- // 定义一个发送指定长度的字符串的函数,采用包头形式,数据头是一块四字节大小的内存空间,用来存储数据包的尺寸大小
- int writen(int fd, const char* msg, int size)
- {
- char* buf = msg;//buf用来指向发送数据的起始位置
- int count = size;//size用来记录数据包内容的总大小,count用来记录剩余数据的大小,一开始没有发送,故为count
- //开始循环发送数据
- while (count>0)//当
- {
- int len = send(fd, buf, count, 0);//len用来记录成功发送出去的数据的大小
- if (len == -1)//数据发送失败
- {
- return -1;
- }
- else if (len == 0)//数据没有发送出去
- {
- continue;//终止当前循环,进入下一次循环
- }
- buf += len;
- count -= len;
- }
- return size;
- }
- //发送数据
- int setMsg(int cfd, const char* msg, int len)
- {
- if (cfd < 0 || msg == NULL || len <= 0)
- {
- return -1;
- }
- char* data = (char*)malloc(len + 4);//在堆上动态申请一块内存
- int biglen = htonl(len);//将字节长度由主机字节序转换为网络字节序
- //将biglen放到data申请的堆内存中
- memcpy(data,&biglen,4);
- //将发送的数据拷贝到申请的堆内存中
- memcpy(data+4, msg, len);
-
- int ret = write(cfd,data, len+4);
- if (ret == -1)
- {
- close(cfd);
- }
- return ret;
- }