mescc: Tinycc support: eputc.
[mes.git] / mlibc / libc-gcc.c
1 /* -*-comment-start: "//";comment-end:""-*-
2  * Mes --- Maxwell Equations of Software
3  * Copyright © 2016,2017 Jan 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 char **g_environment = 0;
22 int g_stdin = 0;
23
24 #include <stdio.h>
25 #include <mlibc.h>
26
27 #if __GNUC__
28 #include <stdlib.h>
29 #endif
30
31 #if __GNUC__ && !POSIX
32
33 void
34 exit (int code)
35 {
36   asm (
37        "mov    %0,%%ebx\n\t"
38        "mov    $1,%%eax\n\t"
39        "int    $0x80"
40        : // no outputs "=" (r)
41        : "" (code)
42        );
43   // not reached
44   exit (0);
45 }
46
47 int
48 read (int fd, void* buf, size_t n)
49 {
50   int r;
51   //syscall (SYS_write, fd, s, n));
52   asm (
53        "mov    %1,%%ebx\n\t"
54        "mov    %2,%%ecx\n\t"
55        "mov    %3,%%edx\n\t"
56
57        "movl $0x3,%%eax\n\t"
58        "int  $0x80\n\t"
59
60        "mov    %%eax,%0\n\t"
61        : "=r" (r)
62        : "" (fd), "" (buf), "" (n)
63        : "eax", "ebx", "ecx", "edx"
64        );
65   return r;
66 }
67
68 int
69 write (int fd, char const* s, int n)
70 {
71   int r;
72   //syscall (SYS_write, fd, s, n));
73   asm (
74        "mov    %1,%%ebx\n\t"
75        "mov    %2,%%ecx\n\t"
76        "mov    %3,%%edx\n\t"
77
78        "mov    $0x4, %%eax\n\t"
79        "int    $0x80\n\t"
80        "mov    %%eax,%0\n\t"
81        : "=r" (r)
82        : "" (fd), "" (s), "" (n)
83        : "eax", "ebx", "ecx", "edx"
84        );
85   return r;
86 }
87
88 int
89 open (char const *s, int flags, ...)
90 {
91   int mode;
92   asm (
93        "mov    %%ebp,%%eax\n\t"
94        "add    $0x10,%%eax\n\t"
95        "mov    (%%eax),%%eax\n\t"
96        "mov    %%eax,%0\n\t"
97        : "=mode" (mode)
98        : //no inputs ""
99        );
100   int r;
101   //syscall (SYS_open, mode));
102   asm (
103        "mov    %1,%%ebx\n\t"
104        "mov    %2,%%ecx\n\t"
105        "mov    %3,%%edx\n\t"
106        "mov    $0x5,%%eax\n\t"
107        "int    $0x80\n\t"
108        "mov    %%eax,%0\n\t"
109        : "=r" (r)
110        : "" (s), "" (flags), "" (mode)
111        : "eax", "ebx", "ecx", "edx"
112        );
113   return r;
114 }
115
116 int
117 access (char const *s, int mode)
118 {
119   int r;
120   //syscall (SYS_access, mode));
121   asm (
122        "mov    %1,%%ebx\n\t"
123        "mov    %2,%%ecx\n\t"
124        "mov    $0x21,%%eax\n\t"
125        "int    $0x80\n\t"
126        "mov    %%eax,%0\n\t"
127        : "=r" (r)
128        : "" (s), "" (mode)
129        : "eax", "ebx", "ecx"
130        );
131   return r;
132 }
133
134 void *
135 brk (void *p)
136 {
137   void *r;
138   asm (
139        "mov    %1,%%ebx\n\t"
140
141        "mov    $0x2d,%%eax\n\t"
142        "int    $0x80\n\t"
143
144        "mov    %%eax,%0\n\t"
145        : "=r" (r)
146        : "" (p)
147        : "eax", "ebx"
148        );
149   return r;
150 }
151
152 int
153 fsync (int fd)
154 {
155   int r;
156   //syscall (SYS_fsync, fd));
157   asm (
158        "mov    %1,%%ebx\n\t"
159
160        "mov    $0x76, %%eax\n\t"
161        "int    $0x80\n\t"
162        "mov    %%eax,%0\n\t"
163        : "=r" (r)
164        : "" (fd)
165        : "eax", "ebx"
166        );
167   return r;
168 }
169
170 int
171 fputc (int c, int fd)
172 {
173   write (fd, (char*)&c, 1);
174   return 0;
175 }
176
177 void
178 free (void *ptr)
179 {
180 }
181
182 char *g_brk = 0;
183
184 void *
185 malloc (size_t size)
186 {
187   if (!g_brk)
188     g_brk = brk (0);
189   if (brk (g_brk + size) == -1)
190     return 0;
191   char *p = g_brk;
192   g_brk += size;
193   return p;
194 }
195
196 void *
197 memcpy (void *dest, void const *src, size_t n)
198 {
199   char* p = dest;
200   char* q = src;
201   while (n--) *p++ = *q++;
202   return dest;
203 }
204
205 int
206 putchar (int c)
207 {
208   write (STDOUT, (char*)&c, 1);
209   return 0;
210 }
211
212 void *
213 realloc (void *ptr, size_t size)
214 {
215   void *new = malloc (size);
216   if (ptr && new)
217     {
218       memcpy (new, ptr, size);
219       free (ptr);
220     }
221   return new;
222 }
223
224 size_t
225 strlen (char const* s)
226 {
227   int i = 0;
228   while (s[i]) i++;
229   return i;
230 }
231
232 int
233 strcmp (char const* a, char const* b)
234 {
235   while (*a && *b && *a == *b) {a++;b++;}
236   return *a - *b;
237 }
238
239 char *
240 strcpy (char *dest, char const *src)
241 {
242   char *p = dest;
243   while (*src) *p++ = *src++;
244   *p = 0;
245   return dest;
246 }
247
248 int
249 fputs (char const* s, int fd)
250 {
251   int i = strlen (s);
252   write (fd, s, i);
253   return 0;
254 }
255
256 int
257 puts (char const* s)
258 {
259   int i = strlen (s);
260   write (STDOUT, s, i);
261   return 0;
262 }
263
264 void
265 assert_fail (char* s)
266 {
267   eputs ("assert fail: ");
268   eputs (s);
269   eputs ("\n");
270   *((int*)0) = 0;
271 }
272
273 #define assert(x) ((x) ? (void)0 : assert_fail (#x))
274
275 int ungetc_char = -1;
276 char ungetc_buf[2];
277
278 int
279 getchar ()
280 {
281   char c;
282   int i;
283   if (ungetc_char == -1)
284     {
285       int r = read (g_stdin, &c, 1);
286       if (r < 1) return -1;
287       i = c;
288     }
289   else
290     i = ungetc_buf[ungetc_char--];
291
292   if (i < 0) i += 256;
293
294   return i;
295 }
296
297 int
298 ungetc (int c, int fd)
299 {
300   assert (ungetc_char < 2);
301   ungetc_buf[++ungetc_char] = c;
302   return c;
303 }
304
305 char const* itoa (int);
306
307 int
308 strncmp (char const* a, char const* b, size_t length)
309 {
310   while (*a && *b && *a == *b && --length) {a++;b++;}
311   return *a - *b;
312 }
313
314 char *
315 getenv (char const* s)
316 {
317   char **p = g_environment;
318   int length = strlen (s);
319   while (*p)
320     {
321       if (!strncmp (s, *p, length) && *(*p + length) == '=') return (*p + length + 1);
322       p++;
323     }
324   return 0;
325 }
326
327 int
328 isdigit (int c)
329 {
330   return (c>='0') && (c<='9');
331 }
332
333 int
334 atoi (char const *s)
335 {
336   int i = 0;
337   int sign = 1;
338   if (*s && *s == '-')
339     {
340       sign = -1;
341       s++;
342     }
343   while (isdigit (*s))
344     {
345       i *= 10;
346       i += (*s - '0');
347       s++;
348     }
349   return i * sign;
350 }
351
352
353 // FIXME: copied from libc-mes.c now
354 #include <stdarg.h>
355
356 int
357 vprintf (char const* format, va_list ap)
358 {
359   char const *p = format;
360   while (*p)
361     if (*p != '%')
362       putchar (*p++);
363     else
364       {
365         p++;
366         char c = *p;
367         switch (c)
368           {
369           case '%': {putchar (*p); break;}
370           case 'c': {char c; c = va_arg (ap, char); putchar (c); break;}
371           case 'd': {int d; d = va_arg (ap, int); puts (itoa (d)); break;}
372           case 's': {char *s; s = va_arg (ap, char *); puts (s); break;}
373           default: putchar (*p);
374           }
375         p++;
376       }
377   va_end (ap);
378   return 0;
379 }
380
381 int
382 printf (char const* format, ...)
383 {
384   va_list ap;
385   va_start (ap, format);
386   int r = vprintf (format, ap);
387   va_end (ap);
388   return r;
389 }
390
391 int
392 sprintf (char *str, char const* format, ...)
393 {
394   va_list ap;
395   va_start (ap, format);
396   char const *p = format;
397   while (*p)
398     if (*p != '%')
399       *str++ = *p++;
400     else
401       {
402         p++;
403         char c = *p;
404         switch (c)
405           {
406           case '%': {*str++ = *p; break;}
407           case 'c': {char c; c = va_arg (ap, char); *str++ = c; break;}
408           case 'd': {int d; d = va_arg (ap, int); char const *s = itoa (d); while (*s) *str++ = *s++; break;}
409           case 's': {char *s; s = va_arg (ap, char *); while (*s) *str++ = *s++; break;}
410           default: *str++ = *p;
411           }
412         p++;
413       }
414   va_end (ap);
415   return 0;
416 }
417 #endif
418
419 char itoa_buf[10];
420
421 char const*
422 itoa (int x)
423 {
424   //static char itoa_buf[10];
425   //char *p = buf+9;
426   char *p = itoa_buf;
427   p += 9;
428   *p-- = 0;
429
430   //int sign = x < 0;
431   int sign;
432   sign = x < 0;
433   if (sign)
434     x = -x;
435
436   do
437     {
438       *p-- = '0' + (x % 10);
439       x = x / 10;
440     } while (x);
441
442   if (sign)
443     *p-- = '-';
444
445   return p+1;
446 }
447
448 #if POSIX
449 #define _GNU_SOURCE
450 #include <assert.h>
451 #include <fcntl.h>
452 #include <string.h>
453 #include <unistd.h>
454 #endif // POSIX
455
456 int
457 fdputs (char const* s, int fd)
458 {
459   int i = strlen (s);
460   write (fd, s, i);
461   return 0;
462 }
463
464 int
465 eputc (int c)
466 {
467   return fdputc (c, STDERR);
468 }
469
470 int
471 eputs (char const* s)
472 {
473   return fdputs (s, STDERR);
474 }
475
476 int
477 fdputc (int c, int fd)
478 {
479   write (fd, (char*)&c, 1);
480   return 0;
481 }
482
483 #if POSIX
484
485 int
486 putchar (int c)
487 {
488   write (STDOUT, (char*)&c, 1);
489   return 0;
490 }
491
492 int ungetc_char = -1;
493 char ungetc_buf[2];
494
495 int
496 getchar ()
497 {
498   char c;
499   int i;
500   if (ungetc_char == -1)
501     {
502       int r = read (g_stdin, &c, 1);
503       if (r < 1) return -1;
504       i = c;
505     }
506   else
507     i = ungetc_buf[ungetc_char--];
508
509   if (i < 0) i += 256;
510
511   return i;
512 }
513
514 int
515 fdungetc (int c, int fd)
516 {
517   assert (ungetc_char < 2);
518   ungetc_buf[++ungetc_char] = c;
519   return c;
520 }
521
522 #endif // POSIX
523
524 #if __GNUC__ && !POSIX
525
526 void
527 _start ()
528 {
529   //  char **;
530   asm (
531        "mov     %%ebp,%%eax\n\t"
532        "addl    $4,%%eax\n\t"
533        "movzbl  (%%eax),%%eax\n\t"
534        "addl    $3,%%eax\n\t"
535        "shl     $2,%%eax\n\t"
536        "add     %%ebp,%%eax\n\t"
537        "movl    %%eax,%0\n\t"
538        : "=g_environment" (g_environment)
539        : //no inputs ""
540        );
541   int r;
542   asm (
543        "mov     %%ebp,%%eax\n\t"
544        "addl    $8,%%eax\n\t"
545        "push    %%eax\n\t"
546
547        "mov     %%ebp,%%eax\n\t"
548        "addl    $4,%%eax\n\t"
549        "movzbl  (%%eax),%%eax\n\t"
550        "push     %%eax\n\t"
551
552        "call    main\n\t"
553        "movl    %%eax,%0\n\t"
554        : "=r" (r)
555        : //no inputs ""
556        );
557   exit (r);
558 }
559
560 #endif // __GNUC__ && !POSIX