实现

client.c

@@ -0,0 +1,88 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <errno.h>
+#include <string.h>
+#include <netdb.h>
+#include <sys/types.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+#include <arpa/inet.h>
+#include <sys/stat.h>
+#include <pthread.h>
+
+int sockfd;
+char buf[BUFSIZ];
+char bufr[BUFSIZ];
+struct sockaddr_in their_addr;
+pthread_t thread;
+struct sf_hdtr hdtr;
+uint32_t file_size;
+int recv_bin = 0;
+
+void getMessage(void *p) {
+    int c = 0;
+    while((c = recv(sockfd, bufr, BUFSIZ, 0)) > 0) {
+        printf("Recv %d bytes: ", c);
+        if(recv_bin) {
+            FILE* fp = fopen("dump.bin", "w+");
+            fwrite(bufr, c, 1, fp);
+            fclose(fp);
+        } else for(int i = 0; i < c; i++) putchar(bufr[i]);
+        putchar('\n');
+    }
+}
+
+off_t fileSize(const char* fname) {
+    struct stat statbuf;
+    if(stat(fname, &statbuf)==0) return statbuf.st_size;
+    else return -1;
+}
+
+int main(int argc,char *argv[]) {   //usage: ./client host port
+    ssize_t numbytes;
+    puts("break!");
+    while((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1);
+    puts("Get sockfd");
+    their_addr.sin_family = AF_INET;
+    their_addr.sin_port = htons(atoi(argv[2]));
+    their_addr.sin_addr.s_addr=inet_addr(argv[1]);
+    bzero(&(their_addr.sin_zero), 8);
+    
+    while(connect(sockfd,(struct sockaddr*)&their_addr,sizeof(struct sockaddr)) == -1);
+    puts("Connected to server");
+    numbytes = recv(sockfd, buf, BUFSIZ,0);
+    buf[numbytes]='\0';  
+    puts(buf);
+    if(!pthread_create(&thread, NULL, (void*)&getMessage, NULL)) {
+        puts("Thread create succeeded");
+        while(1) {
+            printf("Enter command:");
+            scanf("%s",buf);
+            if(!strcmp(buf, "bin")) recv_bin = !recv_bin;
+            else if(!strcmp(buf, "file")) {
+                printf("Enter file path:");
+                scanf("%s",buf);
+                printf("Open:");
+                puts(buf);
+                FILE *fp = NULL;
+                fp = fopen(buf, "rb");
+                if(fp) {
+                    off_t len;
+                    struct iovec headers;
+                    headers.iov_base = &file_size;
+                    headers.iov_len = sizeof(uint32_t);
+                    hdtr.headers = &headers;
+                    hdtr.hdr_cnt = 1;
+                    file_size = (uint32_t)fileSize(buf);
+                    if(!sendfile(fileno(fp), sockfd, 0, &len, &hdtr, 0)) puts("Send file success.");
+                    else puts("Send file error.");
+                    printf("Send count:%lld\n", len);
+                    fclose(fp);
+                } else puts("Open file error!");
+            } else send(sockfd, buf, strlen(buf), 0);
+            sleep(1);
+        }
+    } else perror("Create msg thread failed");
+    close(sockfd);
+    return 0;
+}