3.7作业

网络聊天室：

程序代码：

ser.c

#include <myhead.h>
//定义消息类型结构体
struct xiaoxi {
    char type;
    char name[20];
    char text[100];
};

int main(int argc, const char* argv[])
{
    // 创建套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sfd  == -1) 
	{
        perror("socket error");
        return -1;
    }
    // 设置端口快速重用
    int reuse = 1;
    if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1) 
	{
        perror("setsockopt error");
        return -1;
    }

    // 绑定
    // 定义地址信息结构体
    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)) == -1) 
	{
        perror("bind error\n");
        return -1;
    }

    // 设置监听
    if ((listen(sfd, 128)) == -1)
	{
        perror("listen error\n");
        return -1;
    }
    // 4接受请求
    // 存储客户端信息结构体
    struct sockaddr_in cin;
    cin.sin_family = AF_INET;
    cin.sin_port = htons(atoi(argv[2]));
    cin.sin_addr.s_addr = inet_addr(argv[1]);
    socklen_t socklen = sizeof(cin);
    struct sockaddr_in arr_cin[1024];

    // 定义
    struct pollfd fds[1024];
    for (int i = 0; i < 1024; i++) 
	{
        fds[i].fd = -1;
        fds[i].events = POLLIN;
    }
    //将0和sfd放入数组
    fds[0].fd = 0;
    fds[1].fd = sfd;

    struct xiaoxi rec;//消息类型结构体变量
    char buf[128] = "";//接受消息的容器
    int newfd = -1; //存储文件描述符
    while (1) 
	{
        int res = poll(fds, 1024, -1);
        if (res < 0)
		{
            perror("poll error");
            return -1;
        }
		else if (res == 0)
		{
            printf("time out\n");
            return -1;
        }
        if (fds[0].revents == POLLIN)  //终端标准输入描述符
		{
            bzero(buf, sizeof(buf));
            fgets(buf, sizeof(buf), stdin);
            buf[strlen(buf) - 1] = 0;
            for (int j = 4; j < 1024; j++)
			{
                send(fds[j].fd, buf, sizeof(buf), 0);
            }
        }
        if (fds[1].revents == POLLIN)//sfd
		{ 
            if ((newfd = accept(sfd, (struct sockaddr*)&cin, &socklen)) == -1) 
			{
                perror("accept error\n");
            }
            arr_cin[newfd] = cin; //将地址信息存入cin
            printf("------[%s:%d]连接成功------\n", inet_ntoa(arr_cin[newfd].sin_addr), htons(arr_cin[newfd].sin_port));
            fds[newfd].fd = newfd; //新生成的文件描述符放入数组
            fds[newfd].events = POLLIN;
        }
        for (int i = 4; i < 1024; i++) 
		{
            if (fds[i].fd < 0) 
			{
                continue;
            }
            if (fds[i].revents == POLLIN) 
			{
                bzero(buf, sizeof(buf));
                bzero(buf, sizeof(buf));
                int res = recv(fds[i].fd, &rec, sizeof(rec), 0); //接收客户端消息
                if (res == 0)
				{
                    close(fds[i].fd);
                    printf("------[%s:%d]断开连接------\n", inet_ntoa(arr_cin[fds[i].fd].sin_addr), htons(arr_cin[fds[i].fd].sin_port));
                    fds[i].fd = -1;
                    arr_cin[fds[i].fd].sin_addr.s_addr = 0;
                    arr_cin[fds[i].fd].sin_port = 0;
                    continue;
                }
                // 判断消息类型
                if (rec.type == 'L') // 登录消息
				{ 
                    for (int j = 4; j < 1024; j++)//发送给每个客户端
					{ 
                        if (j == i)
						{
                            continue;
                        }
                        bzero(buf, sizeof(buf));
                        sprintf(buf, "----------%s上线-----------", rec.name);
                        send(fds[j].fd, buf, strlen(buf), 0);
                    }
                } 
				else if (rec.type == 'C') // 聊天消息
				{ 
                    for (int j = 4; j < 1024; j++) 
					{
                        if (j == i) 
						{
                            continue;
                        }
                        bzero(buf, sizeof(buf));
                        sprintf(buf, "%s:%s", rec.name, rec.text);
                        send(fds[j].fd, buf, strlen(buf), 0);
                    }
                } 
				else 
				{                        //退出消息
                    //(rec.type == 'Q')
                    for (int j = 4; j < 1024; j++) 
					{
                        if (j == i)
						{
                            continue;
                        }
                        bzero(buf, sizeof(buf));
                        sprintf(buf, "-----------%s下线------------", rec.name);
                        send(fds[j].fd, buf, strlen(buf), 0);
                    }
                }
            }
        }
    }
    return 0;
}

cli.c

#include <myhead.h>

struct xiaoxi {
    char type;
    char name[20];
    char text[100];
};

int main(int argc, const char* argv[])
{
    // 创建套接字
    int cfd = socket(AF_INET, SOCK_STREAM, 0);
    if (cfd  == -1) 
	{
        perror("socket error\n");
        return -1;
    }

    // 端口快速重用
    int reus = 1;
    if ((reus = setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reus, sizeof(reus))) == -1) 
	{
        perror("setsockopt error\n");
        return -1;
    }
	

    // 定义地址信息结构体
    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 (connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) == -1)
	{
        perror("connect error");
        return -1;
    }


    struct xiaoxi sendi;

    char buf[100] = "";
    char nbuf[20] = "";

    
    struct pollfd fds[2];
    fds[0].fd=0;
    fds[0].events = POLLIN;

    fds[1].fd=cfd;
    fds[1].events = POLLIN;




    printf("请输入用户名:");
    fgets(nbuf,sizeof(nbuf),stdin);
    nbuf[strlen(nbuf)-1]=0;
    sendi.type = 'L';
    strcpy(sendi.name, nbuf);

    send(cfd, &sendi, sizeof(sendi), 0);
    while (1) 
	{
        int res = poll(fds,2,-1);           //阻塞检测
        if (fds[0].revents == POLLIN) 
        {
            bzero(buf, sizeof(buf));
            fgets(buf,sizeof(buf),stdin);
            buf[strlen(buf)-1]=0;
            sendi.type = 'C';    
            if(strcmp(buf,"quit")==0)
            {
                sendi.type = 'Q';
                sendto(cfd, &sendi, sizeof(sendi), 0, (struct sockaddr*)&sin, sizeof(sin));
                break;
            }
            strcpy(sendi.text,buf);
            sendto(cfd, &sendi, sizeof(sendi), 0, (struct sockaddr*)&sin, sizeof(sin));
        }
        if (fds[1].revents == POLLIN)
		{
            bzero(buf, sizeof(buf));
            int res = recv(cfd, buf, sizeof(buf), 0);
            printf("%s\n", buf);
            if (res == 0) 
			{
                printf("服务器下线\n");
                break;
            }
            
        }
    }
    close(cfd);
    return 0;
}

运行结果：