基于Li nux下TCP/UDP聊天室 :客户端一对一聊天(使用tcp协议,服务器转发信息) 客户端与客户端传输文件(使用UDP协议) 客户端上 C语言 代码 + 注释

以下是基于Linux下TCP/UDP聊天室的客户端一对一聊天和文件传输的C语言代码和注释：

TCP聊天室客户端代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
 
#define BUF_SIZE 100
void error_handling(char *message);
 
int main(int argc, char *argv[])
{
    int sock;
    char message[BUF_SIZE];
    int str_len;
    struct sockaddr_in serv_adr;
 
    if(argc!=3) {
        printf("Usage : %s <IP> <port>\n", argv[0]);
        exit(1);
    }
 
    sock=socket(PF_INET, SOCK_STREAM, 0); //创建套接字
    if(sock == -1)
        error_handling("socket() error");
  
    memset(&serv_adr, 0, sizeof(serv_adr));
    serv_adr.sin_family=AF_INET;
    serv_adr.sin_addr.s_addr=inet_addr(argv[1]);
    serv_adr.sin_port=htons(atoi(argv[2]));
 
    if(connect(sock, (struct sockaddr*)&serv_adr, sizeof(serv_adr))==-1) //连接服务器
        error_handling("connect() error!");
    else
        puts("Connected...........");
 
    while(1) //循环发送接收消息
    {
        fputs("Input message(Q to quit): ", stdout);
        fgets(message, BUF_SIZE, stdin);
         
        if(!strcmp(message,"q\n") || !strcmp(message,"Q\n")) //输入'q'或'Q'退出
            break;
 
        write(sock, message, strlen(message)); //发送消息
        str_len=read(sock, message, BUF_SIZE-1); //接收消息
        message[str_len]=0;
        printf("Message from server: %s", message); //输出接收到的消息
    }
 
    close(sock); //关闭套接字
    return 0;
}
 
void error_handling(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

UDP聊天室客户端代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
 
#define BUF_SIZE 30
void error_handling(char *message);
 
int main(int argc, char *argv[])
{
    int sock;
    char message[BUF_SIZE];
    int str_len;
    socklen_t adr_sz;
    struct sockaddr_in serv_adr, from_adr;
 
    if(argc!=3) {
        printf("Usage : %s <IP> <port>\n", argv[0]);
        exit(1);
    }
 
    sock=socket(PF_INET, SOCK_DGRAM, 0); //创建套接字
    if(sock == -1)
        error_handling("sock() error");
  
    memset(&serv_adr, 0, sizeof(serv_adr));
    serv_adr.sin_family=AF_INET;
    serv_adr.sin_addr.s_addr=inet_addr(argv[1]);
    serv_adr.sin_port=htons(atoi(argv[2]));
 
    while(1) //循环发送接收消息
    {
        fputs("Insert message(q to quit): ", stdout);
        fgets(message, sizeof(message), stdin);
        if(!strcmp(message,"q\n") || !strcmp(message,"Q\n")) //输入'q'或'Q'退出
            break;
 
        sendto(sock, message, strlen(message), 0, 
               (struct sockaddr*)&serv_adr, sizeof(serv_adr)); //发送消息
        adr_sz=sizeof(from_adr);
        str_len=recvfrom(sock, message, BUF_SIZE, 0, 
                          (struct sockaddr*)&from_adr, &adr_sz); //接收消息
        message[str_len]=0;
        printf("Message from server: %s", message); //输出接收到的消息
    }
    close(sock); //关闭套接字
    return 0;
}
 
void error_handling(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

文件传输客户端代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <netinet/in.h>
 
#define BUF_SIZE 1024
void error_handling(char *message);
 
int main(int argc, char *argv[])
{
    int sock;
    char buf[BUF_SIZE];
    int read_cnt;
    struct sockaddr_in serv_adr;
    int fd;
    char file_name[BUF_SIZE];
 
    if(argc!=4) {
        printf("Usage : %s <IP> <port> <file name>\n", argv[0]);
        exit(1);
    }
 
    sock=socket(PF_INET, SOCK_STREAM, 0); //创建套接字
    if(sock == -1)
        error_handling("socket() error");
 
    memset(&serv_adr, 0, sizeof(serv_adr));
    serv_adr.sin_family=AF_INET;
    serv_adr.sin_addr.s_addr=inet_addr(argv[1]);
    serv_adr.sin_port=htons(atoi(argv[2]));
 
    if(connect(sock, (struct sockaddr*)&serv_adr, sizeof(serv_adr))==-1) //连接服务器
        error_handling("connect() error!");
 
    sprintf(file_name, "%s", argv[3]);
    if((fd=open(file_name, O_RDONLY))==-1) //打开要发送的文件
        error_handling("open() error");
 
    send(sock, file_name, strlen(file_name), 0); //发送文件名
    while((read_cnt=read(fd, buf, BUF_SIZE ))>0) //循环发送文件内容
    {
        write(sock, buf, read_cnt);
    }
    shutdown(sock, SHUT_WR); //文件传输完成后，向服务器发送FIN包
    read(sock, buf, BUF_SIZE); //接收服务器返回的ACK包
    printf("Message from server: %s \n", buf); //输出接收到的ACK包
    close(fd); //关闭文件
    close(sock); //关闭套接字
    return 0;
}
 
void error_handling(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}