70b6d14546efdcbe2dcb1b74f54a415f261dceb4
[mes.git] / lib / libc.c
1 /* -*-comment-start: "//";comment-end:""-*-
2  * Mes --- Maxwell Equations of Software
3  * Copyright © 2016,2017 Jan (janneke) Nieuwenhuizen <janneke@gnu.org>
4  *
5  * This file is part of Mes.
6  *
7  * Mes is free software; you can redistribute it and/or modify it
8  * under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or (at
10  * your option) any later version.
11  *
12  * Mes is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with Mes.  If not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include <stdarg.h>
22 #include <stdlib.h>
23
24 #if POSIX
25 #define _GNU_SOURCE
26 #include <assert.h>
27 #include <fcntl.h>
28 #include <string.h>
29 #include <unistd.h>
30 #endif // POSIX
31
32 int
33 isdigit (int c)
34 {
35   return (c>='0') && (c<='9');
36 }
37
38 int
39 isxdigit (int c)
40 {
41   return isdigit (c) || (c>='a') && (c<='f');
42 }
43
44 int
45 isspace (int c)
46 {
47   return (c == '\t' || c == '\n' || c == '\v' || c == '\f' || c == '\r' || c == ' ');
48 }
49
50 int
51 isnumber (int c, int base)
52 {
53   if (base == 2)
54     return (c>='0') && (c<='1');
55   if (base == 8)
56     return (c>='0') && (c<='7');
57   if (base == 10)
58     return isdigit (c);
59   if (base == 16)
60     return isxdigit (c);
61 }
62
63 int
64 abtoi (char const **p, int base)
65 {
66   char const *s = *p;
67   int i = 0;
68   int sign = 1;
69   if (!base) base = 10;
70   if (*s && *s == '-')
71     {
72       sign = -1;
73       s++;
74     }
75   while (isnumber (*s, base))
76     {
77       i *= base;
78       int m = *s > '9' ? 'a' - 10 : '0';
79       i += *s - m;
80       s++;
81     }
82   *p = s;
83   return i * sign;
84 }
85
86 int
87 atoi (char const *s)
88 {
89   char const *p = s;
90   return abtoi (&p, 0);
91 }
92
93 char itoa_buf[10];
94
95 char const*
96 itoa (int x)
97 {
98   //static char itoa_buf[10];
99   //char *p = buf+9;
100   char *p = itoa_buf;
101   p += 9;
102   *p-- = 0;
103
104   //int sign = x < 0; // FIXME
105   int sign = 0;
106   if (x < 0) sign = 1;
107   if (sign)
108     x = -x;
109
110   do
111      {
112        *p-- = '0' + (x % 10);
113        x = x / 10;
114      } while (x);
115
116   if (sign && *(p + 1) != '0')
117     *p-- = '-';
118
119   return p+1;
120 }
121
122 char const*
123 itoab (int x, int base)
124 {
125   //static char itoa_buf[10];
126   //char *p = buf+9;
127   char *p = itoa_buf;
128   p += 9;
129   *p-- = 0;
130
131   //int sign = x < 0; // FIXME
132   int sign = 0;
133   if (x < 0) sign = 1;
134   if (sign)
135     x = -x;
136
137   do
138      {
139        int i = x % base;
140        *p-- = i > 9 ? 'a' + i - 10 : '0' + i;
141        x = x / base;
142      } while (x);
143
144   if (sign && *(p + 1) != '0')
145     *p-- = '-';
146
147   return p+1;
148 }
149
150 int
151 fdputc (int c, int fd)
152 {
153   write (fd, (char*)&c, 1);
154   return 0;
155 }
156
157 int
158 fdputs (char const* s, int fd)
159 {
160   int i = strlen (s);
161   write (fd, s, i);
162   return 0;
163 }
164
165 #if !POSIX
166
167 ///char **g_environment = 0; // FIXME: todo extern
168 int g_stdin = 0;
169
170 void _env ();
171
172 int
173 eputc (int c)
174 {
175   return fdputc (c, STDERR);
176 }
177
178 int
179 putchar (int c)
180 {
181   write (STDOUT, (char*)&c, 1);
182   return 0;
183 }
184
185 void
186 assert_fail (char* s)
187 {
188   eputs ("assert fail: ");
189   eputs (s);
190   eputs ("\n");
191   //*((int*)0) = 0;
192   char *fail = s;
193   fail = 0;
194   *fail = 0;
195 }
196
197 int ungetc_char = -1;
198 char ungetc_buf[2];
199
200 int
201 getchar ()
202 {
203   char c;
204   int i;
205   if (ungetc_char == -1)
206     {
207       int r = read (g_stdin, &c, 1);
208       if (r < 1) return -1;
209       i = c;
210    }
211   else
212     {
213        //FIXME
214        //i = ungetc_buf[ungetc_char--];
215        i = ungetc_buf[ungetc_char];
216        //ungetc_char--;
217        ungetc_char = ungetc_char - 1;
218      }
219   if (i < 0) i += 256;
220
221   return i;
222 }
223
224 int
225 fgetc (int fd)
226 {
227   char c;
228   int i;
229   int r = read (fd, &c, 1);
230   if (r < 1) return -1;
231   i = c;
232   return i;
233 }
234
235 void
236 free (void *ptr)
237 {
238 }
239
240 //#define assert(x) ((x) ? (void)0 : assert_fail (#x))
241 int
242 ungetc (int c, int fd)
243 {
244   //FIXME
245   //assert (ungetc_char < 2);
246   //assert (ungetc_char == -1 || ungetc_char < 2);
247   //FIXME
248   //ungetc_buf[++ungetc_char] = c;
249   ungetc_char++;
250   ungetc_buf[ungetc_char] = c;
251   return c;
252 }
253
254 int
255 strcmp (char const* a, char const* b)
256 {
257   while (*a && *b && *a == *b)
258     {
259       a++;b++;
260     }
261   return *a - *b;
262 }
263
264
265 char *
266 strcpy (char *dest, char const *src)
267 {
268   char *p = dest;
269   while (*src) *p++ = *src++;
270   *p = 0;
271   return dest;
272 }
273
274 char *g_brk = 0;
275
276 void *
277 malloc (size_t size)
278 {
279   if (!g_brk)
280     g_brk = brk (0);
281   if (brk (g_brk + size) == -1)
282     return 0;
283   char *p = g_brk;
284   g_brk += size;
285   return p;
286 }
287
288 void *
289 memcpy (void *dest, void const *src, size_t n)
290 {
291   char* p = dest;
292   char* q = src;
293   while (n--) *p++ = *q++;
294   return dest;
295 }
296
297 void *
298 memmove (void *dest, void const *src, size_t n)
299 {
300   if (dest < src)
301     return memcpy (dest, src, n);
302   char *p = dest + n;
303   char const *q = src +n;
304   while (n--)
305     *--p = *--q;
306   return dest;
307 }
308
309 void *
310 realloc (void *ptr, size_t size)
311 {
312   void *new = malloc (size);
313   if (ptr && new)
314     {
315       memmove (new, ptr, size);
316       free (ptr);
317     }
318   return new;
319 }
320
321 int
322 strncmp (char const* a, char const* b, size_t length)
323 {
324   while (*a && *b && *a == *b && --length) {a++;b++;}
325   return *a - *b;
326 }
327
328 char *
329 getenv (char const* s)
330 {
331   char **p = g_environment;
332   int length = strlen (s);
333   while (*p)
334     {
335       if (!strncmp (s, *p, length) && *(*p + length) == '=') return (*p + length + 1);
336       p++;
337     }
338   return 0;
339 }
340
341 int
342 setenv (char const* s, char const* v, int overwrite_p)
343 {
344   char **p = g_environment;
345   int length = strlen (s);
346   while (*p)
347     {
348       if (!strncmp (s, *p, length) && *(*p + length) == '=')
349         break;
350       p++;
351     }
352   char *entry = malloc (length + strlen (v) + 2);
353   int end_p = *p == 0;
354   *p = entry;
355   strcpy (entry, s);
356   strcpy (entry + length, "=");
357   strcpy (entry + length + 1, v);
358   *(entry + length + strlen (v) + 2) = 0;
359   if (end_p)
360     *++p = 0;
361   return 0;
362 }
363
364 int
365 vprintf (char const* format, va_list ap)
366 {
367   char const *p = format;
368   while (*p)
369     if (*p != '%')
370       putchar (*p++);
371     else
372       {
373         p++;
374         char c = *p;
375         switch (c)
376           {
377           case '%': {putchar (*p); break;}
378           case 'c': {char c; c = va_arg (ap, char); putchar (c); break;}
379           case 'd': {int d; d = va_arg (ap, int); puts (itoa (d)); break;}
380           case 's': {char *s; s = va_arg (ap, char *); puts (s); break;}
381           default: {putchar (*p); break;}
382           }
383         p++;
384       }
385   va_end (ap);
386   return 0;
387 }
388
389 int
390 printf (char const* format, ...)
391 {
392   va_list ap;
393   va_start (ap, format);
394   int r = vprintf (format, ap);
395   va_end (ap);
396   return r;
397 }
398
399 int
400 vsprintf (char *str, char const* format, va_list ap)
401 {
402   char const *p = format;
403   while (*p)
404     if (*p != '%')
405       *str++ = *p++;
406     else
407       {
408         p++;
409         char c = *p;
410         switch (c)
411           {
412           case '%': {*str++ = *p; break;}
413           case 'c': {char c; c = va_arg (ap, char); *str++ = c; break;}
414           case 'd': {int d; d = va_arg (ap, int); char const *s; s = itoa (d); while (*s) *str++ = *s++; break;}
415           case 's': {char *s; s = va_arg (ap, char *); while (*s) *str++ = *s++; break;}
416           default: {*str++ = *p; break;}
417           }
418         p++;
419       }
420   va_end (ap);
421   *str = 0;
422   return strlen (str);
423 }
424
425 int
426 sprintf (char *str, char const* format, ...)
427 {
428   va_list ap;
429   va_start (ap, format);
430   int r = vsprintf (str, format, ap);
431   va_end (ap);
432   return r;
433 }
434 #endif //!POSIX