网络编程,udp上传下载udp聊天室

udp下载

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <pthread.h>
#include <semaphore.h>
#include <signal.h>
#include <sys/wait.h>
#include <sys/shm.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <sys/msg.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define ERR_MSG(msg) do{\
		fprintf(stderr,"line;%d;%s %s",__LINE__,__FILE__,__func__);\
		perror(msg);\
}while(0)
#define IP "192.168.2.84"
#define PORT 69
int xia_zai(int cfd,struct sockaddr_in sin)
{
	char name[20]="";
	char buf[516]={0};
	printf("请输入要下载的文件名>");
	scanf("%s",name);
	while(getchar()!=10);
	//printf("%s",name);
//===============================
	short*p1=(short*)buf;
	*p1 = htons(1);
//==============================
	char *p2 = buf+2;
	strcpy(p2,name);
//=============================
	char *p4 = p2+strlen(p2)+1;
	strcpy(p4,"octet");
//===============================
int len = 4+strlen(p2)+strlen(p4);
printf( "%c%c%s%c%s%c\n", 0, 1, name,0, "octet", 0);
if(sendto(cfd, buf, len, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
{
	ERR_MSG("sendto");
	return -1;
}
printf("发送成功\n");

//变量最好初始化,且初始化为无意义的数据
int fd = -1;
socklen_t addrlen = sizeof(sin);
ssize_t res = 0
int ret = 0;
unsigned short num = 0;
while (1)
{
	bzero(buf, sizeof(buf));
	//接收数据包
	res = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
	printf("%d\n",buf[1]);
	if (res < 0)
	{
		ERR_MSG("recvfrom");
		ret = -1;
		break;
	}
	//下列测试可以得到结果,有效操作码存储在buf[1]的位置,buf[0]的位置是0
	//printf("操作码:%#x | %d %d\n", ntohs(*(short*)buf), buf[0], buf[1]);
	//printf("    快编号:%d | %d %d\n", ntohs(*(short*)(buf+2)), buf[2], buf[3]);

	if (3 == buf[1])     //数据包
	{
		//判断数据包中的块编号是否是正确的块编号
		//需要与本地记录的块编号进行对比
		if (*(unsigned short*)(buf + 2) == htons((num + 1)))
		{
			num++;
			if (1 == ntohs(*(unsigned short*)(buf + 2)))
			{
				fd = open(name, O_WRONLY | O_CREAT | O_TRUNC, 0664);
				if (fd < 0)
				{
					ERR_MSG("open");
					return -1;
				}
			}

			//将接收到的数据另存
			if (write(fd, buf + 4, res - 4) < 0)
			{
				ERR_MSG("write");
				ret = -1;
				break;
			}


			//回复ACK,由于ACK包和数据包前四个字节只有操作码不一致
			//所以可以直接修改数据包的操作码,然后将前4个字节回复给服务器即可
			buf[1] = 4;
			if (sendto(cfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				ret = -1;
				break;
			}

			//直到数据小于512个字节,实际获取到的数据包大小为res
			//数据大小= res - 操作码 - 块编号
			if (res - 4 < 512)
			{
				printf("%s 文件拷贝完毕\n", name);
				break;
			}
		}
	}
	else if (5 == buf[1])    //错误包
	{
		fprintf(stderr, "DOWNLOAD_ERROR:%d : %s\n", ntohs(*(short*)(buf + 2)), buf + 4);
		break;
	}
}

close(fd);
return ret;



}
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd = socket(AF_INET,SOCK_DGRAM,0);
	if(cfd < 0)
	{		
		ERR_MSG("socket");
		return -1;
	}
	printf("create soccess sfd=%d __%d++\n",cfd,__LINE__);
	//填充服务器的地址信息结构体,给bind函数绑定
	
	//真实的地址信息结构体柑橘地址族定制
	struct sockaddr_in sin;
	
		sin.sin_family  =AF_INET;//必须填AF_TNET;
		sin.sin_port    = htons(PORT);//1024-49151,网络字节序
		sin.sin_addr.s_addr = inet_addr(IP);//ifconfig出来本机ip
		

		char c=0;
		while(1)
		{
			printf("---------------------------\n");
			printf("----------1. 下载----------\n");
			printf("----------2. 上传----------\n");
			printf("----------3. 退出----------\n");
			printf("---------------------------\n");	
			c = getchar();
			while(getchar() != 10);
			switch(c)
			{
			case '1':
				xia_zai(cfd, sin);
				break;
			case '2':
				
				break;
		//	case '3':
		//		goto END;
		//
		//break;
			default:
				printf("输入错误,请重新输入\n");
			}

			printf("输入任意字符清屏>>>");
			while(getchar()!=10);

		}

		//关闭套接字文件描述符


		close(cfd);
		return 0;
}

UDP上传下载

聊天服务器

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <arpa/inet.h>
 
#define N 128
//协议
#define LOGIN 	1
#define CHAT 	2
#define QUIT 	3
 
#define ERR_LOG(msg) do{\
	printf("%d %s %s \n", __LINE__, __func__, __FILE__);\
	perror(msg);\
}while(0)
 
//收发信息的结构体
typedef struct
{
	int type;
	char name[20];
	char text[N] ;
 
}MSG;
 
 
typedef struct node
{
	struct sockaddr_in cin;
	struct node* next;
}__linklist;
 
int do_recv(int sfd, __linklist* head);
int do_login(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin);
int do_quit(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin);
int do_chat(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin);
int do_system(int sfd, struct sockaddr_in sin);
 
int main(int argc, const char *argv[])
{
	if(argc < 3)
	{
		printf("请输入 ip 和端口号\n");
		return -1;
	}
 
	//1.创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		perror("socket");
		return -1;
	}
 
	//2.绑定服务器ip和端口号
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(atoi(argv[2]));
	sin.sin_addr.s_addr = inet_addr(argv[1]);
 
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		perror("bind");
		return -1;
	}
	
	//创建进程或线程
	int pid = 0;
	pid = fork();
	if(pid > 0)
	{
		//父进程, 接收消息
		__linklist* head = (__linklist*)malloc(sizeof(__linklist));
		head->next = NULL;
		do_recv(sfd , head);
 
	}
	else if(pid==0)
	{
		//子进程,发送系统消息
		do_system(sfd, sin);
	}
 
 
	//4.关闭套接字
	close(sfd);
	return 0;
}
 
int do_system(int sfd, struct sockaddr_in sin)
{
	MSG sys_msg = {htonl(CHAT), "**system**"};
 
	while(1)
	{
		bzero(sys_msg.text, N);
		fgets(sys_msg.text, N, stdin);
		sys_msg.text[strlen(sys_msg.text)-1] = 0;
 
		//将当前进程当做客户端,父进程当做服务器,发送信息;
		if(sendto(sfd, &sys_msg, sizeof(sys_msg), 0, (void*)&sin, sizeof(sin)) < 0)
		{
			ERR_LOG("sendto");
			return -1;
		}
	}
	printf("系统消息发送成功");
	return 0;
}
 
int do_recv(int sfd, __linklist* head)
{
	MSG rcv_msg;
	int recv_len = 0;
	struct sockaddr_in cin;
	socklen_t clen = sizeof(cin);
 
	while(1)
	{
		//接收消息
		recv_len = recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, (void*)&cin, &clen);
		if(recv_len < 0)
		{
			ERR_LOG("recvfrom");
			return -1;
		}
		//提取出协议
		//将网络字节序转换为主机字节序
		int type = ntohl(rcv_msg.type);
 
		switch(type)
		{
		case LOGIN:
			do_login(sfd, head, rcv_msg, cin);
			break;
		case CHAT:
			do_chat(sfd, head, rcv_msg, cin);
			break;
		case QUIT:
			do_quit(sfd, head, rcv_msg, cin);
			break;
		}
 
	}
}
 
int do_chat(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin)
{
	
	printf("%s [%s:%d]chat成功\n", rcv_msg.name,\
			(char*)inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
	//重新拼接群聊消息: 名字+消息
	char buf[258] = "";
	sprintf(buf, "%s:%s", rcv_msg.name, rcv_msg.text);
	strcpy(rcv_msg.text, buf);
 
	//循环发送,除了自己以外的ip地址
	while(head->next != NULL)
	{
		head = head->next;
		if(memcmp(&cin, &head->cin, sizeof(cin)) != 0)
		{	
			if(sendto(sfd, &rcv_msg, sizeof(rcv_msg),0, (void*)&(head->cin), sizeof(head->cin))<0)
			{
				ERR_LOG("sendto");
				return -1;
			}
		}
	}
	return 0;	
}
 
 
 
int do_quit(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin)
{
	sprintf(rcv_msg.text, "-----%s 已下线-----\n", rcv_msg.name);
	//循环遍历链表,发送"xxx一下线"的信息
	//发送给当前客户端外的其余客户端
	while(head->next != NULL)
	{
		if(memcmp(&cin, &head->next->cin, sizeof(cin)) == 0)
		{
			//从链表中删除该客户端信息
			//free
			__linklist* temp = head->next;
			head->next = temp->next;
			free(temp);
		}
		else
		{
			head = head->next;
			if(sendto(sfd, &rcv_msg, sizeof(rcv_msg),0, (void*)&head->cin, sizeof(head->cin))<0)
			{
				ERR_LOG("sendto");
				return -1;
			}
		}
	}
	return 0;
}
 
 
int do_login(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin)
{
	//打印登录成功
	printf("%s [%s:%d]登录成功\n", rcv_msg.name,\
			inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
 
	sprintf(rcv_msg.text, "-----%s登录成功-----", rcv_msg.name);
 
 
	while(head->next != NULL)
	{
		head = head->next;
		if(sendto(sfd, &rcv_msg, sizeof(rcv_msg), 0, (void*)&(head->cin), sizeof(head->cin))<0)
		{
			ERR_LOG("sendto");
			return -1;
		}
	}
 
	//将登录成功的客户端信息添加到链表中;
	__linklist *temp = (__linklist*)malloc(sizeof(__linklist));
	temp->cin = cin;
	temp->next= NULL;
	
	head->next = temp;
	return 0;
 
}

聊天客户端

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/wait.h>
#include <arpa/inet.h>
 
#define N 128
//协议
#define LOGIN 	1
#define CHAT 	2
#define QUIT 	3
 
#define ERR_LOG(msg) do{\
	printf("%d %s %s \n", __LINE__, __func__, __FILE__);\
	perror(msg);\
}while(0)
 
//收发信息的结构体
typedef struct
{
	int type;
	char name[20];
	char text[N] ;
 
}MSG;
 
typedef void (*sighandler_t)(int);
 
int do_recv(int sfd);
int do_chat(int sfd, MSG msg, struct sockaddr_in sin);
 
void handler(int sig)
{
	//回收子进程资源并退出
	while(waitpid(-1,NULL, WNOHANG)>0);
	exit(0);
}
 
int main(int argc, const char *argv[])
{
	if(argc < 3)
	{   
		printf("请输入 ip 和端口号\n");
		return -1; 
	} 
 
	//注册信号处理函数,让子进程退出后,父进程回收子进程资源并退出
	sighandler_t s = signal(SIGCHLD, handler);
	if(s == SIG_ERR)
	{
		ERR_LOG("signal");
		return -1;
	}
 
	//1.创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		perror("socket");
		return -1;
	}
 
	//2.填充服务器ip和端口号
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(atoi(argv[2]));
	sin.sin_addr.s_addr = inet_addr(argv[1]);
	socklen_t addrlen = sizeof(sin);
 
 
	//登录协议
	MSG msg ;
	msg.type = htonl(LOGIN);
 
	printf("请输入姓名>>>");
	fgets(msg.name, 20, stdin);
	msg.name[strlen(msg.name)-1] = 0;
 
	//发送登录信息sendto
	if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
	{
		ERR_LOG("sendto");	
		return -1;
	}
 
	pid_t pid = fork();
 
	if(pid > 0)
	{
		//父进程获取信息
		do_recv(sfd);
	}
	else if(0 == pid)
	{
		//子进程发送信息
		do_chat(sfd, msg, sin);
	}
 
 
	//4.关闭套接字
	close(sfd);
	return 0;
}
 
 
int do_chat(int sfd, MSG msg, struct sockaddr_in sin)
{
	while(1)
	{
		//从终端获取聊天信息
		bzero(msg.text, N);
		fgets(msg.text, N ,stdin);
		msg.text[strlen(msg.text)-1] = 0;
 
		//如果是聊天信息,则封装上 CHAT协议 
		//如果是退出信息,则封装上 QUIT协议
		if(strncasecmp(msg.text, "quit" , 4) == 0)
		{
			msg.type = htonl(QUIT);
		}
		else
		{
			msg.type = htonl(CHAT);
		}
 
		//发送 
		if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
		{
			ERR_LOG("sendto");
			return -1;
		}
 
		//如果是退出信息,则客户端要退出
		//子进程:即当前进程,需要退出
		//且父进程也要退出
		if(msg.type == htonl(QUIT))
		{
			exit(0); 		//退出子进程;
		}
	}
}
 
 
 
int do_recv(int sfd)
{
	MSG rcv_msg ;
	while(1)
	{
		if(recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, NULL, NULL) < 0)
		{
			perror("recvfrom");
			return -1;
		}
		printf("%s\n",rcv_msg.text);
	}
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值