| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613 |
- // See LICENSE for license details.
- #include <stdint.h>
- #include <string.h>
- #include <stdarg.h>
- #include <stdio.h>
- #include <limits.h>
- #include <sys/signal.h>
- #include <sys/times.h>
- #include "encoding.h"
- #define SYS_write 64
- #undef strcmp
- static uintptr_t syscall(uintptr_t which, uint64_t arg0, uint64_t arg1, uint64_t arg2)
- {
- volatile uint64_t magic_mem[8] __attribute__((aligned(64)));
- magic_mem[0] = which;
- magic_mem[1] = arg0;
- magic_mem[2] = arg1;
- magic_mem[3] = arg2;
- __sync_synchronize();
- return magic_mem[0];
- }
- void printstr(const char* s)
- {
- syscall(SYS_write, 1, (uintptr_t)s, strlen(s));
- }
- #ifndef AGRV_ENTRY
- #define AGRV_ENTRY main
- #endif
- int __attribute__((weak)) AGRV_ENTRY(int argc, char** argv)
- {
- // single-threaded programs override this function.
- printstr("Implement main()!\n");
- return -1;
- }
- #ifndef NEED_MINIMAL
- extern uint32_t __data_vma;
- extern uint32_t __data_lma;
- extern uint32_t __data_size;
- extern uint32_t __rodata_vma;
- extern uint32_t __rodata_lma;
- extern uint32_t __rodata_size;
- extern uint32_t __sdata_vma;
- extern uint32_t __sdata_lma;
- extern uint32_t __sdata_size;
- extern uint32_t __sram_text_vma;
- extern uint32_t __sram_text_lma;
- extern uint32_t __sram_text_size;
- extern uint32_t __itim_text_vma;
- extern uint32_t __itim_text_lma;
- extern uint32_t __itim_text_size;
- extern uint32_t __exec_text_vma;
- extern uint32_t __exec_text_lma;
- extern uint32_t __exec_text_size;
- extern uint32_t __bss_vma;
- extern uint32_t __bss_size;
- extern uint32_t __sbss_vma;
- extern uint32_t __sbss_size;
- extern void(*__preinit_array_start)();
- extern void(*__preinit_array_end)();
- extern void(*__init_array_start)();
- extern void(*__init_array_end)();
- extern void(*__fini_array_start)();
- extern void(*__fini_array_end)();
- __attribute__((section(".text.init")))
- void _init(void)
- {
- memcpy(&__exec_text_vma, &__exec_text_lma, (int)&__exec_text_size);
- memcpy(&__itim_text_vma, &__itim_text_lma, (int)&__itim_text_size);
- memcpy(&__data_vma, &__data_lma, (int)&__data_size);
- memcpy(&__rodata_vma, &__rodata_lma, (int)&__rodata_size);
- memcpy(&__sdata_vma, &__sdata_lma, (int)&__sdata_size);
- memcpy(&__sram_text_vma, &__sram_text_lma, (int)&__sram_text_size);
- memset(&__bss_vma, 0, (int)&__bss_size);
- memset(&__sbss_vma, 0, (int)&__sbss_size);
- for (void(**p)() = &__preinit_array_start; p < &__preinit_array_end; ++p) {
- (*p)();
- }
- for (void(**p)() = &__init_array_start; p < &__init_array_end; ++p) {
- (*p)();
- }
- int ret = AGRV_ENTRY(0, 0);
- for (void(**p)() = &__fini_array_start; p < &__fini_array_end; ++p) {
- (*p)();
- }
- while (1);
- }
- #endif
- #undef putchar
- __attribute__((weak)) int putchar(int ch)
- {
- return 0;
- }
- void printhex(uint64_t x)
- {
- char str[17];
- int i;
- for (i = 0; i < 16; i++)
- {
- str[15-i] = (x & 0xF) + ((x & 0xF) < 10 ? '0' : 'a'-10);
- x >>= 4;
- }
- str[16] = 0;
- printstr(str);
- }
- size_t printnum(void (*putch)(int, void**, size_t, size_t), void **putdat,
- unsigned long long num, int base, int width, int padc, size_t idx, size_t size)
- {
- unsigned digs[sizeof(num)*CHAR_BIT];
- int pos = 0;
- int hex_base = 'a';
- if (base < 0) {
- hex_base = 'A';
- base = -base;
- }
- while (1)
- {
- unsigned long long div = num / base;
- digs[pos++] = num - div * base;
- if (num < base)
- break;
- num = div;
- }
- while (width-- > pos)
- putch(padc, putdat, idx++, size);
- while (pos-- > 0)
- putch(digs[pos] + (digs[pos] >= 10 ? hex_base - 10 : '0'), putdat, idx++, size);
- return idx;
- }
- static unsigned long long getuint(va_list *ap, int lflag)
- {
- if (lflag >= 2)
- return va_arg(*ap, unsigned long long);
- else if (lflag)
- return va_arg(*ap, unsigned long);
- else
- return va_arg(*ap, unsigned int);
- }
- static long long getint(va_list *ap, int lflag)
- {
- if (lflag >= 2)
- return va_arg(*ap, long long);
- else if (lflag)
- return va_arg(*ap, long);
- else
- return va_arg(*ap, int);
- }
- static float getfloat(va_list *ap)
- {
- return va_arg(*ap, double);
- }
- int vprintfmt(void (*putch)(int, void**, size_t, size_t), void **putdat, const char *fmt, va_list ap, size_t size)
- {
- register const char* p;
- const char* last_fmt;
- register char ch;
- unsigned long long num;
- float fval;
- int base, lflag, width, precision, precision2, altflag;
- char padc;
- size_t idx = 0;
- while (1) {
- while ((ch = *fmt) != '%') {
- if (ch == '\0')
- return idx;
- fmt++;
- putch(ch, putdat, idx++, size);
- }
- fmt++;
- // Process a %-escape sequence
- last_fmt = fmt;
- padc = ' ';
- width = -1;
- precision = -1;
- precision2 = -1;
- lflag = 0;
- altflag = 0;
- reswitch:
- switch (ch = *fmt++) {
- // flag to pad on the right
- case '-':
- padc = '-';
- goto reswitch;
-
- // flag to pad with 0's instead of spaces
- case '0':
- padc = '0';
- goto reswitch;
- // width field
- case '1':
- case '2':
- case '3':
- case '4':
- case '5':
- case '6':
- case '7':
- case '8':
- case '9':
- case '.':
- precision = 0; precision2 = -1;
- for (; ; ++fmt) {
- if (ch == '.') {
- precision2 = 0;
- } else if (precision2 >= 0) {
- precision2 = precision2 * 10 + ch - '0';
- } else {
- precision = precision * 10 + ch - '0';
- }
- ch = *fmt;
-
- if (ch != '.' && (ch < '0' || ch > '9'))
- break;
- }
- goto process_precision;
- case '*':
- precision = va_arg(ap, int);
- goto process_precision;
- #if 0 //ZZZZZZZZ
- case '.':
- if (width < 0)
- width = 0;
- goto reswitch;
- #endif
- case '#':
- altflag = 1;
- goto reswitch;
- process_precision:
- if (width < 0)
- width = precision, precision = -1;
- goto reswitch;
- // long flag (doubled for long long)
- case 'l':
- lflag++;
- goto reswitch;
- // short flag (h or hh), ignored
- case 'h':
- goto reswitch;
- // character
- case 'c':
- putch(va_arg(ap, int), putdat, idx++, size);
- break;
- // string
- case 's':
- if ((p = va_arg(ap, char *)) == NULL)
- p = "(null)";
- if (width > 0 && padc != '-')
- for (width -= strnlen(p, precision); width > 0; width--)
- putch(padc, putdat, idx++, size);
- for (; (ch = *p) != '\0' && (precision < 0 || --precision >= 0); width--) {
- putch(ch, putdat, idx++, size);
- p++;
- if (lflag && *p == '\0') { // Very simple but incomplete way to handle %ls (wchar)
- p++;
- }
- }
- for (; width > 0; width--)
- putch(' ', putdat, idx++, size);
- break;
- // (signed) decimal
- case 'd':
- case 'i':
- num = getint(&ap, lflag);
- if ((long long) num < 0) {
- putch('-', putdat, idx++, size);
- num = -(long long) num;
- }
- base = 10;
- goto signed_number;
- // unsigned decimal
- case 'u':
- base = 10;
- goto unsigned_number;
- // (unsigned) octal
- case 'o':
- // should do something with padding so it's always 3 octits
- base = 8;
- goto unsigned_number;
- // pointer
- case 'p':
- lflag = 1;
- putch('0', putdat, idx++, size);
- putch('x', putdat, idx++, size);
- /* fall through to 'x' */
- // (unsigned) hexadecimal
- case 'x':
- case 'X':
- base = ch == 'x' ? 16 : -16;
- unsigned_number:
- num = getuint(&ap, lflag);
- signed_number:
- idx = printnum(putch, putdat, num, base, width, padc, idx, size);
- break;
- case 'f':
- floating_number:
- fval = getfloat(&ap);
- if (fval < 0) {
- putch('-', putdat, idx++, size);
- fval = -fval;
- }
- if (precision2 < 0) {
- precision2 = 3;
- }
- if (width >= 0) {
- width -= precision2;
- if (precision2 > 0) {
- --width;
- }
- }
- base = 10;
- #ifdef PRINTF_LONG_LONG_FLOAT_SUPPORT
- // Under riscv32imafc this needs about 6K-byte code of soft float library
- num = (int long long)(fval);
- fval -= num;
- #else
- // Without PRINTF_LONG_LONG_FLOAT_SUPPORT floating point numbers greater than INT_MAX will be printed as INT_MAX.
- num = (int)(fval);
- fval -= (unsigned)num;
- #endif
- idx = printnum(putch, putdat, num, base, width, padc, idx, size);
- if (precision2 > 0) {
- num = 0;
- width = precision2;
- padc = '0';
- putch('.', putdat, idx++, size);
- while (precision2-- > 0) {
- fval *= 10.;
- }
- num = (int)(fval+1e-6);
- idx = printnum(putch, putdat, num, base, width, padc, idx, size);
- }
- break;
- // escaped '%' character
- case '%':
- putch(ch, putdat, idx++, size);
- break;
-
- // unrecognized escape sequence - just print it literally
- default:
- putch('%', putdat, idx++, size);
- fmt = last_fmt;
- break;
- }
- }
- return idx;
- }
- static void sprintf_putch(int ch, void** data, size_t idx, size_t size)
- {
- char** pstr = (char**)data;
- **pstr = ch;
- (*pstr)++;
- }
- static void snprintf_putch(int ch, void** data, size_t idx, size_t size)
- {
- if (idx < size) {
- char** pstr = (char**)data;
- **pstr = ch;
- (*pstr)++;
- }
- }
- int puts(const char *s)
- {
- return printf("%s\n", s);
- }
- int vprintf(const char* fmt, va_list ap)
- {
- return vprintfmt((void*)putchar, 0, fmt, ap, 0);
- }
- int vfprintf(FILE *fp, const char* fmt, va_list ap)
- {
- return vprintf(fmt, ap);
- }
- int vdprintf(int fd, const char* fmt, va_list ap)
- {
- return vprintf(fmt, ap);
- }
- int vsprintf(char* str, const char* fmt, va_list ap)
- {
- int ret = vprintfmt(sprintf_putch, (void**)&str, fmt, ap, 0);
- *str = 0;
- return ret;
- }
- int vsnprintf(char* str, size_t size, const char* fmt, va_list ap)
- {
- int ret = vprintfmt(snprintf_putch, (void**)&str, fmt, ap, size-1);
- *str = 0;
- return ret;
- }
- int printf(const char* fmt, ...)
- {
- va_list ap; va_start(ap, fmt);
- int ret = vprintf(fmt, ap);
- va_end(ap); return ret;
- }
- int fprintf(FILE *fp, const char *fmt, ...)
- {
- va_list ap; va_start(ap, fmt);
- int ret = vfprintf(fp, fmt, ap);
- va_end(ap); return ret;
- }
- int dprintf(int fd, const char *fmt, ...)
- {
- va_list ap; va_start(ap, fmt);
- int ret = vdprintf(fd, fmt, ap);
- va_end(ap); return ret;
- }
- int sprintf(char* str, const char* fmt, ...)
- {
- va_list ap; va_start(ap, fmt);
- int ret = vsprintf(str, fmt, ap);
- va_end(ap); return ret;
- }
- int snprintf(char* str, size_t size, const char* fmt, ...)
- {
- va_list ap; va_start(ap, fmt);
- int ret = vsnprintf(str, size, fmt, ap);
- va_end(ap); return ret;
- }
- __attribute__((section(".text.init")))
- void* memcpy(void* dest, const void* src, size_t len)
- {
- if ((((uintptr_t)dest | (uintptr_t)src | len) & (sizeof(uintptr_t)-1)) == 0) {
- const uintptr_t* s = src;
- uintptr_t *d = dest;
- while (d < (uintptr_t*)(dest + len))
- *d++ = *s++;
- } else {
- const char* s = src;
- char *d = dest;
- while (d < (char*)(dest + len))
- *d++ = *s++;
- }
- return dest;
- }
- __attribute__((section(".text.init")))
- void* memset(void* dest, int byte, size_t len)
- {
- if ((((uintptr_t)dest | len) & (sizeof(uintptr_t)-1)) == 0) {
- uintptr_t word = byte & 0xFF;
- word |= word << 8;
- word |= word << 16;
- word |= word << 16 << 16;
- volatile uintptr_t *d = dest;
- while (d < (uintptr_t*)(dest + len))
- *d++ = word;
- } else {
- volatile char *d = dest;
- while (d < (char*)(dest + len))
- *d++ = byte;
- }
- return dest;
- }
- size_t strlen(const char *s)
- {
- const char *p = s;
- while (*p)
- p++;
- return p - s;
- }
- size_t strnlen(const char *s, size_t n)
- {
- const char *p = s;
- while (n-- && *p)
- p++;
- return p - s;
- }
- int strcmp(const char* s1, const char* s2)
- {
- unsigned char c1, c2;
- do {
- c1 = *s1++;
- c2 = *s2++;
- } while (c1 != 0 && c1 == c2);
- return c1 - c2;
- }
- char* strcpy(char* dest, const char* src)
- {
- char* d = dest;
- while ((*d++ = *src++))
- ;
- return dest;
- }
- long atol(const char* str)
- {
- long res = 0;
- int sign = 0;
- while (*str == ' ')
- str++;
- if (*str == '-' || *str == '+') {
- sign = *str == '-';
- str++;
- }
- while (*str) {
- res *= 10;
- res += *str++ - '0';
- }
- return sign ? -res : res;
- }
- // Dummy implementation of system call functions.
- int _close (int file)
- {
- return 0;
- }
- #include <sys/stat.h>
- int _fstat(int file, struct stat *st)
- {
- return 0;
- }
- int _isatty(int file)
- {
- return 0;
- }
- int _lseek (int file, int offset, int whence)
- {
- return 0;
- }
- int _open(const char *pathname, int flags)
- {
- return 0;
- }
- int _kill(int pid, int sig)
- {
- return 0;
- }
- int _getpid(void)
- {
- return 0;
- }
- int _read_r(void)
- {
- return 0;
- }
- int _write_r(void)
- {
- return 0;
- }
- clock_t _times(struct tms *tt)
- {
- return 0;
- }
|