You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.7 KiB
C
81 lines
2.7 KiB
C
#ifdef __dietlibc__
|
|
#include <sys/cdefs.h>
|
|
#else
|
|
#define __likely(x) x
|
|
#define __unlikely(x) x
|
|
#endif
|
|
#include <sys/types.h>
|
|
#include <stdlib.h>
|
|
#include "safemult.h"
|
|
#include "array.h"
|
|
#include "byte.h"
|
|
|
|
#if 0
|
|
static array x;
|
|
t *p;
|
|
int64 pos;
|
|
|
|
p = array_allocate(&x,sizeof(t),pos);
|
|
|
|
array_allocate makes sure that enough bytes are allocated in x for at
|
|
least pos+1 objects of type t. (The size of t must be positive;
|
|
otherwise the effects are undefined.) If not enough bytes are
|
|
allocated (or x is unallocated), array_allocate allocates more bytes,
|
|
moving the dynamically allocated region if necessary. array_allocate
|
|
often allocates somewhat more bytes than necessary, to save time
|
|
later.
|
|
|
|
array_allocate then makes sure that the number of bytes initialized
|
|
covers at least those pos+1 objects. If not enough bytes are
|
|
initialized, array_allocate initializes more bytes (setting them to
|
|
0), up to exactly the end of the pos+1st object.
|
|
|
|
array_allocate then returns a pointer to the pos+1st object; i.e.,
|
|
object number pos, with objects numbered starting at 0. This pointer
|
|
can be used to change or inspect the object. The pointer can continue
|
|
to be used through subsequent calls to array_get, array_start,
|
|
array_length, and array_bytes, but it must not be used after any
|
|
other operations on this array.
|
|
|
|
If something goes wrong, array_allocate returns 0, setting errno
|
|
appropriately, without touching x. In particular, array_allocate
|
|
returns 0 if
|
|
|
|
* x has failed, or
|
|
* pos is negative, or
|
|
* not enough memory is available.
|
|
|
|
array_allocate does not change x to have failed; if you want to do
|
|
that, use array_fail.
|
|
#endif
|
|
|
|
void* array_allocate(array* x,uint64 membersize,int64 pos) {
|
|
uint64 wanted;
|
|
if (__unlikely(x->allocated<0)) return 0; /* array is failed */
|
|
if (__unlikely(pos+1<1)) return 0;
|
|
/* second case of overflow: pos*membersize too large */
|
|
if (__unlikely(!umult64(membersize,pos+1,&wanted))) return 0;
|
|
|
|
if (__unlikely(wanted >= (uint64)x->allocated)) {
|
|
/* round up a little */
|
|
if (membersize<8)
|
|
wanted=(wanted+127)&(-128ll); /* round up to multiple of 128 */
|
|
else
|
|
wanted=(wanted+4095)&(-4096ll); /* round up to 4k pages */
|
|
|
|
if (__unlikely(wanted<128)) return 0; /* overflow during rounding */
|
|
|
|
if (sizeof(size_t) != sizeof(int64) && __unlikely((size_t)(wanted) != wanted))
|
|
return 0;
|
|
{
|
|
char* tmp=realloc(x->p,wanted);
|
|
if (__unlikely(!tmp)) return 0;
|
|
x->p=tmp;
|
|
}
|
|
x->allocated=wanted;
|
|
byte_zero(x->p+x->initialized,x->allocated-x->initialized);
|
|
}
|
|
x->initialized=(pos+1)*membersize;
|
|
return x->p+pos*membersize;
|
|
}
|