XiYuJing|文章:网络通讯与文件传输工具相关代码

[[ slide_text ]]

[[item.c]]

网络通讯与文件传输工具相关代码

发布时间：2024年12月06日 作者：靖华来源：XiYuJing

项目代码

#网络通讯与文件传输工具相关代码
##client
###client.c
```cpp
#include"tool.h"

int main(int argc,char **argv)
{
    if(argc<3)
    {
        fprintf(stderr,"Usage %s <ip> <port>",argv[0]);
        return -1;
    }
    run(argv[1],argv[2]);
}
```
###tool.c
```cpp
#include"tool.h"

//枚举
enum CMD_NO
{
    FTP_CMD_LS = 1024,	//	从服务器上获取文件名的命令
    FTP_CMD_GET,		//	从服务器上 下载文件 的命令
    FTP_CMD_PUT,		//	向服务器上 上传文件 的命令
    FTP_CMD_BYE,		//	断开连接的命令 
    FTP_CMD_UNKNOW = 0xFFFFFFFF
};

//判断字符串a,b是否相等
int My_Strcmp(const char *a,const char *b)
{
    if(strlen(a)!=strlen(b))
    {
        return 0;
    }
    for(int i=0;i<strlen(a);i++)
    {
        if(a[i]!=b[i])
        {
            return 0;
        }
    }
    return 1;
}

//查找文件是否存在
int Exist_File(char *file)
{
    char buf[2024]={0};
    DIR *dir = opendir(C_PATH);
    struct dirent *dirp =NULL;
    int i=0;
    while(dirp = readdir(dir))
    {
        if(i<2)
        {
            i++;
            continue;
        }    
        if(My_Strcmp(dirp->d_name,file))
        {
            closedir(dir);
            return 1;
        }
    }
    closedir(dir);
    return 0;
}

//上传文件
void My_Put(int socket_id,char *file,int sk)
{
    struct sockaddr_in server;
    if(Exist_File(file))
    { 
        unsigned char  buf[2024]={0};
        char path[520]={0};
        sprintf(path,"%s/%s",C_PATH,file);
        FILE *fd = fopen(path,"r");
        fseek(fd,0,SEEK_END);
        int len = ftell(fd)-sk;
        fseek(fd,sk,SEEK_SET);
        int yn= 0;
        if(len>1000)
        {
            yn=1;
            len=1000;
        }
        int k=0;
        int fn_len = strlen(file);
        //包长+命令+内容长度+是否发完+是否first+文件名长度+文件名+发送内容+文件结尾+包尾
        int b_len = 4+4+4+1+1+4+fn_len+len+1+1;
        //包头
        buf[k++] = 0xAB;
        //包长
        buf[k++] = b_len & 0xff;
        buf[k++] = (b_len>>8) &0xff;
        buf[k++] = (b_len>>16) & 0xff;
        buf[k++] = (b_len>>24) &0xff;
        //命令
        buf[k++] = FTP_CMD_PUT & 0xff;
        buf[k++] = (FTP_CMD_PUT>>8) &0xff;
        buf[k++] = (FTP_CMD_PUT>>16) & 0xff;
        buf[k++] = (FTP_CMD_PUT>>24) &0xff;
        //内容长度
        buf[k++] = len & 0xff;
        buf[k++] = (len>>8) &0xff;
        buf[k++] = (len>>16) & 0xff;
        buf[k++] = (len>>24) &0xff;
        //是否发完
        if(yn)buf[k++] = 'N';
        else buf[k++] = 'Y';
        //是否first
        if(sk==0)buf[k++]='Y';
        else buf[k++]='N';
        //文件名长度
        buf[k++] = fn_len & 0xff;
        buf[k++] = (fn_len>>8) &0xff;
        buf[k++] = (fn_len>>16) & 0xff;
        buf[k++] = (fn_len>>24) &0xff;
        //文件名
        for(int j=0;j<fn_len;j++)
        {
            buf[k++]=file[j];
        }
        
        char ch[len];
        fread(ch,len,1,fd);
        //发送内容
        for(int j=0;j<len;j++)
        {
            buf[k++]=ch[j];
        }
        buf[k++]='\0';
        //包尾
        buf[k++] = 0xAB;
        sendto(socket_id,buf,k,0,(struct sockaddr*)&server,sizeof(server));
        fclose(fd);
        if(yn)
        {
            My_Put(socket_id,file,sk+len);
        }
    }
    else 
    {
        char ch = 0xAC;
        sendto(socket_id,&ch,1,0,(struct sockaddr*)&server,sizeof(server));
    }
}

//接收信息
int Receive_informatione(int socket_id)
{
    struct sockaddr_in server;
    socklen_t len = sizeof(server);
    unsigned char buf[2024]={0};
    int i=0;
    unsigned char ch;
    int ret = read(socket_id,&ch,1);
    if(ret <= 0)
    {
        perror("文件包接收出错");
        return -1;
    }
    // printf("%d %d\n",0xAB,ch);
    if(ch==0xAB)
    {
        do{
            int ret = read(socket_id,&ch,1);
            if(ret <= 0)
            {
                perror("文件包接收出错");
                return -1;
            }
            buf[i++]=ch;
        }while(ch != 0xAB);
        int pack_len = buf[0]|(buf[1]<<8)|(buf[2]<<16)|(buf[3]<<24);
        int cmd_no = buf[4]|(buf[5]<<8)|(buf[6]<<16)|(buf[7]<<24);
        int lens = buf[8]|(buf[9]<<8)|(buf[10]<<16)|(buf[11]<<24);
        // printf("包长:%d,内容大小:%d\n",pack_len,lens);
        if(cmd_no==FTP_CMD_LS)printf("%s\n",&buf[12]);
        else if(cmd_no==FTP_CMD_GET)
        {
            char yn = buf[12];
            char ones = buf[13];
            int fn_len = buf[14]|(buf[15]<<8)|(buf[16]<<16)|(buf[17]<<24);
            unsigned char file[fn_len+1];
            for(int j=0;j<fn_len;j++)
            {
                file[j]=buf[18+j];
            }
            file[fn_len]='\0';
            unsigned char path[2024];
            sprintf(path,"%s/%s",C_PATH,file);
            FILE *fd = NULL;
            if(ones=='Y')fd = fopen(path,"w");
            else fd = fopen(path,"a");
            fwrite(&buf[18+fn_len],lens,1,fd);
            fclose(fd);
            if(yn=='Y')puts("下载成功!");
        }
        else if(cmd_no==FTP_CMD_PUT)
        {
            char file[lens+1];
            for(int j=0;j<lens;j++)
            {
                file[j]=buf[12+j];
            }
            file[lens]='\0';
            My_Put(socket_id,file,0);
        }
        else if(cmd_no==FTP_CMD_BYE)
        {
            puts("下线成功!");
            puts("请登入:");
        }
    }
    else 
    {
        buf[i++]=ch;
        while(ch!='\0')
        {
            int ret = read(socket_id,&ch,1);
            if(ret <= 0)
            {
                perror("文件包接收出错");
                return -1;
            }
            buf[i++]=ch;
        }
        puts(buf);
    }
    return 0;
}

//登录注册
int Login_Create(char *ip,char *port)
{
    int socket_id = socket(AF_INET,SOCK_STREAM,0);
    if(socket_id==-1)
    {
        perror("创建socket失败");
        return -1;
    }
    struct sockaddr_in server;
    server.sin_family=AF_INET;
    server.sin_addr.s_addr=inet_addr(ip);
    server.sin_port=htons(atoi(port));

int ret = connect(socket_id,(struct sockaddr*)&server,sizeof(server));
    if(ret ==-1)
    {
        puts("连接失败");
        return -1;
    }
    puts("连接成功");
    puts("请登入:");
    return socket_id;
}

//运行
int run(char *ip,char *port)
{
    struct timeval tm;
    fd_set rfds;
    int ret=0;
    int socket_id = Login_Create(ip,port);
    struct sockaddr_in server;
    while(1)
    {
        //清空集合
        FD_ZERO(&rfds);
        //添加需要监听的文件描述符进集合
        FD_SET(0,&rfds); 
        FD_SET(socket_id,&rfds);
        //设置超时时间
		// tm.tv_sec = 10;  //10s
		// tm.tv_usec = 0;	//0us

ret = select(socket_id+1,//遍历次数
                        &rfds,//读
                        NULL,//写
                        NULL,//出错
                        0);//等待时间，0死等
        if(ret == -1)
        {
            perror("select 创建失败");
            break;
        }
        else if(ret == 0)
        {
            continue;
        }
        if(FD_ISSET(0,&rfds))
        {
            char buf[512]={0};
            scanf("%s",buf);
            int ret = sendto(socket_id,buf,strlen(buf),0,(struct sockaddr*)&server,sizeof(server));
            if(ret==-1)
            {
                puts("服务器断开");
                return -1;
            }
        }
        if(FD_ISSET(socket_id,&rfds))
        {
            if(Receive_informatione(socket_id)==-1)
            {
                return 0;
            }
        }
    }
}
```
###tool.h
```cpp
#ifndef __TOOL_H__
#define __TOOL_H__

#include<sys/types.h>          /* See NOTES */
#include<sys/socket.h>
#include<stdlib.h>
#include<stdio.h>
#include<arpa/inet.h>
#include<pthread.h>
#include<string.h>
#include<unistd.h>
#include<sys/stat.h>
#include<linux/in.h>
#include <netinet/in.h>
#include<dirent.h>

#define C_PATH "/xxx/.../c_dir"

#define MAX(a,b) ((a)>(b)?(a):(b))
//判断字符串a,b是否相等
int My_Strcmp(const char *a,const char *b);

//查找文件是否存在
int Exist_File(char *file);

//上传文件
void My_Put(int socket_id,char *file,int sk);

//接收信息
int Receive_informatione(int socket_id);

//登录注册
int Login_Create(char *ip,char *port);

//运行
int run(char *ip,char *port);
#endif
```
##server
###server.c
```cpp
#include"tool.h"

int main(int argc,char **argv)
{
    if(argc!=3)
    {
        fprintf(stderr,"Usage %s <ip> <port>\n",argv[0]);
        return -1;
    }
    run(argv[1],argv[2]);
}
```
###tool.c
```cpp
#include"tool.h"
//枚举
enum CMD_NO
{
    FTP_CMD_LS = 1024,	//	从服务器上获取文件名的命令
    FTP_CMD_GET,		//	从服务器上 下载文件 的命令
    FTP_CMD_PUT,		//	向服务器上 上传文件 的命令
    FTP_CMD_BYE,		//	断开连接的命令 
    FTP_CMD_UNKNOW = 0xFFFFFFFF
};

US *head=NULL;

/*
    Create_Tcpsocket:创建socket套接字
        @ip     网络地址
        @port   端口号
        @retval 成功返回socket套接字 失败返回-1
*/
int Create_Tcpsocket(char *ip, char *port)
{
    //创建socket
    int socket_id = socket(AF_INET,SOCK_STREAM,0);
    if(socket_id == -1)
    {
        perror("创建socket失败");
        return -1;
    }

//设置该套接字IP地址和端口号可重用
    int on = 1;
    int ret = setsockopt(socket_id,SOL_SOCKET,SO_REUSEADDR,(void *)&on,sizeof(on));
    if(ret == -1)
    {
        perror("允许流式套接字IP地址重用失败");
        return -1;
    }

ret = setsockopt(socket_id,SOL_SOCKET,SO_REUSEPORT,(void *)&on,sizeof(on));
    if(ret == -1)
    {
        perror("允许流式套接字端口号重用失败");
        return -1;
    }

//保存绑定服务端信息
    struct sockaddr_in my;
    my.sin_family=AF_INET;          //协议簇
    my.sin_addr.s_addr=inet_addr(ip);      //ip
    my.sin_port=htons(atoi(port));  //port
    //绑定
    if(bind(socket_id,(struct sockaddr*)&my,sizeof(my))==-1)
    {
        perror("绑定失败");
        close(socket_id);
        return -1;
    }
    //监听
    if(listen(socket_id,50)==-1)
    {
        perror("监听失败");
        close(socket_id);
        return -1;
    }
    puts("创建socket成功");
    return socket_id;
}

/*
    Add:把客户端信息插入链表保存
*/
US *Add(US *data)
{
    if(head==NULL)
    {
        head=data;
    }
    else 
    {
        head->prev=data;
        data->next=head;
        head=data;
    }
    return head;
}

/*
    遍历客户端用户
*/
US *Print_User(char *name)
{
    for(US *i=head;i;i=i->next)
    {
        int f=0;
        for(int j=0;i->name[j]||name[j];j++)
        {
            if(i->name[j]!=name[j])
            {
                f=1;
                break;
            }
        }
        if(f==0&&i->name[0])
        {
            return i;
        }
    }
    return NULL;
}

//判断字符串a,b是否相等
int My_Strcmp(const char *a,const char *b)
{
    if(strlen(a)!=strlen(b))return 0;
    for(int i=0;i<strlen(a);i++)
    {
        if(a[i]!=b[i])return 0;
    }
    return 1;
}

//删除断开用户信息
void Del(US *p)
{
    if(p==head)
    {
        head=p->next;
        head->prev==NULL;
    }
    else if(p->next==NULL)
    {
        p->prev->next=NULL;
    }
    else 
    {
        p->next->prev=p->prev;
        p->prev=p->next;
    }
    free(p);
}

//登录
int Login_Create(US *us)
{
    char name[512]={0};
    int ret = recvfrom(us->id,name,sizeof(name)-1,0,(struct sockaddr*)&us->cAddr,&us->len);
    if(ret == -1)
    {
        printf("与%s通信失败\n",inet_ntoa(us->cAddr.sin_addr));
        return -1;
    }
    else if(ret > 0)
    {
        if(Print_User(name))
        {
            sendto(us->id,"该用户已被注册",strlen("该用户已被注册")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
            memset(us->name,0,sizeof(us->name));
            return -1;
        }
        else if(My_Strcmp(name,"ls")||My_Strcmp(name,"get")||My_Strcmp(name,"put")||My_Strcmp(name,"bye"))
        {
            sendto(us->id,"登录名非法",strlen("登录名非法")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
            memset(us->name,0,sizeof(us->name));
            return -1;
        }
        else 
        {
            strcpy(us->name,name);
            sendto(us->id,"登入成功",strlen("登入成功")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
            return 1;
        }
    }
    else if(ret == 0)
    {
        printf("客户端%s:%d,退出\n",inet_ntoa(us->cAddr.sin_addr),ntohs(us->cAddr.sin_port));
        Del(us);
        return -1;
    }
}

//聊天
int My_Chat(US *us,char *buf)
{
    char a[512]={0},b[512]={0};
    int j=-1;
    for(int i=0;i<strlen(buf);i++)
    {
        if(j==-1)a[i]=buf[i];
        else 
        {
            b[j]=buf[i];
            j++;
        }
        if(buf[i]==':')
        {
            a[i]='\0';
            j=0;
        }
    }
    if(j==-1)
    {
        sendto(us->id,"发送格式错误,发送对象:发送内容",strlen("发送格式错误,发送对象:发送内容")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
        return -1;
    }
    US *fs = Print_User(a);
    if(fs==NULL)
    {
        sendto(us->id,"该用户不在线",strlen("该用户不在线")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
        return -1;
    }
    char c[2048];
    memset(c,0,sizeof(c));
    sprintf(c,"%s给%s了发消息:%s",us->name,fs->name,b);
    sendto(fs->id,c,strlen(c)+1,0,(struct sockaddr*)&fs->cAddr,sizeof(fs->cAddr));
    sendto(us->id,"发送成功",strlen("发送成功")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
    return 0;
}

//查找文件
void My_Ls(US *us)
{
    char buf[2024]={0};
    DIR *dir = opendir(S_PATH);
    struct dirent *dirp =NULL;
    int i=0;

while(dirp = readdir(dir))
    {
        if(i<2)
        {
            i++;
            continue;
        }
        int k=0;
        int len = strlen(dirp->d_name);
        int b_len = 4+4+4+1+len+1;
        //包头
        buf[k++] = 0xAB;
        //包长
        buf[k++] = b_len & 0xff;
        buf[k++] = (b_len>>8) &0xff;
        buf[k++] = (b_len>>16) & 0xff;
        buf[k++] = (b_len>>24) &0xff;
        //命令
        buf[k++] = FTP_CMD_LS & 0xff;
        buf[k++] = (FTP_CMD_LS>>8) &0xff;
        buf[k++] = (FTP_CMD_LS>>16) & 0xff;
        buf[k++] = (FTP_CMD_LS>>24) &0xff;
        //内容长度
        buf[k++] = len & 0xff;
        buf[k++] = (len>>8) &0xff;
        buf[k++] = (len>>16) & 0xff;
        buf[k++] = (len>>24) &0xff;
        //发送内容
        for(int j=0;j<len;j++)
        {
            buf[k++]=dirp->d_name[j];
        }
        buf[k++]='\0';
        //包尾
        buf[k++] = 0xAB;
        sendto(us->id,buf,k,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
    }
    closedir(dir);
}

//查找文件是否存在
int Exist_File(char *file)
{
    char buf[2024]={0};
    DIR *dir = opendir(S_PATH);
    struct dirent *dirp =NULL;
    int i=0;
    while(dirp = readdir(dir))
    {
        if(i<2)
        {
            i++;
            continue;
        }
        if(My_Strcmp(dirp->d_name,file))
        {
            closedir(dir);
            return 1;
        }
    }
    closedir(dir);
    return 0;
}

//下载文件
void My_Get(US *us,char *file,int sk)
{
    if(Exist_File(file))
    {
        unsigned char  buf[2024]={0};
        char path[520]={0};
        sprintf(path,"%s/%s",S_PATH,file);
        FILE *fd = fopen(path,"r");
        fseek(fd,0,SEEK_END);
        int len = ftell(fd)-sk;
        fseek(fd,sk,SEEK_SET);
        int yn= 0;
        if(len>1000)
        {
            yn=1;
            len=1000;
        }
        int k=0;
        int fn_len = strlen(file);
        //包长+命令+内容长度+是否发完+是否first+文件名长度+文件名+发送内容+文件结尾+包尾
        int b_len = 4+4+4+1+1+4+fn_len+len+1+1;
        //包头
        buf[k++] = 0xAB;
        //包长
        buf[k++] = b_len & 0xff;
        buf[k++] = (b_len>>8) &0xff;
        buf[k++] = (b_len>>16) & 0xff;
        buf[k++] = (b_len>>24) &0xff;
        //命令
        buf[k++] = FTP_CMD_GET & 0xff;
        buf[k++] = (FTP_CMD_GET>>8) &0xff;
        buf[k++] = (FTP_CMD_GET>>16) & 0xff;
        buf[k++] = (FTP_CMD_GET>>24) &0xff;
        //内容长度
        buf[k++] = len & 0xff;
        buf[k++] = (len>>8) &0xff;
        buf[k++] = (len>>16) & 0xff;
        buf[k++] = (len>>24) &0xff;
        //是否发完
        if(yn)buf[k++] = 'N';
        else buf[k++] = 'Y';
        //是否first
        if(sk==0)buf[k++]='Y';
        else buf[k++]='N';
        //文件名长度
        buf[k++] = fn_len & 0xff;
        buf[k++] = (fn_len>>8) &0xff;
        buf[k++] = (fn_len>>16) & 0xff;
        buf[k++] = (fn_len>>24) &0xff;
        //文件名
        for(int j=0;j<fn_len;j++)
        {
            buf[k++]=file[j];
        }
        
        char ch[len];
        fread(ch,len,1,fd);
        //发送内容
        for(int j=0;j<len;j++)
        {
            buf[k++]=ch[j];
        }
        buf[k++]='\0';
        //包尾
        buf[k++] = 0xAB;
        sendto(us->id,buf,k,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
        fclose(fd);
        if(yn)
        {
            My_Get(us,file,sk+len);
        }
    }
    else 
    {
        sendto(us->id,"该文件不存在",strlen("该文件不存在")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
    }
}

//上传文件
void My_Put(US *us,char *file)
{   
    unsigned char buf[2024]={0};
    int k=0;
    int len = sizeof(file);
    int b_len = 4+4+4+1+len+1;
    //包头
    buf[k++] = 0xAB;
    //包长
    buf[k++] = b_len & 0xff;
    buf[k++] = (b_len>>8) &0xff;
    buf[k++] = (b_len>>16) & 0xff;
    buf[k++] = (b_len>>24) &0xff;
    //命令
    buf[k++] = FTP_CMD_PUT & 0xff;
    buf[k++] = (FTP_CMD_PUT>>8) &0xff;
    buf[k++] = (FTP_CMD_PUT>>16) & 0xff;
    buf[k++] = (FTP_CMD_PUT>>24) &0xff;
    //内容长度
    buf[k++] = len & 0xff;
    buf[k++] = (len>>8) &0xff;
    buf[k++] = (len>>16) & 0xff;
    buf[k++] = (len>>24) &0xff;
    //内容
    for(int i=0;i<len;i++)
    {
        buf[k++]=file[i];
    }
    buf[k++] = '\0';
    //包尾
    buf[k++] = 0xAB;
    //告诉客户端还可以传文件了
    sendto(us->id,buf,k,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
    while(1)
    {
        memset(buf,0,sizeof(buf));
        int i=0;
        unsigned char ch;
        int ret = read(us->id,&ch,1);
        if(ret <= 0)
        {
            perror("文件包接收出错");
            return ;
        }
        if(ch==0xAB)
        {
            do
            {
                int ret = read(us->id,&ch,1);
                if(ret <= 0)
                {
                    perror("文件包接收出错");
                    return ;
                }
                buf[i++]=ch;
            }while(ch != 0xAB);
            int pack_len = buf[0]|(buf[1]<<8)|(buf[2]<<16)|(buf[3]<<24);
            int cmd_no = buf[4]|(buf[5]<<8)|(buf[6]<<16)|(buf[7]<<24);
            int lens = buf[8]|(buf[9]<<8)|(buf[10]<<16)|(buf[11]<<24);
            if(cmd_no==FTP_CMD_PUT)
            {
                char yn = buf[12];
                char ones = buf[13];
                int fn_len = buf[14]|(buf[15]<<8)|(buf[16]<<16)|(buf[17]<<24);
                unsigned char file[fn_len+1];
                for(int j=0;j<fn_len;j++)
                {
                    file[j]=buf[18+j];
                }
                file[fn_len]='\0';
                unsigned char path[2024];
                sprintf(path,"%s/%s",S_PATH,file);
                FILE *fd = NULL;
                if(ones=='Y')fd = fopen(path,"w");
                else fd = fopen(path,"a");
                fwrite(&buf[18+fn_len],lens,1,fd);
                fclose(fd);
                if(yn=='Y')
                {
                    sendto(us->id,"上传成功!",strlen("上传成功!")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
                    break;
                }
            }
        }
        else 
        {
            sendto(us->id,"上传失败!",strlen("上传失败!")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
            break;
        }
    }
}

//用户下线
void My_Bye(US *us)
{
    char buf[2024]={0};
    int k=0;
    int len = 0;
    int b_len = 4+4+4+1+len+1;
    //包头
    buf[k++] = 0xAB;
    //包长
    buf[k++] = b_len & 0xff;
    buf[k++] = (b_len>>8) &0xff;
    buf[k++] = (b_len>>16) & 0xff;
    buf[k++] = (b_len>>24) &0xff;
    //命令
    buf[k++] = FTP_CMD_BYE & 0xff;
    buf[k++] = (FTP_CMD_BYE>>8) &0xff;
    buf[k++] = (FTP_CMD_BYE>>16) & 0xff;
    buf[k++] = (FTP_CMD_BYE>>24) &0xff;
    //内容长度
    buf[k++] = len & 0xff;
    buf[k++] = (len>>8) &0xff;
    buf[k++] = (len>>16) & 0xff;
    buf[k++] = (len>>24) &0xff;
    //包尾
    buf[k++] = 0xAB;
    //告诉客户端还有传文件了
    sendto(us->id,buf,k,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
}

//接收信息
int Receive_informatione(US *us)
{
    char buf[512]={0};
    int ret = recvfrom(us->id,buf,sizeof(buf)-1,0,(struct sockaddr*)&us->cAddr,&us->len);
    if(ret == -1)
    {
        printf("与%s通信失败\n",us->name);
        return -1;
    }
    else if(ret>0)
    {
        if(My_Strcmp(buf,"ls"))//查找文件
        {
            puts("ls");
            My_Ls(us);
        }
        else if(My_Strcmp(buf,"get"))//下载文件
        {
            puts("get");
            char file[520]={0};
            sendto(us->id,"输入需要下载的文件名:",strlen("输入需要下载的文件名:")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
            ret = recvfrom(us->id,file,sizeof(file)-1,0,(struct sockaddr*)&us->cAddr,&us->len);
            if(ret == -1)
            {
                printf("与%s通信失败\n",us->name);
                return -1;
            }
            My_Get(us,file,0);
        }
        else if(My_Strcmp(buf,"put"))//上传文件
        {
            puts("put");
            char file[520]={0};
            sendto(us->id,"输入需要上传的文件名:",strlen("输入需要上传的文件名:")+1,0,(struct sockaddr*)&us->cAddr,sizeof(us->cAddr));
            ret = recvfrom(us->id,file,sizeof(file)-1,0,(struct sockaddr*)&us->cAddr,&us->len);
            if(ret == -1)
            {
                printf("与%s通信失败\n",us->name);
                return -1;
            }
            My_Put(us,file);
        }
        else if(My_Strcmp(buf,"bye"))//用户手动下线
        {
            puts("bye");
            us->online=0;
            memset(us->name,0,sizeof(us->name));
            My_Bye(us);
        }
        else //聊天
        {
            My_Chat(us,buf);
        }
    }
    else if(ret == 0)
    {
        printf("客户端%s:%d,退出\n",inet_ntoa(us->cAddr.sin_addr),ntohs(us->cAddr.sin_port));
        Del(us);
        return -1;
    }
}

//用户连接
void Connect(int server)
{
    US *client=(US *)malloc(sizeof(US));
    client->len=sizeof(client->cAddr);
    client->next=NULL;
    client->prev=NULL;
    client->online=0;
    //通过client端的连接请求
    client->id = accept(server,(struct sockaddr*)&client->cAddr,&client->len);
    if(client->id == -1)
    {
        puts("连接失败");
        free(client);
    }
    printf("New Client %s:%d connect!\n",inet_ntoa(client->cAddr.sin_addr),ntohs(client->cAddr.sin_port));
    memset(client->name,0,sizeof(client->name));
    Add(client);
}

//运行
int run(char *ip,char *port)
{
    struct timeval tm;
    fd_set rfds;
    int ret =0;
    int server = Create_Tcpsocket(ip,port);
    if(server==-1)
    {
        puts("创建套接字失败");
        return -1;
    }
    //保存client端
    printf("等待TCP客户端的连接中.......!\n");
    
    //等待client端连接
    while(1)
    {
        //清空集合
        FD_ZERO(&rfds);
        //添加文件描述符进集合
        US *i=head;
        FD_SET(server,&rfds);
        int mx = server;
        while(i)
        {
            FD_SET(i->id,&rfds);
            mx=MAX(mx,i->id);
            i=i->next;
        }
        ret = select(mx+1,&rfds,NULL,NULL,0);
        if(ret == -1)
        {
            perror("select 创建失败");
            break;
        }
        else if(ret == 0)
        {
            continue;
        }
        if(FD_ISSET(server,&rfds))
        {
            Connect(server);
        }
        i=head;
        for(i=head;i;i=i->next)
        {
            if(FD_ISSET(i->id,&rfds))
            {
                if(!i->online)
                {
                    if(Login_Create(i)==-1)continue;
                    else i->online=1;
                }
                else 
                {
                    if(Receive_informatione(i)==-1)continue;
                }
            }
        }
    }
}
```
###tool.h
```cpp
#ifndef __TOOL_H__
#define __TOOL_H__

#define S_PATH "/xxx/.../s_dir"

#define MAX(a,b) ((a)>(b)?(a):(b))

typedef struct users 
{
    struct sockaddr_in cAddr;
    socklen_t len;
    int id;
    char name[512];
    struct users* next;
    struct users* prev;
    int online;
}US;

/*
    Create_Tcpsocket:创建socket套接字
        @ip     网络地址
        @port   端口号
        @retval 成功返回socket套接字 失败返回-1
*/
int Create_Tcpsocket(char *ip, char *port);
/*
    Add:把客户端信息插入链表保存
*/
US *Add(US *data);
/*
    My_Strlen:判断字符串a,b是否相等
*/
int My_Strcmp(const char *a,const char *b);

//删除断开用户信息
void Del(US *p);

//登录
int Login_Create(US *us);

//聊天
int My_Chat(US *us,char *buf);

//查找文件
void My_Ls(US *us);

//查找文件是否存在
int Exist_File(char *file);

//下载文件
void My_Get(US *us,char *file,int sk);

//上传文件
void My_Put(US *us,char *file);

//用户下线
void My_Bye(US *us);

//接收信息
int Receive_informatione(US *us);

//用户连接
void Connect(int server);

//运行
int run(char *ip,char *port);

#endif
```

你觉得文章怎么样

已浏览 76 次，0 条评论