X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=src%2Fodr_mem.c;h=9d502dfa59478962d71913940c6ffbe74d78dda9;hp=555c4435c0e0f92aaaf384ee57e5cce67b936fb8;hb=ffc8ea2eb23fe8c74fbeb96d05b31c477767923f;hpb=6278ee39573b69b59333a876c52671d0ebbca320 diff --git a/src/odr_mem.c b/src/odr_mem.c index 555c443..9d502df 100644 --- a/src/odr_mem.c +++ b/src/odr_mem.c @@ -1,5 +1,5 @@ /* This file is part of the YAZ toolkit. - * Copyright (C) 1995-2009 Index Data + * Copyright (C) Index Data * See the file LICENSE for details. */ /** @@ -10,6 +10,7 @@ #include #endif +#include #include #include "odr-priv.h" #include @@ -27,7 +28,7 @@ NMEM odr_extract_mem(ODR o) return r; } -void *odr_malloc(ODR o, int size) +void *odr_malloc(ODR o, size_t size) { return nmem_malloc(o->mem, size); } @@ -57,17 +58,15 @@ Odr_bool *odr_booldup(ODR o, Odr_bool v) return nmem_booldup(o->mem, v); } -int odr_total(ODR o) +size_t odr_total(ODR o) { return nmem_total(o->mem); } -Odr_oct *odr_create_Odr_oct(ODR o, const unsigned char *buf, int sz) +Odr_oct *odr_create_Odr_oct(ODR o, const char *buf, int sz) { Odr_oct *p = (Odr_oct *) odr_malloc(o, sizeof(Odr_oct)); - p->buf = (unsigned char *) odr_malloc(o, sz); - memcpy(p->buf, buf, sz); - p->size = sz; + p->buf = odr_strdupn(o, buf, sz); p->len = sz; return p; } @@ -81,65 +80,75 @@ int odr_grow_block(ODR b, int min_bytes) if (!b->op->can_grow) return -1; - if (!b->size) + if (!b->op->size) togrow = 1024; else - togrow = b->size; + togrow = b->op->size; if (togrow < min_bytes) togrow = min_bytes; - if (b->size && !(b->buf = - (unsigned char *) xrealloc(b->buf, b->size += togrow))) + if (b->op->size && !(b->op->buf = + (char *) xrealloc(b->op->buf, b->op->size += togrow))) abort(); - else if (!b->size && !(b->buf = (unsigned char *) - xmalloc(b->size = togrow))) + else if (!b->op->size && !(b->op->buf = (char *) + xmalloc(b->op->size = togrow))) abort(); return 0; } -int odr_write(ODR o, unsigned char *buf, int bytes) +int odr_write(ODR o, const char *buf, int bytes) { - if (o->pos + bytes >= o->size && odr_grow_block(o, bytes)) + if (bytes < 0 || o->op->pos > INT_MAX - bytes) { odr_seterror(o, OSPACE, 40); return -1; } - memcpy(o->buf + o->pos, buf, bytes); - o->pos += bytes; - if (o->pos > o->top) - o->top = o->pos; + if (o->op->pos + bytes >= o->op->size && odr_grow_block(o, bytes)) + { + odr_seterror(o, OSPACE, 40); + return -1; + } + memcpy(o->op->buf + o->op->pos, buf, bytes); + o->op->pos += bytes; + if (o->op->pos > o->op->top) + o->op->top = o->op->pos; return 0; } int odr_seek(ODR o, int whence, int offset) { if (whence == ODR_S_CUR) - offset += o->pos; + offset += o->op->pos; else if (whence == ODR_S_END) - offset += o->top; - if (offset > o->size && odr_grow_block(o, offset - o->size)) + offset += o->op->top; + if (offset > o->op->size && odr_grow_block(o, offset - o->op->size)) { odr_seterror(o, OSPACE, 41); return -1; } - o->pos = offset; + o->op->pos = offset; return 0; } -Odr_int odr_atoi(const char *s) +Odr_int odr_strtol(const char *nptr, char **endptr, int base) { #if NMEM_64 - char *endptr; #if WIN32 - return _strtoui64(s, &endptr, 10); + return _strtoui64(nptr, endptr, base); #else - return strtoll(s, &endptr, 10); + return strtoll(nptr, endptr, base); #endif #else - return atoi(s); + return strtol(nptr, endptr, base); #endif } +Odr_int odr_atoi(const char *s) +{ + char *endptr; + return odr_strtol(s, &endptr, 10); +} + /* * Local variables: * c-basic-offset: 4