agsutils/ByteArray.h at master · rofl0r/agsutils · GitHub

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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
/* (C) 2011 - 2025 rofl0r. */

#ifndef BYTEARRAY_H
#define BYTEARRAY_H

#ifdef __cplusplus
extern "C" {
#endif

#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <stdio.h>
#include "MemGrow.h"
#ifdef _WIN32
#define WIN32_LEAN_AND_MEAN
#include <windows.h> /* for HANDLE */
#define FILEDESC HANDLE
#else
#define FILEDESC int
#endif

enum ByteArray_Endianess {
	BAE_BIG,
	BAE_LITTLE,
};

enum ByteArray_Type {
	BAT_MEMSTREAM,
	BAT_FILESTREAM,
};

enum ByteArray_Flags {
	BAF_CANGROW = 1,
	BAF_NONFATAL_READ_OOB = 2,
};

typedef long long ba_off_t;

struct ByteArray {
	int type;
	int flags;
	enum ByteArray_Endianess endian;
	enum ByteArray_Endianess sys_endian;
	ba_off_t pos;
	ba_off_t size;
	MG source_mem;
	FILEDESC source_fd;
	const char *filename;
	ssize_t (*readMultiByte)(struct ByteArray*, char*, size_t);
	unsigned long long (*readUnsignedLongLong)(struct ByteArray*);
	unsigned int (*readUnsignedInt)(struct ByteArray*);
	signed int (*readInt)(struct ByteArray*);
	unsigned short (*readUnsignedShort)(struct ByteArray*);
	signed short (*readShort)(struct ByteArray*);
	unsigned char (*readUnsignedByte)(struct ByteArray*);
	signed char (*readByte)(struct ByteArray*);
	ba_off_t (*readBytes) (struct ByteArray* self, struct ByteArray *dest, ba_off_t start, ba_off_t len);
	ba_off_t (*bytesAvailable)(struct ByteArray*);
	ba_off_t (*writeByte) (struct ByteArray* self, signed char what);
	ba_off_t (*writeUnsignedByte) (struct ByteArray* self, unsigned char what);
	ba_off_t (*writeShort) (struct ByteArray* self, signed short what);
	ba_off_t (*writeUnsignedShort) (struct ByteArray* self, unsigned short what);
	ba_off_t (*writeInt) (struct ByteArray* self, signed int what);
	ba_off_t (*writeUnsignedInt) (struct ByteArray* self, unsigned int what);
	ba_off_t (*writeMem) (struct ByteArray* self, unsigned char* what, size_t len);
	ba_off_t (*writeUTFBytes) (struct ByteArray* self, char* what);
	ba_off_t (*writeBytes) (struct ByteArray* self, struct ByteArray* what);
	ba_off_t (*writeFloat) (struct ByteArray* self, float what);
	int (*set_position_rel) (struct ByteArray* self, int rel);
	int (*set_position) (struct ByteArray* self, ba_off_t pos);
	ba_off_t (*get_position) (struct ByteArray* self);
};

void ByteArray_defaults(struct ByteArray* self);
void ByteArray_ctor(struct ByteArray* self);
struct ByteArray* ByteArray_new(void);

void ByteArray_set_endian(struct ByteArray* self, enum ByteArray_Endianess endian);
void ByteArray_set_flags(struct ByteArray *self, int flags);
enum ByteArray_Endianess ByteArray_get_endian(struct ByteArray* self);

int ByteArray_open_file(struct ByteArray* self, const char* filename);
void ByteArray_close_file(struct ByteArray *self);
int ByteArray_open_mem(struct ByteArray* self, char* data, size_t size);
void* ByteArray_get_mem(struct ByteArray* self, size_t offset, size_t byteswanted);
void ByteArray_clear(struct ByteArray* self);
void ByteArray_close(struct ByteArray* self);

void ByteArray_set_length(struct ByteArray* self, ba_off_t len);
ba_off_t ByteArray_get_length(struct ByteArray* self);

ba_off_t ByteArray_get_position(struct ByteArray* self);
int ByteArray_set_position(struct ByteArray* self, ba_off_t pos);
int ByteArray_set_position_rel(struct ByteArray* self, int rel);
ba_off_t ByteArray_bytesAvailable(struct ByteArray* self);
int ByteArray_is_eof(struct ByteArray* self);
ssize_t ByteArray_readMultiByte(struct ByteArray* self, char* buffer, size_t len);
unsigned long long ByteArray_readUnsignedLongLong(struct ByteArray* self);
unsigned int ByteArray_readUnsignedInt(struct ByteArray* self);
int ByteArray_readInt(struct ByteArray* self);
unsigned short ByteArray_readUnsignedShort(struct ByteArray* self);
short ByteArray_readShort(struct ByteArray* self);
unsigned char ByteArray_readUnsignedByte(struct ByteArray* self);
signed char ByteArray_readByte(struct ByteArray* self);
ba_off_t ByteArray_readBytes(struct ByteArray* self, struct ByteArray *dest, ba_off_t start, ba_off_t len);

ba_off_t ByteArray_writeByte(struct ByteArray* self, signed char what);
ba_off_t ByteArray_writeUnsignedByte(struct ByteArray* self, unsigned char what);
ba_off_t ByteArray_writeShort(struct ByteArray* self, signed short what);
ba_off_t ByteArray_writeUnsignedShort(struct ByteArray* self, unsigned short what);
ba_off_t ByteArray_writeInt(struct ByteArray* self, signed int what);
ba_off_t ByteArray_writeUnsignedInt(struct ByteArray* self, unsigned int what);
ba_off_t ByteArray_writeFloat(struct ByteArray* self, float what);
ba_off_t ByteArray_writeMem(struct ByteArray* self, unsigned char* what, size_t len);
ba_off_t ByteArray_writeUTFBytes(struct ByteArray* self, char* what);
ba_off_t ByteArray_writeBytes(struct ByteArray* self, struct ByteArray* what);

unsigned char ByteArray_getUnsignedByte(struct ByteArray* self, ba_off_t index);
void ByteArray_setUnsignedByte(struct ByteArray* self, ba_off_t index, unsigned char what);

int ByteArray_search(struct ByteArray *self, unsigned char* bytes, size_t len);

void ByteArray_dump_to_stream(struct ByteArray* self, FILE *out);
void ByteArray_dump_to_file(struct ByteArray* self, char* filename);

#ifdef __cplusplus
}
#endif

#pragma RcB2 DEP "ByteArray.c"

#endif