我正在尝试创建一个将任意类型的缓冲区组合在一起的函数。想想基本的 RPC。所以缓冲区看起来像
{ char opcode, uint32_t param1_size, param1, ... , uint32_t paramN_size, paramN }
它似乎正在工作,但是从缓冲区中提取数据时,它似乎没有正确解释字符串。
int enqueue_into_buf(char *buf, size_t buf_pos, const uint32_t param_len, const void *param) {
/* If param_len is NULL, then change opcode */
if(param_len == 0) {
memcpy( buf, param, 1 );
if(buf_pos == 0)
buf_pos++;
return buf_pos;
}
memcpy( buf + buf_pos, ¶m_len, sizeof(param_len) );
buf_pos += sizeof(param_len);
memcpy( buf + buf_pos, param, param_len );
buf_pos += param_len;
return buf_pos;
}
int main(int argc, char *argv[])
{
char opcode;
uint32_t param_len, num_params, buf_size, buf_pos = 0, received_num;
char *buf, *temp;
char *input_string = "file01"; /* example string to use as parameter */
size_t input_size = 4, received_size; /* example variable to also use as a parameter */
opcode = '1'; /* opcode equals function 1 */
num_params = 2; /* number of parameters */
/* setting the size of the buffer that will be sent over the network */
buf_size = sizeof(opcode) + ( num_params * sizeof(uint32_t) ) + (strlen(input_string) * sizeof(char)) + sizeof(input_size);
buf = malloc( buf_size );
/* Notice the ampersand! */
buf_pos = enqueue_into_buf(buf, buf_pos, 0, &opcode);
buf_pos = enqueue_into_buf(buf, buf_pos, strlen(input_string), &input_string);
buf_pos = enqueue_into_buf(buf, buf_pos, sizeof(input_size), &input_size);
/* At this point, since we inserted everything into the buffer,
the buffer size and current buffer position should be equal */
if(buf_pos == buf_size)
printf("Calculated buffer size correctly and inserted everything correctly as well. Buffer size = %d\n", buf_size);
/** Extract from buffer **/
buf_pos = 0;
printf("Opcode: %c\n", buf[buf_pos]);
buf_pos++;
memcpy(&received_num, buf + buf_pos, sizeof(uint32_t));
printf("Size of parameter 1: %d\n", received_num);
buf_pos += sizeof(uint32_t);
temp = malloc(received_num + 1);
memcpy(temp, buf + buf_pos, received_num);
temp[received_num] = '\0';
printf("Parameter 1: %s\n", temp);
buf_pos += received_num;
memcpy(&received_num, buf + buf_pos, sizeof(uint32_t));
printf("Size of parameter 2: %d\n", received_num);
buf_pos += sizeof(uint32_t);
memcpy(&received_size, buf + buf_pos, sizeof(size_t));
printf("Parameter 2: %d\n", received_size);
buf_pos += sizeof(size_t);
return 0;
}
编辑:代码输出:
Calculated buffer size correctly and inserted everything correctly as well. Buffer size = 23
Opcode: 1
Size of parameter 1: 6
Parameter 1: @ @
Size of parameter 2: 8
Parameter 2: 4
我认为我没有将数据正确复制到缓冲区中,因为使用相同的参数(input_string =“file01”,input_size = 4),此代码有效......
/* OPCODE */
buf[buf_pos] = opcode;
buf_pos++;
/* PARAMETER 1 */
param_len = (strlen(input_string) * sizeof(char)); /* size of parameter 1 */
memcpy(buf + buf_pos, ¶m_len, sizeof(uint32_t));
buf_pos += sizeof(uint32_t);
//memcpy( buf + buf_pos, &input_string, (strlen(input_string) * sizeof(char)) );
strcat( buf + buf_pos, input_string );
buf_pos += strlen(input_string) * sizeof(char);
/* PARAMETER 2 */
param_len = sizeof(input_size);
memcpy(buf + buf_pos, ¶m_len, sizeof(param_len)); /* same as saying sizeof(uint32_t) */
buf_pos += sizeof(uint32_t);
memcpy(buf + buf_pos, &input_size, sizeof(input_size));
buf_pos += sizeof(input_size);
编辑:代码输出:
Calculated buffer size correctly and inserted everything correctly as well. Buffer size = 23
Opcode: 1
Size of parameter 1: 6
Parameter 1: file01
Size of parameter 2: 8
Parameter 2: 4
但显然我不想使用 strcat() 因为我不知道它将是哪种数据类型。我是否错误地使用了 memcpy?
编辑:放置程序的输出