2020/2/17自定义协议的学习

1. 写一个自定义协议，封装了发送和接收message的模块，src/msg.c内容如下：

#include "msg.h"
#include <unistd.h>
#include <string.h>
#include <memory.h>
#include <sys/types.h>

// calculate check code 计算校验码函数
static unsigned char msg_check(Msg *message)
{
   unsigned char s = 0;
   int i;
   for(i = 0; i < sizeof(message->head); i++)
   {
       s += message->head[i];
   }
   for(i = 0; i < sizeof(message->buff); i++)
   {
       s += message->buff[i];
   }
   return s;
}

//send a message: data in buff 把参数buff中的数据发送出去
int write_msg(int sockfd, char *buff, size_t len)
{
   Msg message;
   memset(&message, 0, sizeof(message));
   strcpy(message.head, "iotek2012");
   memcpy(message.buff, buff, len);
   message.checknum = msg_check(&message);
   if(write(sockfd, &message, sizeof(message)) != sizeof(message)){
       return -1;
   }

}
//receive message: data in buff 把接收到的sockfd中的数据message参数buff中的读取出来
int read_msg(int sockfd, char *buff, size_t len)
{
   Msg message;
   memset(&message, 0, sizeof(message));
   size_t size;
   if((size = read(sockfd, &message, sizeof(message))) < 0){
       return -1;
   }
   else if(size == 0){
       return 0;
   }

   //check the code 检查校验码是否正确
   unsigned char s = msg_check(&message);
   if((s == (unsigned char)message.checknum) && (!strcmp("iotek2012",message.head))){//strcmp is a compare function for string
       memcpy(buff, message.buff, len);
       return sizeof(message);
   }
   return -1;
}

其中， size_t 这个类型可理解为无符号整型unsigned int，它定义在<stddef.h>, <stdio.h>, <stdlib.h>, <string.h>, <time.h>和<wchar.h>这些标准C头文件中，增加了代码的可移植性。

2. 编译测试

在编译时使用的命令行中，-Iinclude 表示把当前目录下的include目录做为头文件的搜索目录之一

来源：CSDN

作者：Xinyue_Lu

链接：https://blog.csdn.net/Xinyue_Lu/article/details/104347825

标签

include

sizeof