首页 > 代码库 > ftp服务器的简单实现(二)——客户端
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);
}
}
ftp服务器的简单实现(二)——客户端