c++ 文件传输项目源码(含socket服务器端以及客户端)

#include "ServerSocket.h" #include <iostream> using namespace std; #include <sys/socket.h> #include <sys/types.h> #include <arpa/inet.h> #include <netinet/in.h> #include <sys/epoll.h> #include <assert.h> #include <string.h> const int maxEvents = 100; ServerSocket::ServerSocket(char *ip, short port) { listenFd = socket(AF_INET, SOCK_STREAM, 0); assert(-1 != listenFd); struct sockaddr_in ser; memset(&ser, 0, sizeof(ser)); ser.sin_family = AF_INET; ser.sin_port = htons(port); ser.sin_addr.s_addr = inet_addr(ip); int res = bind(listenFd, (struct sockaddr*)&ser, sizeof(ser)); assert(-1 != res); listen(listenFd, 5); epollFd = epoll_create(5); assert(-1 != epollFd); struct epoll_event event; event.events = EPOLLIN; event.data.fd = listenFd; epoll_ctl(epollFd, EPOLL_CTL_ADD, listenFd, &event); } ServerSocket::~ServerSocket() { close(listenFd); close(epollFd); } void ServerSocket::Start() { while(1) { struct epoll_event orderEvents[maxEvents]; int n = epoll_wait(epollFd, orderEvents, maxEvents, -1); int i = 0; for(; i < n; ++i) { int fd = orderEvents[i].data.fd; if(fd == listenFd) { AcceptClientLink(); } else { if(orderEvents[i].events & EPOLLRDHUP) { close(fd); epoll_ctl(epollFd, EPOLL_CTL_DEL, fd, NULL); } else if(orderEvents[i].events & EPOLLIN) { RecvData(fd); } else { cout<<"event is error"<<endl; } } } } } void ServerSocket::AcceptClientLink() { struct sockaddr_in cli; socklen_t len = (socklen_t)sizeof(cli); int c = accept(listenFd, (struct sockaddr*)&cli, &len); if(-1 == c) { cout<<"one client link error"<<endl; return; } struct epoll_event event; event.events = EPOLLIN | EPOLLRDHUP; event.data.fd = c; epoll_ctl(epollFd, EPOLL_CTL_ADD, c, &event); } void ServerSocket::RecvData(int fd) { Data data; // type command memset(&data, 0, sizeof(data)); int n = recv(fd, &data, sizeof(data), 0); if(n <= 0) { return; } switch(data.type) { case 1: CarryOutCmd(fd, data.command); break; case 2: RecvFile(fd, data.command); break; case 3: SendFile(fd, data.command); break; default: send(fd, "type error", 10, 0); break; } } void ServerSocket::CmdCd(char *path) { chdir(path); } // 服务器执行系统命令，将执行结果发送给客户端 void ServerSocket::CarryOutCmd(int fd, char *cmd) { char *argv[128] = {0}; char *p = strtok(cmd, " "); int count = 0; while(p != NULL) { argv[count++] = p; p = strtok(NULL, " "); } if(strncmp(argv[0], "cd", 2) == 0) { CmdCd(argv[1]); char path[128] = {0}; getcwd(path, 127); send(fd, path, strlen(path), 0); return; } int fds[2] = {-1,-1}; pipe(fds); pid_t pid = fork(); assert(-1 != pid); if(pid == 0) { close(1); close(2); dup(fds[1]); dup(fds[1]); char path[128] = "/bin/"; strcat(path, argv[0]); execv(path, argv); } else { close(fds[1]); int flag = 0; while(1) { char buff[128] = {0}; int n = read(fds[0], buff, 127); if(n <= 0) { break; } flag = 1; send(fd, buff, n, 0); } if(!flag) send(fd, "success", strlen("success"), 0); close(fds[0]); } return; }