套接字通信--(UDP)

proto.h

#ifndef __PROTO_H__
#define __PROTO_H__

#include <stdint.h>//typede各种数据类型

//server-->bind	client-->sendto
#define SERVERIP	"10.11.3.157"
#define SERVERPORT			3366

//数据类型
#define MSGSIZE		128

struct msg_st {
	int8_t id;
	char msg[MSGSIZE];
}__attribute__((packed));//单字节对齐，__attribute__((packed))属性就是单字节对齐。

#endif

client.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>

#include "proto.h"

int main(int argc, char **argv)
{
	int udp_socket;
	//myaddr自己的地址	remote_addr接收的地址
	struct sockaddr_in server_addr;
	//定义一个(自己定义的结构体msg_st的)变量
	struct msg_st rcvbuf, sndbuf;
	int cnt;

	if(argc < 3)
		return 1;

	//创建报式套接字
	udp_socket = socket(AF_INET, SOCK_DGRAM, 0);
	if(-1 == udp_socket) {
		perror("socket()");
		exit(1);
	}
	//bind可省略

	//发送的数据
	sndbuf.id = atoi(argv[1]);
	strncpy(sndbuf.msg, argv[2], MSGSIZE);
	//绑定地址
	//ipv4协议
	server_addr.sin_family = AF_INET;
	//网络字节序,htons()将主机字节序(本地字节序)转换为网络字节序
	//网络字节序都是大端字节序，主机字节序(本地字节序)有大端字节序和小端字节序
	server_addr.sin_port = htons(SERVERPORT);
	//字符串的点分十进制ip转换为ip结构体
	//通过回填参数的形式，给myaddr.sin_addr赋值
	inet_aton(SERVERIP, &server_addr.sin_addr);
	sendto(udp_socket, &sndbuf, sizeof(sndbuf.id) + strlen(sndbuf.msg), 0, (struct sockaddr *)&server_addr, sizeof(server_addr));
	//接收数据，因为是客户端，知道通信的服务端的地址，直接为NULL也可以，recvfrom的最后两个参数是用来保存发送过来信息的对方的地址
	cnt = recvfrom(udp_socket, &rcvbuf, sizeof(rcvbuf), 0, NULL, NULL);
	if(-1 == cnt) {
		perror("revefrom()");
		goto ERROR;
	}
	//调试语句
	printf("msg id:%d, msg:%s\n", rcvbuf.id, rcvbuf.msg);

	close(udp_socket);
	exit(0);
ERROR:
	close(udp_socket);
	exit(1);
}

server.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>

#include "proto.h"

int main(void)
{
	int udp_socket;
	//myaddr自己的地址	remote_addr接收的地址
	struct sockaddr_in myaddr, remote_addr;
	//remote_addr_len接收的地址的长度
	socklen_t remote_addr_len;
	//定义一个(自己定义的结构体msg_st的)变量
	struct msg_st rcvbuf;
	int cnt;

	//创建报式套接字
	udp_socket = socket(AF_INET, SOCK_DGRAM, 0);
	if(-1 == udp_socket) {
		perror("socket()");
		exit(1);
	}

	//绑定地址
	//ipv4协议
	myaddr.sin_family = AF_INET;
	//网络字节序,htons()将主机字节序(本地字节序)转换为网络字节序
	//网络字节序都是大端字节序，主机字节序(本地字节序)有大端字节序和小端字节序
	myaddr.sin_port = htons(SERVERPORT);
	//字符串的点分十进制ip转换为ip结构体
	//通过回填参数的形式，给myaddr.sin_addr赋值
	inet_aton(SERVERIP, &myaddr.sin_addr);
	if(-1 == bind(udp_socket, (struct sockaddr *)&myaddr, sizeof(myaddr))) {
		perror("bind()");
		goto BIND_ERROR;
	}

	//接收client的数据包之前一定要将地址长度赋值
	while(1) {
		//等待接收客户端的请求
		cnt = recvfrom(udp_socket, &rcvbuf, sizeof(rcvbuf), 0, (struct sockaddr *)&remote_addr, &remote_addr_len);
		if(-1 == cnt) {
			perror("recvfrom()");
			goto BIND_ERROR;
		}
		//调试语句
		printf("*****************************\n");
		printf("client ip:%s, port:%d\n", inet_ntoa(remote_addr.sin_addr), ntohs(remote_addr.sin_port));
		printf("msg id:%d, msg:%s\n", rcvbuf.id, rcvbuf.msg);
		printf("****************************\n");
		rcvbuf.id = 0;
		strcpy(rcvbuf.msg, "ok");
		sendto(udp_socket, &rcvbuf, sizeof(rcvbuf.id) + strlen(rcvbuf.msg) + 1, 0, (struct sockaddr *)&remote_addr, remote_addr_len);
	}

	close(udp_socket);

	exit(0);
BIND_ERROR:
	close(udp_socket);
	exit(1);
}