1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
/* libjodycode: fileinfo batch handlers
*
* Copyright (C) 2014-2026 by Jody Bruchon <jody@jodybruchon.com>
* Released under The MIT License
*/
#include <stdlib.h>
#include <string.h>
#include "libjodycode.h"
#include "likely_unlikely.h"
/* allocs: 0 = just the batch, 1 = with dirents, 2 = with stats, 3 = with both */
struct jc_fileinfo_batch *jc_fileinfo_batch_alloc(const int filecnt, const int stat, const int namlen)
{
struct jc_fileinfo_batch *batch;
struct JC_STAT *stats;
struct JC_DIRENT *dirents;
int i;
if (unlikely(filecnt <= 0)) return NULL;
batch = calloc(1, (size_t)((int)sizeof(struct jc_fileinfo_batch) + ((int)sizeof(struct jc_fileinfo) * filecnt)));
if (batch == NULL) return NULL;
batch->count = filecnt;
if (stat != 0) {
stats = calloc(1, (size_t)(((int)sizeof(struct JC_STAT) * filecnt)));
if (stats == NULL) goto error_cleanup;
for (i = 0; i < filecnt; i++)
batch->files[i].stat = (struct JC_STAT *)((uintptr_t)stats + (uintptr_t)(((int)sizeof(struct JC_STAT) * i)));
}
if (namlen != 0) {
dirents = calloc(1, (size_t)(((int)sizeof(struct JC_DIRENT) + namlen) * filecnt));
if (dirents == NULL) goto error_cleanup;
for (i = 0; i < filecnt; i++)
batch->files[i].dirent = (struct JC_DIRENT *)((uintptr_t)dirents + (uintptr_t)(((int)sizeof(struct JC_DIRENT) + namlen) * i));
}
return batch;
error_cleanup:
jc_fileinfo_batch_free(batch);
return NULL;
}
void jc_fileinfo_batch_free(struct jc_fileinfo_batch * const restrict batch)
{
if (unlikely(batch == NULL)) return;
if (batch->files[0].stat != NULL) free(batch->files[0].stat);
if (batch->files[0].dirent != NULL) free(batch->files[0].dirent);
free(batch);
return;
}
|
