// // fs_stream.h // fsnet // // Created by Vincent on 14-5-22. // Copyright (c) 2014年 Vincent. All rights reserved. // #ifndef fsnet_fs_stream_h #define fsnet_fs_stream_h #include "fs_define.h" #define OUTPUT_STRAEM_DEFAULT_MAX 128 struct fs_stream; struct fs_input_stream; struct fs_output_stream; struct fs_output_stream* fs_create_output_stream(size_t len); #define fs_create_output_stream_ext fs_create_output_stream(OUTPUT_STRAEM_DEFAULT_MAX) void fs_stream_free_output( struct fs_output_stream* ); void fs_stream_write_byte( struct fs_output_stream*, BYTE); void fs_stream_write_uint32( struct fs_output_stream*, uint32_t); void fs_stream_write_uint16( struct fs_output_stream*, uint16_t); void fs_stream_write_int32( struct fs_output_stream*, int32_t); void fs_stream_write_int16( struct fs_output_stream*, int16_t); void fs_stream_write_float( struct fs_output_stream*, float ); void fs_stream_write_double( struct fs_output_stream*, double ); void fs_stream_write_data( struct fs_output_stream*, BYTE*, size_t len ); const BYTE* fs_stream_get_dataptr( struct fs_output_stream* ); size_t fs_stream_get_len( struct fs_output_stream* ); size_t fs_stream_output_sub( struct fs_output_stream*, size_t start, size_t len ); struct fs_input_stream* fs_create_input_stream(const BYTE* data, size_t len); struct fs_input_stream* fs_create_input_stream_with_copy(const BYTE* data, size_t len); void fs_input_stream_set_data( struct fs_input_stream*, const BYTE* data, size_t len); void fs_input_stream_set_data_copy( struct fs_input_stream*, const BYTE* data, size_t len); size_t fs_input_stream_get_len( struct fs_input_stream* stream); void fs_input_stream_set_order( struct fs_input_stream*, BYTE order); void fs_stream_free_input( struct fs_input_stream* ); BYTE fs_stream_read_byte( struct fs_input_stream* ); uint32_t fs_stream_read_uint32( struct fs_input_stream* ); uint16_t fs_stream_read_uint16( struct fs_input_stream* ); int32_t fs_stream_read_int32( struct fs_input_stream* ); int16_t fs_stream_read_int16( struct fs_input_stream* ); float fs_stream_read_float( struct fs_input_stream* ); double fs_stream_read_double( struct fs_input_stream* ); size_t fs_stream_read_string( struct fs_input_stream* , char**); void fs_output_stream_set_script_id( struct fs_output_stream* , fs_script_id _id ); fs_script_id fs_output_stream_get_script_id( struct fs_output_stream* ); void fs_input_stream_set_script_id( struct fs_input_stream* , fs_script_id _id ); fs_script_id fs_input_stream_get_script_id( struct fs_input_stream* ); #endif
// // fs_stream.c // fsnet // // Created by Vincent on 14-5-22. // Copyright (c) 2014年 Vincent. All rights reserved. // #include <stdio.h> #include <string.h> #include "fs_define.h" #include "fs_stream.h" #include "fs_malloc.h" #define BASE_STREAM BYTE byte_order; BYTE* data; struct fs_input_stream{ BASE_STREAM size_t pos; size_t len; fs_bool copy; fs_script_id script_id; }; struct fs_output_stream{ BASE_STREAM size_t pos; size_t buff_len; fs_script_id script_id; }; struct fs_output_stream* fs_create_output_stream(size_t len){ struct fs_output_stream* ret = (struct fs_output_stream*)fs_malloc(sizeof(*ret)); fs_zero(ret, sizeof(*ret)); ret->data = (BYTE*)fs_malloc(len); ret->pos = 0; ret->buff_len = len; return ret; } void fs_stream_free_output( struct fs_output_stream* stream){ fs_free(stream->data); fs_free(stream); } void fs_stream_write_byte( struct fs_output_stream* stream, BYTE v){ fs_stream_write_data(stream, (BYTE*)&v, sizeof(BYTE)); } void fs_stream_write_uint32( struct fs_output_stream* stream, uint32_t v){ fs_stream_write_data(stream, (BYTE*)&v, sizeof(uint32_t)); } void fs_stream_write_uint16( struct fs_output_stream* stream, uint16_t v){ fs_stream_write_data(stream, (BYTE*)&v, sizeof(uint16_t)); } void fs_stream_write_int32( struct fs_output_stream* stream, int32_t v){ fs_stream_write_data(stream, (BYTE*)&v, sizeof(int32_t)); } void fs_stream_write_int16( struct fs_output_stream* stream, int16_t v){ fs_stream_write_data(stream, (BYTE*)&v, sizeof(int16_t)); } void fs_stream_write_float( struct fs_output_stream* stream, float v){ fs_stream_write_data(stream, (BYTE*)&v, sizeof(float)); } void fs_stream_write_double( struct fs_output_stream* stream, double v){ fs_stream_write_data(stream, (BYTE*)&v, sizeof(double)); } void fs_stream_write_data( struct fs_output_stream* stream, BYTE* data, size_t len ){ while(stream->pos + len > stream->buff_len){ void* data = fs_realloc(stream->data, stream->buff_len << 1); stream->data = data; stream->buff_len = stream->buff_len << 1; } memcpy(stream->data + stream->pos, data, len); stream->pos += len; } const BYTE* fs_stream_get_dataptr( struct fs_output_stream* stream){ return stream->data; } size_t fs_stream_get_len( struct fs_output_stream* stream){ return stream->pos; } size_t fs_stream_output_sub( struct fs_output_stream* stream, size_t start, size_t len ){ memcpy(stream->data, stream->data + start, len); stream->pos = len; return len; } struct fs_input_stream* fs_create_input_stream(const BYTE* data, size_t len){ struct fs_input_stream* ret = (struct fs_input_stream*)fs_malloc(sizeof(*ret)); fs_zero(ret, sizeof(*ret)); fs_input_stream_set_data(ret, data, len); return ret; } struct fs_input_stream* fs_create_input_stream_with_copy(const BYTE* data, size_t len){ struct fs_input_stream* ret = (struct fs_input_stream*)fs_malloc(sizeof(*ret)); fs_zero(ret, sizeof(*ret)); fs_input_stream_set_data_copy(ret, data, len); return ret; } void fs_input_stream_set_data( struct fs_input_stream* stream, const BYTE* data, size_t len){ if(stream->data){ if(stream->copy){ fs_free(stream->data); } } stream->data = (BYTE*)data; stream->pos = 0; stream->len = len; stream->copy = fs_false; } void fs_input_stream_set_data_copy( struct fs_input_stream* stream, const BYTE* data, size_t len){ if(stream->data){ if(stream->copy){ fs_free(stream->data); } } stream->data = (BYTE*)fs_malloc(len); stream->pos = 0; stream->len = len; stream->copy = fs_true; memcpy(stream->data, data, len); } size_t fs_input_stream_get_len( struct fs_input_stream* stream){ return stream->len; } void fs_input_stream_set_order( struct fs_input_stream* stream, BYTE order){ stream->byte_order = order; } void fs_stream_free_input( struct fs_input_stream* stream){ if(stream->copy){ fs_free(stream->data); } fs_free(stream); } #define _stream_read(TYPE)TYPE v = 0; memcpy(&v, stream->data + stream->pos, sizeof(v)); stream->pos += sizeof(v); return v; BYTE fs_stream_read_byte( struct fs_input_stream* stream){ _stream_read(BYTE); } uint32_t fs_stream_read_uint32( struct fs_input_stream* stream){ _stream_read(uint32_t); } uint16_t fs_stream_read_uint16( struct fs_input_stream* stream){ _stream_read(uint16_t); } int32_t fs_stream_read_int32( struct fs_input_stream* stream){ _stream_read(int32_t); } int16_t fs_stream_read_int16( struct fs_input_stream* stream){ _stream_read(int16_t); } float fs_stream_read_float( struct fs_input_stream* stream){ _stream_read(float); } double fs_stream_read_double( struct fs_input_stream* stream){ _stream_read(double); } size_t fs_stream_read_string( struct fs_input_stream* stream, char** out){ return 0; } void fs_output_stream_set_script_id( struct fs_output_stream* stream, fs_script_id _id ){ stream->script_id = _id; } fs_script_id fs_output_stream_get_script_id( struct fs_output_stream* stream){ return stream->script_id; } void fs_input_stream_set_script_id( struct fs_input_stream* stream, fs_script_id _id ){ stream->script_id = _id; } fs_script_id fs_input_stream_get_script_id( struct fs_input_stream* stream){ return stream->script_id; }
c 实现IO流
时间: 2024-10-18 15:12:04