多客户端连接服务器传输文件

1、/*可以实现文件的传送功能，且可以选择存放的路径，若该路径有相同的文件名则询问*是否需要更改存储的文件名，如果需要则重新命名，否则退出。*20110908*/*外加实现多线程，即多个客户端能同时传送文件给服务器*2011-09-19*/服务器#include#include#include#include#include#include#include#include#include#include#include#include#include#include#define LISTENQ20#define MAXLINE40#define BUFFERSIZE4096#define POR

2、T3336/*将buf中的字节内容写入文件描述符fd*/ssize_t writen(int fd, constvoid * vptr, size_t n)size_tnleft;ssize_t nwritten;constchar * ptr;ptr = vptr;nleft = n;while ( nleft 0) /start to writeif (nwritten = write(fd, ptr, nleft) = 0)if (nwritten 0 & errno = EINTR) /interupt errornwritten = 0;elsereturn -1;nleft -=

3、nwritten;ptr += nwritten; /continue to write from the leftreturn n;ssize_t readline(int fd, void * vptr, size_t maxlen)ssize_t n, rc;char c, *ptr;ptr = vptr;for (n = 1; n maxlen; n+)again:if (rc = read(fd, &c, 1) = 1)*ptr+ = c;if (c = n)break;elseif (rc = 0) *ptr = 0;return (n - 1);elseif (errno = E

4、INTR)goto again;return (-1);*ptr = 0;return (n);int Socket(int domain, int type, int protocol)int sockfd;if (sockfd = socket(domain, type, protocol) 0)fprintf(stderr, socket errorn);exit(1);return sockfd;int Accept(int sockfd, struct sockaddr * addr, socklen_t * addrlen)int ret;if (ret = accept(sock

5、fd, addr, addrlen) 0)fprintf(stderr, accept errorn);exit(1);return ret;int Bind(int sockfd, conststruct sockaddr *addr, socklen_t addrlen)int ret;if (ret = bind(sockfd, addr, addrlen) 0)fprintf(stderr, bind errorn);exit(1);return ret;int Listen(int sockfd, int backlog)int ret;if (ret = listen(sockfd

6、, backlog) 0)fprintf(stderr, listen errorn);exit(1);return ret;int Close(int fd)int ret;if (ret = close(fd) 0)fprintf(stderr, close errorn);exit(1);return ret;void *create_thread(int connfd)charbuffBUFFERSIZE + 1;charfilenameBUFFERSIZE + 1;charcdBUFFERSIZE+1;intfilefd; /* file descriptor */intcount;

7、DIR*dir;struct dirent*ptr;charchoose10;printf(readinging filename.n);if (readline(connfd, buff, BUFFERSIZE) 0)fprintf(stderr, readline errorn);exit(1);buffstrlen(buff) - 1 = 0; /* change n to NUL */memcpy(filename, buff, BUFFERSIZE + 1);printf(will save to file: %sn, buff);/change the direct to stor

8、e the fileprintf(Input the direct you want to store %s:n, buff);scanf(%s, cd);if(chdir(cd) d_name) = 0)printf(Already have the %sn, buff);printf(If you want to rename, please input yes, or you can input non);scanf(%s, choose);if(strcmp(choose, yes) = 0)printf(You can rename:t);scanf(%s, buff);elsepr

9、intf(Already have the file, now exit!n);exit (1);filefd = open(buff, O_WRONLY | O_CREAT);if (filefd 0)fprintf(stderr, cant open the file: %sn, buff);exit(1);printf(which thread is %dn,connfd);while(count = read(connfd, buff, BUFFERSIZE)if (count 0)fprintf(stderr, connfd read errorn);exit(1);if (writ

10、en(filefd, buff, count) 0) fprintf(stderr, writing to filefd errorn);exit(1);closedir(dir);Close(filefd);Close(connfd);printf(file %s received!n, filename);int main()intlistenfd, connfdFD_MAX,i = 0,j;charbuffBUFFERSIZE + 1;charfilenameBUFFERSIZE + 1;charcdBUFFERSIZE+1;charchoose10;struct sockaddr_in

11、servaddr, cliaddr;intcliaddrlen;intfilefd; /* file descriptor */DIR*dir;struct dirent*ptr;listenfd = Socket(AF_INET, SOCK_STREAM, 0);bzero(&servaddr, sizeof(servaddr);servaddr.sin_family = AF_INET;servaddr.sin_addr.s_addr = INADDR_ANY;servaddr.sin_port = htons(PORT);Bind(listenfd, (struct sockaddr *

12、)&servaddr, sizeof(servaddr);Listen(listenfd, LISTENQ);pthread_tchild_thread;pthread_attr_tchild_thread_attr;pthread_attr_setdetachstate(&child_thread_attr,PTHREAD_CREATE_DETACHED);/while(1)printf(listening.n);cliaddrlen = sizeof(cliaddr);printf(i is %d n, i);connfdi = Accept(listenfd, (struct socka

13、ddr *)&cliaddr, &cliaddrlen);if(pthread_create(&child_thread,&child_thread_attr,create_thread,(void *)connfdi)0)printf(pthread_create Failed:%sn,strerror(errno);i+;Close(listenfd);pthread_attr_destroy(&child_thread_attr); pthread_exit (NULL);return 0;/客户端#include#include#include#include#include#incl

14、ude#include#include#include#include#include#include#define LISTENQ 20#define MAXLINE 40#define BUFFERSIZE 4096#define PORT 3336ssize_t writen(int fd, constvoid * vptr, size_t n)size_t nleft;ssize_t nwritten;constchar * ptr;ptr = vptr;nleft = n;while (nleft 0) if (nwritten = write(fd, ptr, nleft) = 0

15、) if (nwritten 0 & errno = EINTR) nwritten = 0;elsereturn -1;nleft -= nwritten;ptr += nwritten;return n;int Socket(int domain, int type, int protocol)int sockfd;if (sockfd = socket(domain, type, protocol) 0) fprintf(stderr, socket errorn);exit(1);return sockfd;int Connect(int sockfd, conststruct soc

16、kaddr *addr, socklen_t addrlen)int ret;if (ret = connect(sockfd, addr, addrlen) 0) fprintf(stderr, connect errorn);exit(1);return ret;int Close(int fd)int ret;if (ret = close(fd) 0) fprintf(stderr, close errorn);exit(1);return ret;int main(int argc, char *argv)if (argc != 3) fprintf(stderr, Usage: .

17、/fileclient n);exit(1);int sockfd;char buffBUFFERSIZE + 1;char filenameheaderBUFFERSIZE + 1;struct sockaddr_in servaddr;int filefd; /* file descriptor */int count;sockfd = Socket(AF_INET, SOCK_STREAM, 0);bzero(&servaddr, sizeof(servaddr);servaddr.sin_family = AF_INET;servaddr.sin_addr.s_addr=INADDR_

18、ANY;servaddr.sin_port = htons(PORT);printf(connecting.n);Connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr);printf(transferring file name: %s.n, argv1);memcpy(filenameheader, argv1, strlen(argv1);filenameheaderstrlen(argv1) = n;filenameheaderstrlen(argv1) + 1 = 0;writen(sockfd, filenameheader, strlen(filenameheader);printf(will transfer file: %sn, argv1);filefd = open(argv1, O_RDONLY);if (filefd 0) fprintf(stderr, cant open the file