最近在寫Linux網絡方面的demo,用TCP實現了一個簡單的傳輸文件程序，適用於網卡設備及TCP/IP協議棧及網絡環境測試時使用。用法容易，代碼簡潔，已經測試成功。其功能是使用TCP協議從client端向server端傳輸文件，用法如下:
編譯：
client:gcc -o client client.c
server:gcc -o server server.c
運行：
client端：./client <server IP> <端口號> <上傳文件名>
server端：./server <端口號> <保存為文件名>
其中，server端先運行，client端與server端的端口號必須一致並且不能與已知端口沖突（例如8888即可）
下面將代碼貼上：（使用UDP傳輸文件請參閱Linux網絡編程之使用UDP傳輸文件 【 http://www.linuxidc.com/Linux/2013-06/85843.htm】）
server端代碼:server.c
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define MAXLINE 1024
void usage(char *command)
{
printf("usage :%s portnum filename\n", command);
exit(0);
}
int main(int argc,char **argv)
{
struct sockaddr_in serv_addr;
struct sockaddr_in clie_addr;
char buf[MAXLINE];
int sock_id;
int link_id;
int recv_len;
int write_leng;
int clie_addr_len;
FILE *fp;
if (argc != 3) {
usage(argv[0]);
}
if ((fp = fopen(argv[2], "w")) == NULL) {
perror("Open file failed\n");
exit(0);
}
if ((sock_id = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
perror("Create socket failed\n");
exit(0);
}
/*fill the server sockaddr_in struct commented by guoqingbo*/
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(atoi(argv[1]));
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(sock_id, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0 ) {
perror("Bind socket failed\n");
exit(0);
}
if (-1 == listen(sock_id, 10)) {
perror("Listen socket failed\n");
exit(0);
}
/* server part commented by guoqingbo*/
while (1) {
clie_addr_len = sizeof(clie_addr);
link_id = accept(sock_id, (struct sockaddr *)&clie_addr, &clie_addr_len);
if (-1 == link_id) {
perror("Accept socket failed\n");
exit(0);
}
bzero(buf, MAXLINE);
while (recv_len = recv(link_id, buf, MAXLINE, 0)) {
/* receiver data part commented by guoqingbo*/
if(recv_len < 0) {
printf("Recieve Data From Server Failed!\n");
break;
}
printf("#");
write_leng = fwrite(buf, sizeof(char), recv_len, fp);
if (write_leng < recv_len) {
printf("Write file failed\n");
break;
}
bzero(buf,MAXLINE);
}
printf("\nFinish Recieve\n");
fclose(fp);
close(link_id);
}
close(sock_id);
return 0;
}
