diff --git a/include/stdalign.h b/include/stdalign.h new file mode 100644 index 0000000000000000000000000000000000000000..b6e50aebdbfd65c8f142c26174710fb22e02b4e3 --- /dev/null +++ b/include/stdalign.h @@ -0,0 +1,15 @@ +#ifndef _STDALIGN_H +#define _STDALIGN_H + +/* this whole header only works in C11 or with compiler extensions */ +#if __STDC_VERSION__ < 201112L && defined( __GNUC__) +#define _Alignas(t) __attribute__((__aligned__(t))) +#define _Alignof(t) __alignof__(t) +#endif + +#define alignas _Alignas +#define alignof _Alignof +#define __alignas_is_defined 1 +#define __alignof_is_defined 1 + +#endif diff --git a/include/stdio.h b/include/stdio.h index 19ab8bcdda3e87bba4dd367688af3d7161ae1fe9..9aa2f782627ca8acbfcc76a472388d942de1b902 100644 --- a/include/stdio.h +++ b/include/stdio.h @@ -90,7 +90,9 @@ int putc(int, FILE *); int putchar(int); char *fgets(char *, int, FILE *); +#if __STDC_VERSION__ < 201112L char *gets(char *); +#endif int fputs(const char *, FILE *); int puts(const char *); diff --git a/include/stdlib.h b/include/stdlib.h index c1d93e71595744d521a7786f4bb2011a928298b9..1749cb3b8c10b520cc32f76fbfb59b2fbfc13ae8 100644 --- a/include/stdlib.h +++ b/include/stdlib.h @@ -38,6 +38,7 @@ void *malloc (size_t); void *calloc (size_t, size_t); void *realloc (void *, size_t); void free (void *); +void *aligned_alloc(size_t alignment, size_t size); void abort (void); int atexit (void (*) (void)); diff --git a/src/malloc/aligned_alloc.c b/src/malloc/aligned_alloc.c new file mode 100644 index 0000000000000000000000000000000000000000..d623420b42c8427bfce2babe8c1a9ce055a939a0 --- /dev/null +++ b/src/malloc/aligned_alloc.c @@ -0,0 +1,51 @@ +#include +#include +#include + +/* This function should work with most dlmalloc-like chunk bookkeeping + * systems, but it's only guaranteed to work with the native implementation + * used in this library. */ + +void *aligned_alloc(size_t align, size_t len) +{ + unsigned char *mem, *new, *end; + size_t header, footer; + + if ((align & -align & -sizeof(void *)) != align) { + errno = EINVAL; + return NULL; + } + + if (len > SIZE_MAX - align) { + errno = ENOMEM; + return NULL; + } + + if (align <= 4*sizeof(size_t)) { + if (!(mem = malloc(len))) + return NULL; + return mem; + } + + if (!(mem = malloc(len + align-1))) + return NULL; + + header = ((size_t *)mem)[-1]; + end = mem + (header & -8); + footer = ((size_t *)end)[-2]; + new = (void *)((uintptr_t)mem + align-1 & -align); + + if (!(header & 7)) { + ((size_t *)new)[-2] = ((size_t *)mem)[-2] + (new-mem); + ((size_t *)new)[-1] = ((size_t *)mem)[-1] - (new-mem); + return new; + } + + ((size_t *)mem)[-1] = header&7 | new-mem; + ((size_t *)new)[-2] = footer&7 | new-mem; + ((size_t *)new)[-1] = header&7 | end-new; + ((size_t *)end)[-2] = footer&7 | end-new; + + if (new != mem) free(mem); + return new; +} diff --git a/src/malloc/memalign.c b/src/malloc/memalign.c index 61f456e48667e6a717802eadc5a471d46a8121a7..a594c636884f08c331d6dfc1ebc290b45b5b1a7c 100644 --- a/src/malloc/memalign.c +++ b/src/malloc/memalign.c @@ -3,11 +3,5 @@ void *memalign(size_t align, size_t len) { - void *mem; - int ret; - if ((ret = posix_memalign(&mem, align, len))) { - errno = ret; - return 0; - } - return mem; + return aligned_alloc(align, len); } diff --git a/src/malloc/posix_memalign.c b/src/malloc/posix_memalign.c index 2ae928c84e3439d6782adc2882f116c2f3d4603a..42cf27405a23d4b50dfb32df6bbba854aa0675ff 100644 --- a/src/malloc/posix_memalign.c +++ b/src/malloc/posix_memalign.c @@ -1,47 +1,10 @@ #include -#include #include -/* This function should work with most dlmalloc-like chunk bookkeeping - * systems, but it's only guaranteed to work with the native implementation - * used in this library. */ - int posix_memalign(void **res, size_t align, size_t len) { - unsigned char *mem, *new, *end; - size_t header, footer; - - if ((align & -align & -sizeof(void *)) != align) return EINVAL; - if (len > SIZE_MAX - align) return ENOMEM; - - if (align <= 4*sizeof(size_t)) { - if (!(mem = malloc(len))) - return errno; - *res = mem; - return 0; - } - - if (!(mem = malloc(len + align-1))) - return errno; - - header = ((size_t *)mem)[-1]; - end = mem + (header & -8); - footer = ((size_t *)end)[-2]; - new = (void *)((uintptr_t)mem + align-1 & -align); - - if (!(header & 7)) { - ((size_t *)new)[-2] = ((size_t *)mem)[-2] + (new-mem); - ((size_t *)new)[-1] = ((size_t *)mem)[-1] - (new-mem); - *res = new; - return 0; - } - - ((size_t *)mem)[-1] = header&7 | new-mem; - ((size_t *)new)[-2] = footer&7 | new-mem; - ((size_t *)new)[-1] = header&7 | end-new; - ((size_t *)end)[-2] = footer&7 | end-new; - - if (new != mem) free(mem); - *res = new; + void *mem = aligned_alloc(align, len); + if (!mem) return errno; + *res = mem; return 0; } diff --git a/src/stdio/fopen.c b/src/stdio/fopen.c index 084cc73ce5ab7014821c3774f120c2fbd07d60c3..0d5b1af25c81f74ffe7b19e2d1dab5d2d6cb3717 100644 --- a/src/stdio/fopen.c +++ b/src/stdio/fopen.c @@ -5,7 +5,6 @@ FILE *fopen(const char *filename, const char *mode) FILE *f; int fd; int flags; - int plus = !!strchr(mode, '+'); /* Check for valid initial mode character */ if (!strchr("rwa", *mode)) { @@ -14,9 +13,10 @@ FILE *fopen(const char *filename, const char *mode) } /* Compute the flags to pass to open() */ - if (plus) flags = O_RDWR; + if (strchr(mode, '+')) flags = O_RDWR; else if (*mode == 'r') flags = O_RDONLY; else flags = O_WRONLY; + if (strchr(mode, 'x')) flags |= O_EXCL; if (*mode != 'r') flags |= O_CREAT; if (*mode == 'w') flags |= O_TRUNC; if (*mode == 'a') flags |= O_APPEND;