summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat
-rw-r--r--netbytes.c332
-rw-r--r--netbytes.h60
2 files changed, 391 insertions, 1 deletions
diff --git a/netbytes.c b/netbytes.c
index 7250aac..dca7d22 100644
--- a/netbytes.c
+++ b/netbytes.c
@@ -25,6 +25,80 @@ int nb_u8arr_create( nb_u8arr *s, __NBT_U8ARR_LEN len, uint8_t *val )
return 0;
}
+int nb_u16_create( nb_u16 *s, uint16_t val )
+{
+ if ( s == NULL )
+ return -1;
+
+ s->type = NBT_U16;
+ s->val = val;
+
+ return 0;
+}
+
+
+int nb_u16arr_create( nb_u16arr *s, __NBT_U16ARR_LEN len, uint16_t *val )
+{
+ if ( s == NULL )
+ return -1;
+
+ s->type = NBT_U16ARRAY;
+ s->len = len;
+ s->val = val;
+
+ return 0;
+}
+
+
+int nb_u32_create( nb_u32 *s, uint32_t val )
+{
+ if ( s == NULL )
+ return -1;
+
+ s->type = NBT_U32;
+ s->val = val;
+
+ return 0;
+}
+
+
+int nb_u32arr_create( nb_u32arr *s, __NBT_U32ARR_LEN len, uint32_t *val )
+{
+ if ( s == NULL )
+ return -1;
+
+ s->type = NBT_U32ARRAY;
+ s->len = len;
+ s->val = val;
+
+ return 0;
+}
+
+
+int nb_u64_create( nb_u64 *s, uint64_t val )
+{
+ if ( s == NULL )
+ return -1;
+
+ s->type = NBT_U64;
+ s->val = val;
+
+ return 0;
+}
+
+
+int nb_u64arr_create( nb_u64arr *s, __NBT_U64ARR_LEN len, uint64_t *val )
+{
+ if ( s == NULL )
+ return -1;
+
+ s->type = NBT_U64ARRAY;
+ s->len = len;
+ s->val = val;
+
+ return 0;
+}
+
int nb_init( netbyte_store *store )
{
@@ -71,6 +145,96 @@ int nb_add_u8arr( netbyte_store *store, nb_u8arr *u8arr )
}
+int nb_add_u16( netbyte_store *store, nb_u16 *u16 )
+{
+ if ( store->count >= __NBT_MAX_TYPES - 1 )
+ return -1;
+
+ if ( u16->type != NBT_U16 )
+ return -2;
+
+ store->types[store->count].type = NBT_U16;
+ store->types[store->count].nb_val = (uint8_t *)u16;
+ store->count += 1;
+ return 0;
+}
+
+
+int nb_add_u16arr( netbyte_store *store, nb_u16arr *u16arr )
+{
+ if ( store->count >= __NBT_MAX_TYPES - 1 )
+ return -1;
+
+ if ( u16arr->type != NBT_U16ARRAY )
+ return -2;
+
+ store->types[store->count].type = NBT_U16ARRAY;
+ store->types[store->count].nb_val = (uint8_t *)u16arr;
+ store->count += 1;
+ return 0;
+}
+
+
+int nb_add_u32( netbyte_store *store, nb_u32 *u32 )
+{
+ if ( store->count >= __NBT_MAX_TYPES - 1 )
+ return -1;
+
+ if ( u32->type != NBT_U32 )
+ return -2;
+
+ store->types[store->count].type = NBT_U32;
+ store->types[store->count].nb_val = (uint8_t *)u32;
+ store->count += 1;
+ return 0;
+}
+
+
+int nb_add_u32arr( netbyte_store *store, nb_u32arr *u32arr )
+{
+ if ( store->count >= __NBT_MAX_TYPES - 1 )
+ return -1;
+
+ if ( u32arr->type != NBT_U32ARRAY )
+ return -2;
+
+ store->types[store->count].type = NBT_U32ARRAY;
+ store->types[store->count].nb_val = (uint8_t *)u32arr;
+ store->count += 1;
+ return 0;
+}
+
+
+int nb_add_u64( netbyte_store *store, nb_u64 *u64 )
+{
+ if ( store->count >= __NBT_MAX_TYPES - 1 )
+ return -1;
+
+ if ( u64->type != NBT_U64 )
+ return -2;
+
+ store->types[store->count].type = NBT_U64;
+ store->types[store->count].nb_val = (uint8_t *)u64;
+ store->count += 1;
+ return 0;
+}
+
+
+int nb_add_u64arr( netbyte_store *store, nb_u64arr *u64arr )
+{
+ if ( store->count >= __NBT_MAX_TYPES - 1 )
+ return -1;
+
+ if ( u64arr->type != NBT_U64ARRAY )
+ return -2;
+
+ store->types[store->count].type = NBT_U64ARRAY;
+ store->types[store->count].nb_val = (uint8_t *)u64arr;
+ store->count += 1;
+ return 0;
+}
+
+
uint8_t *nb_create( netbyte_store *store )
{
uint8_t *ret = NULL;
@@ -95,7 +259,7 @@ uint8_t *nb_create( netbyte_store *store )
switch( iter.type )
{
case NBT_U8:
- store->size += sizeof(__NBT_TYPED) + 1;
+ store->size += sizeof(__NBT_TYPED) + sizeof(uint8_t);
break;
case NBT_U8ARRAY:
{
@@ -105,6 +269,39 @@ uint8_t *nb_create( netbyte_store *store )
u8arr->len;
}
break;
+ case NBT_U16:
+ store->size += sizeof(__NBT_TYPED) + sizeof(uint16_t);
+ break;
+ case NBT_U16ARRAY:
+ {
+ nb_u16arr *u16arr = (nb_u16arr *)iter.nb_val;
+ store->size += sizeof(__NBT_TYPED) +
+ sizeof(__NBT_U16ARR_LEN) +
+ u16arr->len*sizeof(uint16_t);
+ }
+ break;
+ case NBT_U32:
+ store->size += sizeof(__NBT_TYPED) + sizeof(uint32_t);
+ break;
+ case NBT_U32ARRAY:
+ {
+ nb_u32arr *u32arr = (nb_u32arr *)iter.nb_val;
+ store->size += sizeof(__NBT_TYPED) +
+ sizeof(__NBT_U32ARR_LEN) +
+ u32arr->len*sizeof(uint32_t);
+ }
+ break;
+ case NBT_U64:
+ store->size += sizeof(__NBT_TYPED) + sizeof(uint64_t);
+ break;
+ case NBT_U64ARRAY:
+ {
+ nb_u64arr *u64arr = (nb_u64arr *)iter.nb_val;
+ store->size += sizeof(__NBT_TYPED) +
+ sizeof(__NBT_U64ARR_LEN) +
+ u64arr->len*sizeof(uint64_t);
+ }
+ break;
default:
printf("Unknow type be carefull1\n");
}
@@ -143,6 +340,57 @@ uint8_t *nb_create( netbyte_store *store )
c += u8arr->len;
break;
}
+ case NBT_U16:
+ memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
+ c += sizeof(__NBT_TYPED);
+ memcpy( c, &((nb_u16 *)(iter.nb_val))->val, sizeof(uint16_t) );
+ c += sizeof(uint16_t);
+ break;
+ case NBT_U16ARRAY:
+ {
+ nb_u16arr *u16arr = (nb_u16arr *)iter.nb_val;
+ memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
+ c += sizeof(__NBT_TYPED);
+ memcpy( c, &u16arr->len, sizeof(__NBT_U16ARR_LEN));
+ c += sizeof(__NBT_U16ARR_LEN);
+ memcpy( c, u16arr->val, u16arr->len );
+ c += u16arr->len*sizeof(uint16_t);
+ break;
+ }
+ case NBT_U32:
+ memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
+ c += sizeof(__NBT_TYPED);
+ memcpy( c, &((nb_u32 *)(iter.nb_val))->val, sizeof(uint32_t) );
+ c += sizeof(uint32_t);
+ break;
+ case NBT_U32ARRAY:
+ {
+ nb_u32arr *u32arr = (nb_u32arr *)iter.nb_val;
+ memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
+ c += sizeof(__NBT_TYPED);
+ memcpy( c, &u32arr->len, sizeof(__NBT_U32ARR_LEN));
+ c += sizeof(__NBT_U32ARR_LEN);
+ memcpy( c, u32arr->val, u32arr->len );
+ c += u32arr->len*sizeof(uint32_t);
+ break;
+ }
+ case NBT_U64:
+ memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
+ c += sizeof(__NBT_TYPED);
+ memcpy( c, &((nb_u64 *)(iter.nb_val))->val, sizeof(uint64_t) );
+ c += sizeof(uint64_t);
+ break;
+ case NBT_U64ARRAY:
+ {
+ nb_u64arr *u64arr = (nb_u64arr *)iter.nb_val;
+ memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
+ c += sizeof(__NBT_TYPED);
+ memcpy( c, &u64arr->len, sizeof(__NBT_U64ARR_LEN));
+ c += sizeof(__NBT_U64ARR_LEN);
+ memcpy( c, u64arr->val, u64arr->len );
+ c += u64arr->len*sizeof(uint64_t);
+ break;
+ }
default:
printf("Unknow type be carefull2\n");
}
@@ -223,6 +471,88 @@ int nb_load( netbyte_store *store, uint8_t *data )
break;
}
+ case NBT_U16:
+ {
+ nb_u16 *u16 = malloc( sizeof(nb_u16) );
+
+ memcpy( u16, c, sizeof(nb_u16) );
+ c += sizeof(nb_u16);
+
+ nb_add_u16( store, u16 );
+
+ break;
+ }
+ case NBT_U16ARRAY:
+ {
+ nb_u16arr *u16arr = malloc( sizeof(nb_u16arr) );
+
+ memcpy( u16arr, c, sizeof(__NBT_TYPED));
+ c += sizeof(__NBT_TYPED);
+
+ memcpy( &u16arr->len, c, sizeof(__NBT_U16ARR_LEN) );
+ c += sizeof(__NBT_U16ARR_LEN);
+
+ u16arr->val = (uint16_t *)c;
+ //printf("->2 %s\n",c );
+ c += u16arr->len*sizeof(uint16_t);
+
+ nb_add_u16arr( store, u16arr );
+ break;
+ }
+ case NBT_U32:
+ {
+ nb_u32 *u32 = malloc( sizeof(nb_u32) );
+
+ memcpy( u32, c, sizeof(nb_u32) );
+ c += sizeof(nb_u32);
+
+ nb_add_u32( store, u32 );
+ break;
+ }
+ case NBT_U32ARRAY:
+ {
+ nb_u32arr *u32arr = malloc( sizeof(nb_u32arr) );
+
+ memcpy( u32arr, c, sizeof(__NBT_TYPED));
+ c += sizeof(__NBT_TYPED);
+
+ memcpy( &u32arr->len, c, sizeof(__NBT_U32ARR_LEN) );
+ c += sizeof(__NBT_U32ARR_LEN);
+
+ u32arr->val = (uint32_t *)c;
+ //printf("->2 %s\n",c );
+ c += u32arr->len*sizeof(uint32_t);
+
+ nb_add_u32arr( store, u32arr );
+ break;
+ }
+ case NBT_U64:
+ {
+ nb_u64 *u64 = malloc( sizeof(nb_u64) );
+
+ memcpy( u64, c, sizeof(nb_u64) );
+ c += sizeof(nb_u64);
+
+ nb_add_u64( store, u64 );
+ break;
+ }
+ case NBT_U64ARRAY:
+ {
+ nb_u64arr *u64arr = malloc( sizeof(nb_u64arr) );
+
+ memcpy( u64arr, c, sizeof(__NBT_TYPED));
+ c += sizeof(__NBT_TYPED);
+
+ memcpy( &u64arr->len, c, sizeof(__NBT_U64ARR_LEN) );
+ c += sizeof(__NBT_U64ARR_LEN);
+
+ u64arr->val = (uint64_t *)c;
+ //printf("->2 %s\n",c );
+ c += u64arr->len;
+
+ nb_add_u64arr( store, u64arr );
+ break;
+ }
default:
printf("Unknown type");
}
diff --git a/netbytes.h b/netbytes.h
index cab61f6..e4937fb 100644
--- a/netbytes.h
+++ b/netbytes.h
@@ -38,6 +38,9 @@
#define __NBT_SIZE uint32_t
#define __NBT_TYPED uint8_t
#define __NBT_U8ARR_LEN uint16_t
+#define __NBT_U16ARR_LEN uint16_t
+#define __NBT_U32ARR_LEN uint16_t
+#define __NBT_U64ARR_LEN uint16_t
#define __NBT_MINIMAL_SIZE (sizeof(__NBT_SIZE))
#define __NBT_MAX_TYPES (256)
@@ -68,6 +71,51 @@ typedef struct nb_u8arr
uint8_t *val;
} nb_u8arr;
+
+/*-----------------*/
+typedef struct nb_u16
+{
+ __NBT_TYPED type;
+ uint16_t val;
+} nb_u16;
+
+typedef struct nb_u16arr
+{
+ __NBT_TYPED type;
+ __NBT_U16ARR_LEN len;
+ uint16_t *val;
+} nb_u16arr;
+
+
+/*-----------------*/
+typedef struct nb_u32
+{
+ __NBT_TYPED type;
+ uint32_t val;
+} nb_u32;
+
+typedef struct nb_u32arr
+{
+ __NBT_TYPED type;
+ __NBT_U32ARR_LEN len;
+ uint32_t *val;
+} nb_u32arr;
+
+
+/*------------------*/
+typedef struct nb_u64
+{
+ __NBT_TYPED type;
+ uint64_t val;
+} nb_u64;
+
+typedef struct nb_u64arr
+{
+ __NBT_TYPED type;
+ __NBT_U64ARR_LEN len;
+ uint64_t *val;
+} nb_u64arr;
+
//loading/parsing netbyte structure
typedef struct netbyte_load
{
@@ -77,11 +125,23 @@ typedef struct netbyte_load
int nb_u8_create( nb_u8 *s, uint8_t val );
int nb_u8arr_create( nb_u8arr *s, __NBT_U8ARR_LEN len, uint8_t *val );
+int nb_u16_create( nb_u16 *s, uint16_t val );
+int nb_u16arr_create( nb_u16arr *s, __NBT_U16ARR_LEN len, uint16_t *val );
+int nb_u32_create( nb_u32 *s, uint32_t val );
+int nb_u32arr_create( nb_u32arr *s, __NBT_U32ARR_LEN len, uint32_t *val );
+int nb_u64_create( nb_u64 *s, uint64_t val );
+int nb_u64arr_create( nb_u64arr *s, __NBT_U64ARR_LEN len, uint64_t *val );
int nb_init( netbyte_store *store );
int nb_add_u8( netbyte_store *store, nb_u8 *u8 );
int nb_add_u8arr( netbyte_store *store, nb_u8arr *u8arr );
+int nb_add_u16( netbyte_store *store, nb_u16 *u16 );
+int nb_add_u16arr( netbyte_store *store, nb_u16arr *u16arr );
+int nb_add_u32( netbyte_store *store, nb_u32 *u32 );
+int nb_add_u32arr( netbyte_store *store, nb_u32arr *u32arr );
+int nb_add_u64( netbyte_store *store, nb_u64 *u64 );
+int nb_add_u64arr( netbyte_store *store, nb_u64arr *u64arr );
uint8_t *nb_create( netbyte_store *store );
int nb_load( netbyte_store *store, uint8_t *data );
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-14 17:10:08 +0000