ftp服务器的简单实现（二）——客户端-优快云博客

客户端：

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>

#define TRUE 1
#define dataLen 1024
char user_cmd[10], cmd_arg[20];
char buf[dataLen];

void cmd_pwd(int sock, int sockmsg);
void cmd_dir(int sock, int sockmsg);
void cmd_cd(int sock, int sockmsg, char *dirName);
void cmd_cdback(int sock, int sockmsg);
void cmd_help(int sock, int sockmsg);
void cmd_get(int sock, int sockmsg, char *fileName);
void cmd_put(int sock, int sockmsg, char *fileName);
void cmd_quit(int sock, int sockmsg);

int main(int argc, char *argv[])
{
	int cmd_len, arg_len;
	int sock, sockmsg;
	struct sockaddr_in server, servermsg;
	struct hostent *hp;

	sock = socket(AF_INET, SOCK_STREAM, 0);
	sockmsg = socket(AF_INET, SOCK_STREAM, 0);
	
	if(sock < 0 || sockmsg < 0)
	{
		perror("opening stream socket");
		exit(1);
	}

	printf("argc is %d, argv[1] is %s\n",argc, argv[1]);

	hp = gethostbyname(argv[1]);
	//hp = gethostbyname("localhost");
	if(hp == 0)
	{
		fprintf(stderr, "%s:unknown host\n", argv[1]);
		exit(2);
	}

	server.sin_family = AF_INET;
	server.sin_port = htons(atoi(argv[2]));
	memcpy((char*)&server.sin_addr, (char*)hp->h_addr, hp->h_length);
	
	servermsg.sin_family = AF_INET;
	servermsg.sin_port = htons(atoi(argv[2]) +1);
	memcpy((char*)&servermsg.sin_addr, (char*)hp->h_addr, hp->h_length);

	if(connect(sock, (struct sockaddr *)&server, sizeof(server)) < 0 ||
		connect(sockmsg, (struct sockaddr *)&servermsg, sizeof(servermsg)) < 0)
	{
		perror("connecting stream socket");
		exit(1);
	}

	while(TRUE)
	{
		memset(user_cmd, 0, 10);
		memset(cmd_arg, 0, 20);

		//First get the input character by user
		printf("command: ");
		scanf("%s", user_cmd);

		//Then get the command charater and argument
		if(strcmp(user_cmd, "quit") == 0)
		{
			cmd_quit(sock, sockmsg);
			close(sockmsg);
			close(sock);
			printf("connection closed\n\n");
			exit(0);
		}
		else if(strcmp(user_cmd, "?") == 0)
		{
			cmd_help(sock, sockmsg);
		}
		else if(strcmp(user_cmd, "pwd") == 0)
		{
			printf("stcmp pwd --------------->>>\n");
			cmd_pwd(sock, sockmsg);
		}
		else if(strcmp(user_cmd, "dir") == 0)
		{
			cmd_dir(sock, sockmsg);
		}
		else if(strcmp(user_cmd, "cd") == 0)
		{
			scanf("%s", cmd_arg);
			cmd_cd(sock, sockmsg, cmd_arg);
		}
		else if(strcmp(user_cmd, "cd..") == 0)
		{
			cmd_cdback(sock, sockmsg);
		}
		else if(strcmp(user_cmd, "get") == 0 )
		{
			scanf("%s", cmd_arg);
			cmd_get(sock, sockmsg, cmd_arg);
		}
		else if(strcmp(user_cmd, "put") == 0)
		{
			scanf("%s", cmd_arg);
			cmd_put(sock, sockmsg, cmd_arg);
		}
		else
		{
			printf("bad command!\n");
		}
	}
}


void cmd_pwd(int sock, int sockmsg)
{
	//int numRead
	char dirName[30];
	write(sockmsg, user_cmd, sizeof(user_cmd));
	read(sock, dirName, 30);
	printf("dirName is ------->>  %s\n", dirName);
}

//command dir
void cmd_dir(int sock, int sockmsg)
{
	int i, fileNum = 0;
	char fileInfo[50];

	write(sockmsg, user_cmd, sizeof(user_cmd));
	read(sock, &fileNum, sizeof(int));

	printf("---------------------------------------\n");
	printf("file number : %d\n", fileNum);
	if(fileNum > 0)
	{
		for(i=0; i<fileNum; i++)
		{
			memset(fileInfo, 0, sizeof(fileInfo));
			read(sock, fileInfo, sizeof(fileInfo));
			printf("%s\n", fileInfo);
		}
		printf("-----------------------------------\n");
	}
	else if(fileNum == 0)
	{
		printf("directory of server point is empty.\n");
		return;
	}
	else
	{
		printf("error in command 'dir'\n");
		return;
	}
} 

void cmd_cd(int sock, int sockmsg, char *dirName)
{
	char currentDirPath[200];
	write(sockmsg, user_cmd, sizeof(user_cmd));
	write(sockmsg, cmd_arg, sizeof(cmd_arg));
	printf("172 ----------->>\n");
	read(sock, currentDirPath, sizeof(currentDirPath));
	printf("174 ------------>>\n");
	printf("now is directory : %s\n", currentDirPath);
}

//command cd..
void cmd_cdback(int sock, int sockmsg)
{
	write(sockmsg, user_cmd, sizeof(user_cmd));
}

//command quit
void cmd_quit(int sock, int sockmsg)
{
	write(sockmsg, user_cmd, sizeof(user_cmd));
}

//command help
void cmd_help(int sock, int sockmsg)
{
	char help[300];
	write(sockmsg, user_cmd, sizeof(user_cmd));
	read(sock, help, 300);

	printf("%s\n", help);
}

//command get
void cmd_get(int sock, int sockmsg, char *fileName)
{
	int fd;
	long fileSize;
	char localFilePath[200];

	write(sockmsg, user_cmd, sizeof(user_cmd));
	write(sockmsg, cmd_arg, sizeof(cmd_arg));
	printf("%s\n%s\n", user_cmd, cmd_arg);

	memset(localFilePath, 0, sizeof(localFilePath));
	getcwd(localFilePath, sizeof(localFilePath));
	strcat(localFilePath, "/");
	strcat(localFilePath, fileName);

	fd = open(localFilePath, O_RDWR|O_CREAT, S_IREAD|S_IWRITE);
	if(fd!=-1)
	{
		memset(buf, 0, dataLen);
		read(sock, &fileSize, sizeof(long));
		while(fileSize > dataLen)
		{
			read(sock, buf, dataLen);
			write(fd, buf, dataLen);
			fileSize = fileSize - dataLen;
		}

		read(sock, buf, fileSize);
		write(fd, buf, fileSize);
		close(fd);
		printf("download completed\n");
	}
	else
	{
		printf("open file %s failed\n", localFilePath);
	}
}

//command put
void cmd_put(int sock, int sockmsg, char *fileName)
{
	write(sockmsg, user_cmd, sizeof(user_cmd));
	write(sockmsg, cmd_arg, sizeof(cmd_arg));

	int fd;
	long fileSize;
	int numRead;
	char filePath[200];

	struct stat fileSta;
	
	memset(filePath, 0, sizeof(filePath));
	getcwd(filePath, sizeof(filePath));

	strcat(filePath, "/");
	strcat(filePath, fileName);

	fd = open(filePath, O_RDONLY, S_IREAD);
	if(fd != -1)
	{
		fstat(fd, &fileSta);
		fileSize = (long)fileSta.st_size;

		write(sock, &fileSize, sizeof(long));
		memset(buf, 0, dataLen);
		while(fileSize > dataLen)
		{
			read(fd, buf, dataLen);
			write(sock, buf, dataLen);
			fileSize = fileSize -dataLen;
		}

		read(fd, buf, fileSize);
		write(sock, buf, fileSize);
		close(fd);
		printf("upload completed\n");
	}
	else
	{
		printf("open file %s failed\n", filePath);
	}
}

编译的话，用gcc -g -o client client.c

运行的话，需要注意，需要两个参数：

./client localhost 3499

一个是本地主机，一个是端口号

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>

#define TRUE 1
#define dataLen 1024
char user_cmd[10], cmd_arg[20];
char buf[dataLen];

void cmd_pwd(int sock, int sockmsg);
void cmd_dir(int sock, int sockmsg);
void cmd_cd(int sock, int sockmsg, char *dirName);
void cmd_cdback(int sock, int sockmsg);
void cmd_help(int sock, int sockmsg);
void cmd_get(int sock, int sockmsg, char *fileName);
void cmd_put(int sock, int sockmsg, char *fileName);
void cmd_quit(int sock, int sockmsg);

int main(int argc, char *argv[])
{
   int cmd_len, arg_len;
   int sock, sockmsg;
   struct sockaddr_in server, servermsg;
   struct hostent *hp;

   sock = socket(AF_INET, SOCK_STREAM, 0);
   sockmsg = socket(AF_INET, SOCK_STREAM, 0);

   if(sock < 0 || sockmsg < 0)
   {
       perror("opening stream socket");
       exit(1);
   }

   printf("argc is %d, argv[1] is %s\n",argc, argv[1]);

   hp = gethostbyname(argv[1]);
   //hp = gethostbyname("localhost");
   if(hp == 0)
   {
       fprintf(stderr, "%s:unknown host\n", argv[1]);
       exit(2);
   }

   server.sin_family = AF_INET;
   server.sin_port = htons(atoi(argv[2]));
   memcpy((char*)&server.sin_addr, (char*)hp->h_addr, hp->h_length);

   servermsg.sin_family = AF_INET;
   servermsg.sin_port = htons(atoi(argv[2]) +1);
   memcpy((char*)&servermsg.sin_addr, (char*)hp->h_addr, hp->h_length);

   if(connect(sock, (struct sockaddr *)&server, sizeof(server)) < 0 ||
       connect(sockmsg, (struct sockaddr *)&servermsg, sizeof(servermsg)) < 0)
   {
       perror("connecting stream socket");
       exit(1);
   }

   while(TRUE)
   {
       memset(user_cmd, 0, 10);
       memset(cmd_arg, 0, 20);

       //First get the input character by user
       printf("command: ");
       scanf("%s", user_cmd);

       //Then get the command charater and argument
       if(strcmp(user_cmd, "quit") == 0)
       {
           cmd_quit(sock, sockmsg);
           close(sockmsg);
           close(sock);
           printf("connection closed\n\n");
           exit(0);
       }
       else if(strcmp(user_cmd, "?") == 0)
       {
           cmd_help(sock, sockmsg);
       }
       else if(strcmp(user_cmd, "pwd") == 0)
       {
           printf("stcmp pwd --------------->>>\n");
           cmd_pwd(sock, sockmsg);
       }
       else if(strcmp(user_cmd, "dir") == 0)
       {
           cmd_dir(sock, sockmsg);
       }
       else if(strcmp(user_cmd, "cd") == 0)
       {
           scanf("%s", cmd_arg);
           cmd_cd(sock, sockmsg, cmd_arg);
       }
       else if(strcmp(user_cmd, "cd..") == 0)
       {
           cmd_cdback(sock, sockmsg);
       }
       else if(strcmp(user_cmd, "get") == 0 )
       {
           scanf("%s", cmd_arg);
           cmd_get(sock, sockmsg, cmd_arg);
       }
       else if(strcmp(user_cmd, "put") == 0)
       {
           scanf("%s", cmd_arg);
           cmd_put(sock, sockmsg, cmd_arg);
       }
       else
       {
           printf("bad command!\n");
       }
   }
}

void cmd_pwd(int sock, int sockmsg)
{
   //int numRead
   char dirName[30];
   write(sockmsg, user_cmd, sizeof(user_cmd));
   read(sock, dirName, 30);
   printf("dirName is ------->> %s\n", dirName);
}

//command dir
void cmd_dir(int sock, int sockmsg)
{
   int i, fileNum = 0;
   char fileInfo[50];

   write(sockmsg, user_cmd, sizeof(user_cmd));
   read(sock, &fileNum, sizeof(int));

   printf("---------------------------------------\n");
   printf("file number : %d\n", fileNum);
   if(fileNum > 0)
   {
       for(i=0; i<fileNum; i++)
       {
           memset(fileInfo, 0, sizeof(fileInfo));
           read(sock, fileInfo, sizeof(fileInfo));
           printf("%s\n", fileInfo);
       }
       printf("-----------------------------------\n");
   }
   else if(fileNum == 0)
   {
       printf("directory of server point is empty.\n");
       return;
   }
   else
   {
       printf("error in command 'dir'\n");
       return;
   }
}

void cmd_cd(int sock, int sockmsg, char *dirName)
{
   char currentDirPath[200];
   write(sockmsg, user_cmd, sizeof(user_cmd));
   write(sockmsg, cmd_arg, sizeof(cmd_arg));
   printf("172 ----------->>\n");
   read(sock, currentDirPath, sizeof(currentDirPath));
   printf("174 ------------>>\n");
   printf("now is directory : %s\n", currentDirPath);
}

//command cd..
void cmd_cdback(int sock, int sockmsg)
{
   write(sockmsg, user_cmd, sizeof(user_cmd));
}

//command quit
void cmd_quit(int sock, int sockmsg)
{
   write(sockmsg, user_cmd, sizeof(user_cmd));
}

//command help
void cmd_help(int sock, int sockmsg)
{
   char help[300];
   write(sockmsg, user_cmd, sizeof(user_cmd));
   read(sock, help, 300);

   printf("%s\n", help);
}

//command get
void cmd_get(int sock, int sockmsg, char *fileName)
{
   int fd;
   long fileSize;
   char localFilePath[200];

   write(sockmsg, user_cmd, sizeof(user_cmd));
   write(sockmsg, cmd_arg, sizeof(cmd_arg));
   printf("%s\n%s\n", user_cmd, cmd_arg);

   memset(localFilePath, 0, sizeof(localFilePath));
   getcwd(localFilePath, sizeof(localFilePath));
   strcat(localFilePath, "/");
   strcat(localFilePath, fileName);

   fd = open(localFilePath, O_RDWR|O_CREAT, S_IREAD|S_IWRITE);
   if(fd!=-1)
   {
       memset(buf, 0, dataLen);
       read(sock, &fileSize, sizeof(long));
       while(fileSize > dataLen)
       {
           read(sock, buf, dataLen);
           write(fd, buf, dataLen);
           fileSize = fileSize - dataLen;
       }

       read(sock, buf, fileSize);
       write(fd, buf, fileSize);
       close(fd);
       printf("download completed\n");
   }
   else
   {
       printf("open file %s failed\n", localFilePath);
   }
}

//command put
void cmd_put(int sock, int sockmsg, char *fileName)
{
   write(sockmsg, user_cmd, sizeof(user_cmd));
   write(sockmsg, cmd_arg, sizeof(cmd_arg));

   int fd;
   long fileSize;
   int numRead;
   char filePath[200];

   struct stat fileSta;

   memset(filePath, 0, sizeof(filePath));
   getcwd(filePath, sizeof(filePath));

   strcat(filePath, "/");
   strcat(filePath, fileName);

   fd = open(filePath, O_RDONLY, S_IREAD);
   if(fd != -1)
   {
       fstat(fd, &fileSta);
       fileSize = (long)fileSta.st_size;

       write(sock, &fileSize, sizeof(long));
       memset(buf, 0, dataLen);
       while(fileSize > dataLen)
       {
           read(fd, buf, dataLen);
           write(sock, buf, dataLen);
           fileSize = fileSize -dataLen;
       }

       read(fd, buf, fileSize);
       write(sock, buf, fileSize);
       close(fd);
       printf("upload completed\n");
   }
   else
   {
       printf("open file %s failed\n", filePath);
   }
}