1 /*-
2 * Copyright (c) 2014 Pedro Souza <[email protected]>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 */
27
28 #include <sys/cdefs.h>
29 __FBSDID("$FreeBSD$");
30
31 #include "lstd.h"
32 #include "math.h"
33
34 #ifdef LOADER_VERIEXEC
35 #include <verify_file.h>
36 #endif
37
38 FILE *
fopen(const char * filename,const char * mode)39 fopen(const char *filename, const char *mode)
40 {
41 struct stat st;
42 int fd, m, o;
43 FILE *f;
44
45 if (mode == NULL)
46 return NULL;
47
48 switch (*mode++) {
49 case 'r': /* open for reading */
50 m = O_RDONLY;
51 o = 0;
52 break;
53
54 case 'w': /* open for writing */
55 m = O_WRONLY;
56 /* These are not actually implemented yet */
57 o = O_CREAT | O_TRUNC;
58 break;
59
60 default: /* illegal mode */
61 return (NULL);
62 }
63
64 if (*mode == '+')
65 m = O_RDWR;
66
67 fd = open(filename, m | o);
68 if (fd < 0)
69 return NULL;
70
71 f = malloc(sizeof(FILE));
72 if (f == NULL) {
73 close(fd);
74 return NULL;
75 }
76
77 if (fstat(fd, &st) != 0) {
78 free(f);
79 close(fd);
80 return (NULL);
81 }
82
83 #ifdef LOADER_VERIEXEC
84 /* only regular files and only reading makes sense */
85 if (S_ISREG(st.st_mode) && !(m & O_WRONLY)) {
86 if (verify_file(fd, filename, 0, VE_GUESS) < 0) {
87 free(f);
88 close(fd);
89 return (NULL);
90 }
91 }
92 #endif
93
94 f->fd = fd;
95 f->offset = 0;
96 f->size = st.st_size;
97
98 return (f);
99 }
100
101
102 FILE *
freopen(const char * filename,const char * mode,FILE * stream)103 freopen(const char *filename, const char *mode, FILE *stream)
104 {
105 fclose(stream);
106 return (fopen(filename, mode));
107 }
108
109 size_t
fread(void * ptr,size_t size,size_t count,FILE * stream)110 fread(void *ptr, size_t size, size_t count, FILE *stream)
111 {
112 size_t r;
113
114 if (stream == NULL)
115 return 0;
116 r = (size_t)read(stream->fd, ptr, size * count);
117 stream->offset += r;
118
119 return (r);
120 }
121
122 size_t
fwrite(const void * ptr,size_t size,size_t count,FILE * stream)123 fwrite(const void *ptr, size_t size, size_t count, FILE *stream)
124 {
125 ssize_t w;
126
127 if (stream == NULL || ptr == NULL)
128 return (0);
129 w = write(stream->fd, ptr, size * count);
130 if (w == -1)
131 return (0);
132
133 stream->offset += w;
134 return ((size_t)w);
135 }
136
137 int
fclose(FILE * stream)138 fclose(FILE *stream)
139 {
140 if (stream == NULL)
141 return EOF;
142 close(stream->fd);
143 free(stream);
144
145 return (0);
146 }
147
148 int
ferror(FILE * stream)149 ferror(FILE *stream)
150 {
151
152 return (stream == NULL || stream->fd < 0);
153 }
154
155 int
feof(FILE * stream)156 feof(FILE *stream)
157 {
158
159 if (stream == NULL)
160 return 1;
161
162 return (stream->offset >= stream->size);
163 }
164
165 int
getc(FILE * stream)166 getc(FILE *stream)
167 {
168 char ch;
169 size_t r;
170
171 if (stream == NULL)
172 return EOF;
173 r = read(stream->fd, &ch, 1);
174 if (r == 1)
175 return ch;
176 return EOF;
177 }
178
179 DIR *
opendir(const char * name)180 opendir(const char *name)
181 {
182 DIR *dp;
183 int fd;
184
185 fd = open(name, O_RDONLY);
186 if (fd < 0)
187 return NULL;
188 dp = fdopendir(fd);
189 if (dp == NULL)
190 close(fd);
191 return dp;
192 }
193
194 DIR *
fdopendir(int fd)195 fdopendir(int fd)
196 {
197 DIR *dp;
198
199 dp = malloc(sizeof(*dp));
200 if (dp == NULL)
201 return NULL;
202 dp->fd = fd;
203 return dp;
204 }
205
206 int
closedir(DIR * dp)207 closedir(DIR *dp)
208 {
209 close(dp->fd);
210 dp->fd = -1;
211 free(dp);
212 return 0;
213 }
214
215 void
luai_writestring(const char * s,int i)216 luai_writestring(const char *s, int i)
217 {
218
219 while (i-- > 0)
220 putchar(*s++);
221 }
222
223 /*
224 * These routines from here on down are to implement the lua math
225 * library, but that's not presently included by default. They are
226 * little more than placeholders to allow compilation due to linkage
227 * issues with upstream Lua.
228 */
229
230 int64_t
lstd_pow(int64_t x,int64_t y)231 lstd_pow(int64_t x, int64_t y)
232 {
233 int64_t rv = 1;
234
235 if (y < 0)
236 return 0;
237 rv = x;
238 while (--y)
239 rv *= x;
240
241 return rv;
242 }
243
244 int64_t
lstd_floor(int64_t x)245 lstd_floor(int64_t x)
246 {
247
248 return (x);
249 }
250
251 int64_t
lstd_fmod(int64_t a,int64_t b)252 lstd_fmod(int64_t a, int64_t b)
253 {
254
255 return (a % b);
256 }
257
258 /*
259 * This can't be implemented, so maybe it should just abort.
260 */
261 int64_t
lstd_frexp(int64_t a,int * y)262 lstd_frexp(int64_t a, int *y)
263 {
264 *y = 0;
265
266 return 0;
267 }
268