ArrayList.c

 
#include <winpr/config.h>
 
#include <stdarg.h>
 
#include <winpr/crt.h>
#include <winpr/assert.h>
#include <winpr/collections.h>
 
#if defined(_WIN32) && (_MSC_VER < 1800) && !defined(__MINGW32__)
#define va_copy(dest, src) (dest = src)
#endif
 
struct s_wArrayList
{
  size_t capacity;
  size_t growthFactor;
  BOOL synchronized;
 
  size_t size;
  void** array;
  CRITICAL_SECTION lock;
 
  wObject object;
};
 
size_t ArrayList_Capacity(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  return arrayList->capacity;
}
 
size_t ArrayList_Count(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  return arrayList->size;
}
 
size_t ArrayList_Items(wArrayList* arrayList, ULONG_PTR** ppItems)
{
  WINPR_ASSERT(arrayList);
  *ppItems = (ULONG_PTR*)arrayList->array;
  return arrayList->size;
}
 
BOOL ArrayList_IsFixedSized(WINPR_ATTR_UNUSED wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  return FALSE;
}
 
BOOL ArrayList_IsReadOnly(WINPR_ATTR_UNUSED wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  return FALSE;
}
 
BOOL ArrayList_IsSynchronized(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  return arrayList->synchronized;
}
 
static void ArrayList_Lock_Conditional(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  if (arrayList->synchronized)
    EnterCriticalSection(&arrayList->lock);
}
 
void ArrayList_Lock(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  EnterCriticalSection(&arrayList->lock);
}
 
static void ArrayList_Unlock_Conditional(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  if (arrayList->synchronized)
    LeaveCriticalSection(&arrayList->lock);
}
 
void ArrayList_Unlock(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  LeaveCriticalSection(&arrayList->lock);
}
 
void* ArrayList_GetItem(wArrayList* arrayList, size_t index)
{
  void* obj = NULL;
 
  WINPR_ASSERT(arrayList);
  if (index < arrayList->size)
  {
    obj = arrayList->array[index];
  }
 
  return obj;
}
 
BOOL ArrayList_SetItem(wArrayList* arrayList, size_t index, const void* obj)
{
  WINPR_ASSERT(arrayList);
  if (index >= arrayList->size)
    return FALSE;
 
  if (arrayList->object.fnObjectNew)
  {
    arrayList->array[index] = arrayList->object.fnObjectNew(obj);
    if (obj && !arrayList->array[index])
      return FALSE;
  }
  else
  {
    union
    {
      const void* cpv;
      void* pv;
    } cnv;
    cnv.cpv = obj;
    arrayList->array[index] = cnv.pv;
  }
  return TRUE;
}
 
static BOOL ArrayList_EnsureCapacity(wArrayList* arrayList, size_t count)
{
  WINPR_ASSERT(arrayList);
  WINPR_ASSERT(count > 0);
 
  if (arrayList->size + count > arrayList->capacity)
  {
    void** newArray = NULL;
    size_t newCapacity = arrayList->capacity * arrayList->growthFactor;
    if (newCapacity < arrayList->size + count)
      newCapacity = arrayList->size + count;
 
    newArray = (void**)realloc((void*)arrayList->array, sizeof(void*) * newCapacity);
 
    if (!newArray)
      return FALSE;
 
    arrayList->array = newArray;
    arrayList->capacity = newCapacity;
  }
 
  return TRUE;
}
static BOOL ArrayList_Shift(wArrayList* arrayList, size_t index, SSIZE_T count)
{
  WINPR_ASSERT(arrayList);
  if (count > 0)
  {
    if (!ArrayList_EnsureCapacity(arrayList, (size_t)count))
      return FALSE;
 
    MoveMemory((void*)&arrayList->array[index + (size_t)count], (void*)&arrayList->array[index],
               (arrayList->size - index) * sizeof(void*));
    arrayList->size += (size_t)count;
  }
  else if (count < 0)
  {
    const size_t scount = WINPR_ASSERTING_INT_CAST(size_t, -count);
    const size_t off = index + scount;
    if (off < arrayList->size)
    {
      const size_t chunk = arrayList->size - off;
      MoveMemory((void*)&arrayList->array[index], (void*)&arrayList->array[off],
                 chunk * sizeof(void*));
    }
 
    arrayList->size -= scount;
  }
 
  return TRUE;
}
 
void ArrayList_Clear(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  for (size_t index = 0; index < arrayList->size; index++)
  {
    if (arrayList->object.fnObjectFree)
      arrayList->object.fnObjectFree(arrayList->array[index]);
 
    arrayList->array[index] = NULL;
  }
 
  arrayList->size = 0;
 
  ArrayList_Unlock_Conditional(arrayList);
}
 
BOOL ArrayList_Contains(wArrayList* arrayList, const void* obj)
{
  BOOL rc = FALSE;
 
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  for (size_t index = 0; index < arrayList->size; index++)
  {
    rc = arrayList->object.fnObjectEquals(arrayList->array[index], obj);
 
    if (rc)
      break;
  }
 
  ArrayList_Unlock_Conditional(arrayList);
 
  return rc;
}
 
#if defined(WITH_WINPR_DEPRECATED)
int ArrayList_Add(wArrayList* arrayList, const void* obj)
{
  WINPR_ASSERT(arrayList);
  if (!ArrayList_Append(arrayList, obj))
    return -1;
  return (int)ArrayList_Count(arrayList) - 1;
}
#endif
 
BOOL ArrayList_Append(wArrayList* arrayList, const void* obj)
{
  size_t index = 0;
  BOOL rc = FALSE;
 
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  if (!ArrayList_EnsureCapacity(arrayList, 1))
    goto out;
 
  index = arrayList->size++;
  rc = ArrayList_SetItem(arrayList, index, obj);
out:
 
  ArrayList_Unlock_Conditional(arrayList);
 
  return rc;
}
 
/*
 * Inserts an element into the ArrayList at the specified index.
 */
 
BOOL ArrayList_Insert(wArrayList* arrayList, size_t index, const void* obj)
{
  BOOL ret = TRUE;
 
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  if (index < arrayList->size)
  {
    if (!ArrayList_Shift(arrayList, index, 1))
    {
      ret = FALSE;
    }
    else
    {
      ArrayList_SetItem(arrayList, index, obj);
    }
  }
 
  ArrayList_Unlock_Conditional(arrayList);
 
  return ret;
}
 
BOOL ArrayList_Remove(wArrayList* arrayList, const void* obj)
{
  BOOL found = FALSE;
  BOOL ret = TRUE;
 
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  size_t index = 0;
  for (; index < arrayList->size; index++)
  {
    if (arrayList->object.fnObjectEquals(arrayList->array[index], obj))
    {
      found = TRUE;
      break;
    }
  }
 
  if (found)
  {
    if (arrayList->object.fnObjectFree)
      arrayList->object.fnObjectFree(arrayList->array[index]);
 
    ret = ArrayList_Shift(arrayList, index, -1);
  }
 
  ArrayList_Unlock_Conditional(arrayList);
 
  return ret;
}
 
BOOL ArrayList_RemoveAt(wArrayList* arrayList, size_t index)
{
  BOOL ret = TRUE;
 
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  if (index < arrayList->size)
  {
    if (arrayList->object.fnObjectFree)
      arrayList->object.fnObjectFree(arrayList->array[index]);
 
    ret = ArrayList_Shift(arrayList, index, -1);
  }
 
  ArrayList_Unlock_Conditional(arrayList);
 
  return ret;
}
 
SSIZE_T ArrayList_IndexOf(wArrayList* arrayList, const void* obj, SSIZE_T startIndex, SSIZE_T count)
{
  BOOL found = FALSE;
 
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  SSIZE_T sindex = startIndex;
  if (startIndex < 0)
    sindex = 0;
 
  SSIZE_T index = sindex;
  SSIZE_T cindex = count;
  if (count < 0)
  {
    if (arrayList->size > SSIZE_MAX)
      goto fail;
    cindex = (SSIZE_T)arrayList->size;
  }
 
  for (; index < sindex + cindex; index++)
  {
    if (arrayList->object.fnObjectEquals(arrayList->array[index], obj))
    {
      found = TRUE;
      break;
    }
  }
 
fail:
  if (!found)
    index = -1;
 
  ArrayList_Unlock_Conditional(arrayList);
 
  return index;
}
 
SSIZE_T ArrayList_LastIndexOf(wArrayList* arrayList, const void* obj, SSIZE_T startIndex,
                              SSIZE_T count)
{
  SSIZE_T sindex = 0;
  SSIZE_T cindex = 0;
  BOOL found = FALSE;
 
  WINPR_ASSERT(arrayList);
  ArrayList_Lock_Conditional(arrayList);
 
  sindex = startIndex;
  if (startIndex < 0)
    sindex = 0;
 
  cindex = count;
  if (count < 0)
  {
    WINPR_ASSERT(arrayList->size <= SSIZE_MAX);
    cindex = (SSIZE_T)arrayList->size;
  }
 
  SSIZE_T index = sindex + cindex;
  for (; index > sindex; index--)
  {
    if (arrayList->object.fnObjectEquals(arrayList->array[index - 1], obj))
    {
      found = TRUE;
      break;
    }
  }
 
  if (!found)
    index = -1;
 
  ArrayList_Unlock_Conditional(arrayList);
 
  return index;
}
 
static BOOL ArrayList_DefaultCompare(const void* objA, const void* objB)
{
  return objA == objB ? TRUE : FALSE;
}
 
wObject* ArrayList_Object(wArrayList* arrayList)
{
  WINPR_ASSERT(arrayList);
  return &arrayList->object;
}
 
BOOL ArrayList_ForEach(wArrayList* arrayList, ArrayList_ForEachFkt fkt, ...)
{
  BOOL rc = 0;
  va_list ap = { 0 };
  va_start(ap, fkt);
  rc = ArrayList_ForEachAP(arrayList, fkt, ap);
  va_end(ap);
 
  return rc;
}
 
BOOL ArrayList_ForEachAP(wArrayList* arrayList, ArrayList_ForEachFkt fkt, va_list ap)
{
  BOOL rc = FALSE;
  va_list cap;
 
  WINPR_ASSERT(arrayList);
  WINPR_ASSERT(fkt);
 
  ArrayList_Lock_Conditional(arrayList);
  size_t count = ArrayList_Count(arrayList);
  for (size_t index = 0; index < count; index++)
  {
    BOOL rs = 0;
    void* obj = ArrayList_GetItem(arrayList, index);
    va_copy(cap, ap);
    rs = fkt(obj, index, cap);
    va_end(cap);
    if (!rs)
      goto fail;
  }
  rc = TRUE;
fail:
  ArrayList_Unlock_Conditional(arrayList);
  return rc;
}
 
wArrayList* ArrayList_New(BOOL synchronized)
{
  wObject* obj = NULL;
  wArrayList* arrayList = NULL;
  arrayList = (wArrayList*)calloc(1, sizeof(wArrayList));
 
  if (!arrayList)
    return NULL;
 
  arrayList->synchronized = synchronized;
  arrayList->growthFactor = 2;
  obj = ArrayList_Object(arrayList);
  if (!obj)
    goto fail;
  obj->fnObjectEquals = ArrayList_DefaultCompare;
  if (!ArrayList_EnsureCapacity(arrayList, 32))
    goto fail;
 
  InitializeCriticalSectionAndSpinCount(&arrayList->lock, 4000);
  return arrayList;
fail:
  WINPR_PRAGMA_DIAG_PUSH
  WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
  ArrayList_Free(arrayList);
  WINPR_PRAGMA_DIAG_POP
  return NULL;
}
 
void ArrayList_Free(wArrayList* arrayList)
{
  if (!arrayList)
    return;
 
  ArrayList_Clear(arrayList);
  DeleteCriticalSection(&arrayList->lock);
  free((void*)arrayList->array);
  free(arrayList);
}