网络编程第三天

服务器：

#include<sys/types.h> // 支持套接字地址结构
#include <sys/socket.h> // 提供套接字API
#include <netinet/in.h> // 定义IP地址结构体
#include <string.h> // 提供字符串操作函数
#include <stdio.h> // 提供标准I/O操作函数
#include <arpa/inet.h> // 提供IP地址转换函数
#include <stdlib.h>
#include "string.h"
#include "pthread.h"

// udo组播代码
int main(int argc, char *argv[])
{
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0); // 创建UDP套接字
    if (sockfd < 0)
    {
        perror("socket");
        return -1;
    }

    struct ip_mreq mreq; // 定义IP多播结构体
    mreq.imr_multiaddr.s_addr = inet_addr("224.10.10.10"); // 设置多播组IP地址.
    mreq.imr_interface.s_addr = INADDR_ANY; // 设置本地接口为任何接口

    // 加入多播组
    if (setsockopt(sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq)) < 0)
    {
        perror("setsockopt");
        return -1;
    }

    struct sockaddr_in server_addr; // 定义服务器地址结构体
    memset(&server_addr, 0, sizeof(server_addr)); // 初始化服务器地址结构体
    server_addr.sin_family = AF_INET; // 设置地址族为IPv4
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY); // 设置服务器IP地址为任何地址
    server_addr.sin_port = htons(atoi(argv[1])); // 设置服务器端口号为8000

    // 绑定套接字到服务器地址
    if (bind(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0)
    {
        perror("bind");
        return -1;
    }

    char buffer[1024]; // 定义缓冲区
    struct sockaddr_in client_addr; // 定义客户端地址结构体
    socklen_t client_addr_len = sizeof(client_addr); // 定义客户端地址长度

    while (1)
    {
        memset(buffer, 0, sizeof(buffer)); // 清空缓冲区
        int len = recvfrom(sockfd, buffer, sizeof(buffer), 0, (struct sockaddr *)&client_addr, &client_addr_len); // 接收数据
        if(strncmp(buffer,"start",5)==0)
        {
            char buf[1024];
            snprintf(buf,1024,"%s:%d 上线了",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
            sendto(sockfd,buf,strlen(buf),0,(struct sockaddr *)&server_addr,client_addr_len);
            continue;
        }
        else if(strncmp(buffer,"end",3)==0)
        {
            char buf[1024];
            snprintf(buf,1024,"%s:%d 下线了",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
            sendto(sockfd,buf,strlen(buf),0,(struct sockaddr *)&server_addr,client_addr_len);
            continue;
        }
        if (len < 0)
        {
            perror("recvfrom");
            continue;
        }
        buffer[len] = '\0';
        printf("Received message: %s\n", buffer); // 打印接收到的消息
    }

    return 0;
}

客户端：

#include<sys/types.h> // 支持套接字地址结构
#include <sys/socket.h> // 提供套接字API
#include <netinet/in.h> // 定义IP地址结构体
#include <string.h> // 提供字符串操作函数
#include <stdio.h> // 提供标准I/O操作函数
#include <arpa/inet.h> // 提供IP地址转换函数
#include <stdlib.h>
#include "string.h"
#include "pthread.h"
#include "signal.h"

int sockfd;
struct sockaddr_in mcast_addr; // 定义组播地址结构体


void func(void)
{
    sendto(sockfd, "end", strlen("end"), 0, (struct sockaddr *)&mcast_addr, sizeof(mcast_addr));
    
    exit(0);
}


// UDP组播
int main(int argc, char *argv[])
{
    
    sockfd = socket(AF_INET, SOCK_DGRAM, 0); // 创建UDP套接字
    if (sockfd < 0)
    {
        perror("socket");
        return -1;
    }

    
    mcast_addr.sin_family = AF_INET; // 设置地址族为IPv4
    mcast_addr.sin_addr.s_addr = inet_addr("224.10.10.10"); // 设置组播组IP地址
    mcast_addr.sin_port = htons(atoi(argv[1])); // 设置组播组端口号
    sendto(sockfd, "start", strlen("start"), 0, (struct sockaddr *)&mcast_addr, sizeof(mcast_addr));
    signal(SIGINT, (void *)func);
    // 循环发送数据
    while (1)
    {
        char buffer[1024];
        printf("Enter message to send: ");
        fgets(buffer, sizeof(buffer), stdin);
        buffer[strlen(buffer) - 1] = '\0'; // 去掉fgets获取的回车

        int len = sendto(sockfd, buffer, strlen(buffer), 0, (struct sockaddr *)&mcast_addr, sizeof(mcast_addr));
        if (len < 0)
        {
            perror("sendto");
            continue;
        }
    }

    return 0;
}