diff options
-rw-r--r-- | lib/lib.c | 36 | ||||
-rw-r--r-- | lib/lib.h | 4 |
2 files changed, 25 insertions, 15 deletions
@@ -586,26 +586,34 @@ int64_t peek_be(void *ptr, unsigned size) int64_t peek(void *ptr, unsigned size) { - return IS_BIG_ENDIAN ? peek_be(ptr, size) : peek_le(ptr, size); + return (IS_BIG_ENDIAN ? peek_be : peek_le)(ptr, size); } -void poke(void *ptr, uint64_t val, int size) +void poke_le(void *ptr, long long val, unsigned size) { - if (size & 8) { - volatile uint64_t *p = (uint64_t *)ptr; - *p = val; - } else if (size & 4) { - volatile int *p = (int *)ptr; - *p = val; - } else if (size & 2) { - volatile short *p = (short *)ptr; - *p = val; - } else { - volatile char *p = (char *)ptr; - *p = val; + char *c = ptr; + + while (size--) { + *c++ = val&255; + val >>= 8; } } +void poke_be(void *ptr, long long val, unsigned size) +{ + char *c = ptr + size; + + while (size--) { + *--c = val&255; + val >>=8; + } +} + +void poke(void *ptr, long long val, unsigned size) +{ + (IS_BIG_ENDIAN ? poke_be : poke_le)(ptr, val, size); +} + // Iterate through an array of files, opening each one and calling a function // on that filehandle and name. The special filename "-" means stdin if // flags is O_RDONLY, stdout otherwise. An empty argument list calls @@ -209,7 +209,9 @@ int highest_bit(unsigned long l); int64_t peek_le(void *ptr, unsigned size); int64_t peek_be(void *ptr, unsigned size); int64_t peek(void *ptr, unsigned size); -void poke(void *ptr, uint64_t val, int size); +void poke_le(void *ptr, long long val, unsigned size); +void poke_be(void *ptr, long long val, unsigned size); +void poke(void *ptr, long long val, unsigned size); struct string_list *find_in_path(char *path, char *filename); long long estrtol(char *str, char **end, int base); long long xstrtol(char *str, char **end, int base); |