int main(int argc,char *argv[])
{
int tmpres;
char cmdLine[1024];
status = 0;
//get current directory
getcwd(localAddress, sizeof localAddress);
//Create socket
strcpy(ip,argv[1]);
cmdAddr.sin_family = AF_INET;
cmdAddr.sin_addr.s_addr=inet_addr(ip);
cmdAddr.sin_port = htons(cmdPort);
memset(cmdAddr.sin_zero, '\0', sizeof cmdAddr.sin_zero);
dataAddr.sin_family=AF_INET;
dataAddr.sin_addr.s_addr=inet_addr(ip);
memset(dataAddr.sin_zero, '\0', sizeof dataAddr.sin_zero);
cmdSock = socket(AF_INET, SOCK_STREAM, 0);
if(cmdSock == -1)
{
printf("Socket creation failed");
exit(1);
}
tmpres = connect(cmdSock,(struct sockaddr*)&cmdAddr, sizeof(struct sockaddr));
if(tmpres == -1)
{
printf("Connect Error");
exit(1);
}
char * str;
int codeftp;
printf("Connection established, waiting for welcome message...\n");
while((tmpres = recv(cmdSock, buf, BUFSIZ, 0)) > 0){
sscanf(buf,"%d", &codeftp);
printf("%s", buf);
if(codeftp != 220)
{
replylogcode(codeftp);
exit(1);
}
str = strstr(buf, "\r\n");
if(str != NULL){
break;
}
memset(buf, 0, tmpres);
}
while(true) {
//Send Username
char info[50];
printf("Name (%s): ", ip);
memset(buf, 0, sizeof buf);
fgets (info, sizeof info, stdin);
sprintf(buf,"USER %s\r\n",info);
tmpres = send(cmdSock, buf, strlen(buf), 0);
memset(buf, 0, sizeof buf);
tmpres = recv(cmdSock, buf, BUFSIZ, 0);
sscanf(buf,"%d", &codeftp);
if(codeftp != 331)
{
replylogcode(codeftp);
continue;
}
printf("%s", buf);
//Send Password
memset(info, 0, sizeof info);
printf("Password: "******"PASS %s\r\n",info);
tmpres = send(cmdSock, buf, strlen(buf), 0);
memset(buf, 0, sizeof buf);
tmpres = recv(cmdSock, buf, BUFSIZ, 0);
sscanf(buf,"%d", &codeftp);
if(codeftp != 230)
{
replylogcode(codeftp);
continue;
}
printf("%s", buf);
break;
}
strcpy(mode,"off");
do {
memset(cmdLine, 0, sizeof cmdLine);
printf("ftp>");
fflush(stdin);
fgets (cmdLine, 1024, stdin);
CmdLineHandle(cmdLine);
}while(status == 0);
/*flush(stdin);
scanf("%s",cmdLine);
CmdLineHandle(cmdLine);*/
}
int main(int argc,char *argv[])
{
//######### Xu ly tham so dong lenh o day #####
//######### Stub, se con nang cap them ########
help();
if(argc < 2)
{
//***Them phan xu ly phan biet hostname va IP!!
printf("Not enough arguments\n\n");
usage();
exit(EXIT_FAILURE);
}
// Tuc co tham so -p (passive mode)
bool is_passive = false;
if(argc > 2)
{
is_passive = true;
}
printf("Trying to connect to %s\n",argv[1]);
//----- Ket thuc xu ly tham so dong lenh ------
// Cac bien su dung trong chuong trinh
int mode; //0 = active, 1 = passive
struct sockaddr_in *remote;
struct stat obj;
int sock;
int choice;
int tmpres, size;
int filehandle;
int status;
char buf[BUFSIZ+1], filename[20], *f;
char path[50];
//----------------------------------------
sock = socket(AF_INET, SOCK_STREAM, 0);
if(sock == -1)
{
printf("socket creation failed");
exit(EXIT_FAILURE);
}
char *ip = argv[1];
remote = (struct sockaddr_in *)malloc(sizeof(struct sockaddr_in *));
remote->sin_family = AF_INET;
tmpres = inet_pton(AF_INET, ip, (void *)(&(remote->sin_addr.s_addr)));
if( tmpres < 0)
{
perror("Can't set remote->sin_addr.s_addr");
exit(EXIT_FAILURE);
}
else if(tmpres == 0)
{
fprintf(stderr, "%s is not a valid IP address\n", ip);
exit(EXIT_FAILURE);
}
//Dang xu ly Passive Port
remote->sin_port = htons(PASSIVE_PORT);
tmpres = connect(sock,(struct sockaddr*)remote, sizeof(struct sockaddr));
if(tmpres == -1)
{
printf("Connect Error");
exit(EXIT_FAILURE);
}
/*
Connection Establishment
120
220
220
421
Login
USER
230
530
500, 501, 421
331, 332
PASS
230
202
530
500, 501, 503, 421
332
*/
char * str;
int codeftp;
printf("Connection established, waiting for welcome message...\n");
//How to know the end of welcome message: http://stackoverflow.com/questions/13082538/how-to-know-the-end-of-ftp-welcome-message
// - Nhan welcome message - !!
while((tmpres = recv(sock, buf, BUFSIZ, 0)) > 0)
{
sscanf(buf,"%d", &codeftp);
printf("%s", buf);
if(codeftp != 220) //120, 240, 421: something wrong
{
replylogcode(codeftp);
exit(EXIT_FAILURE);
}
str = strstr(buf, "220 \r\n");//Why ???
if(str != NULL){
break;
}
memset(buf, 0, tmpres);
}
//Man hinh huong dan su cmn dung
printf("##### Welcome to our FTP CLIENT ####\n");
//Send Username
char info[50];
printf("Name (%s): ", ip);
memset(buf, 0, sizeof buf);
scanf("%s", info);
sprintf(buf,"USER %s\r\n",info);
tmpres = send(sock, buf, strlen(buf), 0);
memset(buf, 0, sizeof buf);
tmpres = recv(sock, buf, BUFSIZ, 0);
sscanf(buf,"%d", &codeftp);
if(codeftp != 331)
{
replylogcode(codeftp);
exit(1);
}
printf("%s", buf);
//Send Password
memset(info, 0, sizeof info);
//printf("Password: "******"%s", info);
char *password;
password = getpass("Enter Password: "******"%s\n",password);
sprintf(buf,"PASS %s\r\n",password);
tmpres = send(sock, buf, strlen(buf), 0);
memset(buf, 0, sizeof buf);
tmpres = recv(sock, buf, BUFSIZ, 0);
sscanf(buf,"%d", &codeftp);
if(codeftp != 230)
{
replylogcode(codeftp);
exit(1);
}
printf("%s", buf);
//~ printf("Choose passive mode or active mode: \n");
//~ printf(" 1 - Passvive mode\n");
//~ printf(" 0 - Active mode\n");
//~ printf("Please enter your choice: ");
//~ scanf("%d", &mode);
//Nen chia thanh 2 phan`
//passive_mode_handler()
//active_mode_handler()
//***Khuc duoi nay bo het vo ham cho gon
//~ if (mode == 1) //Passive mode
//~ {
memset(buf, 0, sizeof buf);
sprintf(buf,"PASS \r\n");
tmpres = send(sock, buf, strlen(buf), 0);
memset(buf, 0, sizeof buf);
tmpres = recv(sock, buf, BUFSIZ, 0);
char passbuf[BUFSIZ+1];
strcpy(passbuf, buf);
sscanf(buf,"%d", &codeftp);
if(codeftp == 227)
{
int a1, a2, a3, a4, p1, p2, dataPort; //PASV Information
sscanf(passbuf, "227 Entering Passive Mode (%d,%d,%d,%d,%d,%d)", &a1,&a2,&a3,&a4,&p1,&p2);
//Shift right 8 bits
dataPort = (p1 * 256) + p2;
sock = socket(AF_INET, SOCK_STREAM, 0);
if(sock == -1)
{
printf("socket creation failed");
exit(1);
}
remote->sin_port = htons(dataPort);
tmpres = connect(sock,(struct sockaddr*)remote, sizeof(struct sockaddr));
if(tmpres == -1)
{
printf("Connect Error");
exit(1);
}
//***Khong nen xai choice cho nay, nen lam nhu cai FTP Client, xong lenh thi chay tiep
//vòng lặp thực hiện các lệnh
int i = 1;
while(true)
{
printf("Enter a choice:\n1- get\n2- put\n3- pwd\n4- ls\n5- cd\n6- delete\n7- quit\n");
scanf("%d", &choice);
switch(choice)
{
case 1:
printf("Enter filename to get: ");
scanf("%s", filename);
sprintf(buf,"GET %s\r\n",filename);
send(sock, buf, 100, 0);
recv(sock, &size, sizeof(int), 0);
if(!size)
{
printf("No such file on the remote directory\n\n");
break;
}
f = malloc(size);
recv(sock, f, size, 0);
while(1)
{
//Ok, khuc nay cung dc
filehandle = open(filename, O_CREAT | O_EXCL | O_WRONLY, 0666);
if(filehandle == -1)
{
sprintf(filename + strlen(filename), "%d", i);//needed only if same directory is used for both server and client
}
else
{
break;
}
}
write(filehandle, f, size);
close(filehandle);
strcpy(buf, "cat ");
strcat(buf, filename);
system(buf);
break;
case 2:
printf("Enter filename to put to server: ");
scanf("%s", filename);
filehandle = open(filename, O_RDONLY);
if(filehandle == -1)
{
printf("No such file on the local directory\n\n");
break;
}
sprintf(buf,"PUT %s\r\n", filename);
send(sock, buf, 100, 0);
stat(filename, &obj);
size = obj.st_size;
send(sock, &size, sizeof(int), 0);
sendfile(sock, filehandle, NULL, size);
recv(sock, &status, sizeof(int), 0);
if(status)
{
printf("File stored successfully\n");
}
else
{
printf("File failed to be stored to remote machine\n");
}
break;
case 3:
sprintf(buf,"PWD \r\n");
send(sock, buf, 100, 0);
recv(sock, buf, 100, 0);
printf("The path of the remote directory is: %s\n", buf);
break;
case 4:
sprintf(buf,"PWD \r\n");
send(sock, buf, 100, 0);
recv(sock, buf, BUFSIZ, 0);
//recv(sock, &size, sizeof(int), 0);
//f = malloc(size);
//recv(sock, f, size, 0);
//filehandle = creat("temp.txt", O_WRONLY);
//write(filehandle, f, size, 0);
//close(filehandle);
//printf("The remote directory listing is as follows:\n");
//system("cat temp.txt");
break;
case 5:
printf("Enter the path to change the remote directory: ");;
scanf("%s", path);
sprintf(buf,"CD %s\r\n",path);
send(sock, buf, 100, 0);
recv(sock, &status, sizeof(int), 0);
if(status)
{
printf("Remote directory successfully changed\n");
}
else
{
printf("Remote directory failed to change\n");
}
break;
case 6:
printf("Enter the path to change the remote directory: ");;
scanf("%s", path);
sprintf(buf,"DELETE %s\r\n",path);
send(sock, buf, 100, 0);
recv(sock, &status, sizeof(int), 0);
if(status)
{
printf("Remote directory successfully deleted\n");
}
else
{
printf("Remote directory failed to delete\n");
}
break;
case 7:
strcpy(buf, "QUIT \r\n");
send(sock, buf, 100, 0);
recv(sock, &status, 100, 0);
if(status)
{
printf("Server closed\nQuitting..\n");
exit(0);
}
printf("Server failed to close connection\n");
}
}
}
}
