网络编程——聊天程序实现

1、UDP实现

服务器端:

cpp 复制代码
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h> 
#include <time.h>
#include <unistd.h>
#include<pthread.h>
typedef struct sockaddr *(SA);
//th函数是一个线程函数,负责持续接收来自客户端的消息
void* th(void* arg)
{
    int fd=*(int*)arg;
    char buf[512];
    while (1)
    {
        recvfrom(fd, buf, sizeof(buf), 0, NULL,NULL);
        if(strcmp(buf, "#quit")==0)
        {
            printf("client has left\n");
            exit(0);
        }
        printf("Me:%s\n",buf);
    }
    return NULL;
}
int main()
{
//1创建UDP套接字
    int udpfd=socket(AF_INET, SOCK_DGRAM, 0);
    if(-1==udpfd)
    {
        perror("socket");
        return 1;
    }
//2设置服务器地址
    struct sockaddr_in ser,cli;
    bzero(&ser, sizeof(ser));
    bzero(&cli, sizeof(cli));
    ser.sin_family=AF_INET;
    ser.sin_port=htons(50000);
    ser.sin_addr.s_addr=inet_addr("192.168.31.86");
//3绑定套接字
    int ret=bind(udpfd, (SA)&ser, sizeof(ser));
    if(-1==ret)
    {
        perror("bind");
        return 1;
    }
//4等待客户端连接
    char buf[512];
    socklen_t len=sizeof(cli);
    recvfrom(udpfd, buf, sizeof(buf), 0, (SA)&cli, &len);
//5创建接收线程
    pthread_t tid;
    pthread_create(&tid,NULL,th,&udpfd);
//6主线程循环发送消息给客户端
    printf("You:");
    while (1)
    {
        fgets(buf, sizeof(buf), stdin);
        sendto(udpfd, buf, sizeof(buf), 0, (SA)&cli, len);
        if(0==strcmp("#quit", buf))
        {
            exit(0);
        }
        printf("You:");
    }
//清理工作
    pthread_join(tid1, NULL);
    close(udpfd);
    return 0;
}

客户端:

cpp 复制代码
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h> 
#include <time.h>
#include <unistd.h>
#include<pthread.h>
typedef struct sockaddr *(SA);
//th函数是一个线程函数,负责持续接收来自服务器的消息
void* th1(void* arg)
{
    int fd=*(int*)arg;
    while (1)
    {
        char buf[128]={0};
        recvfrom(fd, buf, sizeof(buf), 0, NULL, NULL);
        if(strcmp(buf, "#quit")==0)
        {
            printf("server has left\n");
            exit(0);
        }
        printf("You:%s\n",buf);
    }
    return NULL;
}

int main()
{
//1创建UDP套接字
    int udpfd=socket(AF_INET, SOCK_DGRAM, 0);
    if(-1==udpfd)
    {
        perror("socket");
        return 1;
    }
//2设置服务器地址
    struct sockaddr_in ser;
    bzero(&ser, sizeof(ser));
    ser.sin_family=AF_INET;
    ser.sin_port=htons(50000);
    ser.sin_addr.s_addr=inet_addr("192.168.31.86");
//3发送消息建立初始链接
    char buf[512]="start";
    sendto(udpfd, buf, strlen(buf), 0, (SA)&ser, sizeof(ser));
//4创建接收线程
    pthread_t tid1,tid2;
    pthread_create(&tid1,NULL,th1,&udpfd);
//5主线程循环发送消息到服务器
    printf("You:");
    while (1)
    {
        char buf[512];
        fgets(buf, sizeof(buf), stdin);
        sendto(udpfd, buf, sizeof(buf), 0, (SA)&ser, sizeof(ser));
        if(0==strcmp("#quit", buf))
        {
            exit(0);
        }
        printf("You:");
    }
//清理工作
    pthread_join(tid1, NULL);
    close(udpfd);
    return 0;
}

2、TCP实现

服务器端:

cpp 复制代码
#include<netinet/in.h>
#include<netinet/ip.h>
#include <pthread.h>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include <strings.h>
#include<sys/socket.h>
#include<sys/types.h>
#include <thread_db.h>
#include<time.h>
#include<unistd.h>
typedef struct sockaddr*(SA);
//th函数是一个线程函数,负责持续接收来自客户端的消息
void *th(void *arg)
{
    int fd=*(int*)arg;
    char buf[512];
    while (1)
    {
        recv(fd, buf, sizeof(buf), 0);
        if(strcmp("#quit\n",buf)==0)
        {
            exit(0);
        }
        printf("You:%s",buf);
    }
    return NULL;
}
int main()
{
//1创建TCP监听套接字
    int listfd=socket(AF_INET, SOCK_STREAM, 0);
    if(-1==listfd)
    {
        perror("socket");
        return 1;
    }
//2设置服务器地址
    struct sockaddr_in ser,cli;
    bzero(&ser, sizeof(ser));
    bzero(&cli, sizeof(cli));
    ser.sin_family=AF_INET;
    ser.sin_port=htons(50000);
    ser.sin_addr.s_addr=INADDR_ANY;
//3绑定套接字
    int ret=bind(listfd, (SA)&ser, sizeof(ser));
    if(-1==ret)
    {
        perror("bind");
        return -1;
    }
//4开始监听
    listen(listfd, 3);//3代表同一时间可以与服务器建立连接的排队数
//5接受客户端连接
    socklen_t len=sizeof(cli);
    int conn=accept(listfd, (SA)&cli, &len);
    if(-1==conn)
    {
        perror("conn");
        return -1;
    }
//6接收初始消息建立连接
    char  buf[512];
    recv(conn, buf, strlen(buf), 0);
//7创建接收线程
    pthread_t tid;
    pthread_create(&tid, NULL, th, &conn);
//8主线程循环发送消息到客户端
    while (1)
    {
        printf("You:");
        fgets(buf, sizeof(buf), stdin);
        // buf[strlen(buf)-1]='\0';
        send(conn, buf, strlen(buf), 0);
        if(0==strcmp("#quit\n", buf))
        {
            exit(0);
        }
    }
//清理工作
    pthread_join(tid, NULL);
    close(listfd);
    close(conn);
    return 0;
}

客户端:

cpp 复制代码
#include <arpa/inet.h>
#include<netinet/in.h>
#include<netinet/ip.h>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include <strings.h>
#include<sys/socket.h>
#include<sys/types.h>
#include <thread_db.h>
#include<time.h>
#include<unistd.h>
#include<pthread.h>
typedef struct sockaddr*(SA);
//th函数是一个线程函数,负责持续接收来自服务器的消息
void *th(void *arg)
{
    int fd=*(int*)arg;
    while (1) 
    {
        char buf[128]={0};
        recv(fd, buf, sizeof(buf), 0);
        if(strcmp("#quit\n",buf)==0)
        {
            exit(0);
        }
        printf("You:%s",buf);
    }
    return NULL;
}
int main()
{
//1创建TCP套接字
    int conn=socket(AF_INET, SOCK_STREAM, 0);
    if(-1==conn)
    {
        perror("socket");
        return 1;
    }
//2设置服务器地址
    struct sockaddr_in ser;
    bzero(&ser, sizeof(ser));
    ser.sin_family=AF_INET;
    ser.sin_port=htons(50000);
    ser.sin_addr.s_addr=inet_addr("192.168.31.86");
//3连接服务器
    int ret=connect(conn, (SA)&ser, sizeof(ser));
    if(-1==ret)
    {
        perror("connect");
        return 1;
    }
//4发送初始消息建立连接
    char buf[512]="start";
    send(conn, buf, strlen(buf), 0);
//5创建接收线程
    pthread_t tid;
    pthread_create(&tid, NULL, th, &conn);
//6主线程循环发送消息到服务器  
    while (1)
    {
        printf("You:");
        char buf[512];
        fgets(buf, sizeof(buf), stdin);
        // buf[strlen(buf)-1]='\0';
        send(conn, buf, strlen(buf), 0);
        if(0==strcmp("#quit\n", buf))
        {
            exit(0);
        }
    }
//清理工作
    pthread_join(tid, NULL);
    close(conn);
    return 0;
}
相关推荐
FPGA小徐10 小时前
STM32五路气敏传感器阵列电路详解:从HHC1000分压到AD8606跟随器和ADC采样
stm32·单片机·嵌入式硬件
程序员-Benothing10 小时前
Shell输入输出详解:read、echo、printf、颜色输出
linux·运维·服务器
FPGA小徐10 小时前
从 NEMU 到 AbstractMachine:程序、运行时环境与 AM 完全解析
linux·运维·服务器
Horn Still Sounds12 小时前
IMX6ULL裸机|I2C通信协议 + ADC模数转换学习笔记
arm开发·笔记·单片机·嵌入式硬件
智感子12 小时前
无人值守称重:地磅系统架构与流程
嵌入式硬件
新晨单片机设计14 小时前
S011C-基于STM32智能灯光控制系统(光强、声音、人体)【Proteus仿真+Keil程序+原理图】
stm32·嵌入式硬件·proteus·智能灯光控制系统
..Dauntless..14 小时前
【Linux】进程地址空间初步理解
linux·运维·服务器
Julien200414 小时前
Docker 网络(一)
linux·运维·服务器·ssh·学习方法
杨云龙UP15 小时前
一次数据库查询缓慢故障复盘:大表数据增长、SQL全表扫描导致系统响应异常
linux·运维·服务器·数据库·sql·mysql
女神下凡16 小时前
芯参谋(26):DDR4_软件设计规范
arm开发·单片机·嵌入式硬件