rooget/put_unique.c at master · Napat/rooget · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
#include "common.h"
#include "passive_connect.h"

/*
Upload files on server.
*/
void put_unique(char *arg,char *user_input,int sockfd)
{
	/* Temporary variables*/
	int no_of_bytes;
	int port;
	int newsockfd;
	int fd;
	int p;
	int total;
	int size;
	int file_size;
	int down = 2;
	int temp;
	int temp1;

	struct timeval tm;/* time structure to set time wait for receive buffer */
	tm.tv_sec = 1;
	tm.tv_usec = 750000;

	struct stat buff;

	char message_from_server[MAXSZ];
	char message_to_server[MAXSZ];
	char file[MAXSZ];// File name
	char file_name[MAXSZ];// File name with instruction to server
	char data[MAXSZ];// Data transfer

	/* Initialise all the character arrays */
	bzero(message_from_server,MAXSZ);
	bzero(message_to_server,MAXSZ);
	bzero(file_name,MAXSZ);
	bzero(file,MAXSZ);
	bzero(data,MAXSZ);

	/* Tell server to change to BINARY mode */
	send(sockfd,"TYPE I\r\n",8,0);

	while((no_of_bytes = recv(sockfd,message_from_server,MAXSZ,0)) > 0)
	{
		message_from_server[no_of_bytes] = '\0';
		printf("%s",message_from_server);
		fflush(stdout);
		if(strstr(message_from_server,"200 ") > 0 || strstr(message_from_server,"501 ") > 0 ||strstr(message_from_server,"500 ") > 0 ||strstr(message_from_server,"504 ") > 0 ||strstr(message_from_server,"421 ") > 0 || strstr(message_from_server,"530 ") > 0)
			break;
	}
	printf("\n");

	if(strstr(message_from_server,"501 ") > 0 ||strstr(message_from_server,"500 ") > 0 ||strstr(message_from_server,"504 ") > 0 ||strstr(message_from_server,"421 ") > 0 || strstr(message_from_server,"530 ") > 0)
		return;

	/* Send request for PASSIVE connection */
	send(sockfd,passive,strlen(passive),0);

	while((no_of_bytes = recv(sockfd,message_from_server,MAXSZ,0)) > 0)
	{
		message_from_server[no_of_bytes] = '\0';
		printf("%s",message_from_server);
		fflush(stdout);
		if(strstr(message_from_server,"227 ") > 0 || strstr(message_from_server,"501 ") > 0 ||strstr(message_from_server,"500 ") > 0 ||strstr(message_from_server,"502 ") > 0 ||strstr(message_from_server,"421 ") > 0 || strstr(message_from_server,"530 ") > 0)
			break;

	}
	printf("\n");

	if(strstr(message_from_server,"501 ") > 0 ||strstr(message_from_server,"500 ") > 0 ||strstr(message_from_server,"502 ") > 0 ||strstr(message_from_server,"421 ") > 0 || strstr(message_from_server,"530 ") > 0)
		return;

	/* Server accepts request and sends PORT variables */
	if(strncmp(message_from_server,"227",3) == 0)
	{
		/* Generate a PORT number using PORT variables */
		port = passive_port_number(message_from_server);

		/* Connect to server using another PORT for file transfers */
		newsockfd = func_to_connect_passive(arg,port);

		/* Send file name to server */
		sprintf(file_name,"STOU %s\r\n",user_input + 8);
		send(sockfd,file_name,strlen(file_name),0);

		while((no_of_bytes = recv(sockfd,message_from_server,MAXSZ,0)) > 0)
		{
			message_from_server[no_of_bytes] = '\0';
			printf("%s",message_from_server);
			fflush(stdout);
			if(strstr(message_from_server,"550 ") > 0 ||strstr(message_from_server,"125 ") > 0 ||strstr(message_from_server,"150 ") > 0 || strstr(message_from_server,"501 ") > 0 ||strstr(message_from_server,"500 ") > 0 ||strstr(message_from_server,"452 ") > 0 ||strstr(message_from_server,"421 ") > 0 || strstr(message_from_server,"530 ") > 0 || strstr(message_from_server,"553 ") > 0 ||strstr(message_from_server,"532 ") > 0)
				break;
		}
		printf("\n");

		/* Send file data to server */
		if(strncmp(message_from_server,"150",3) == 0 || strncmp(message_from_server,"125",3) == 0)
		{
			sprintf(file,"%s",user_input + 8);

			fd = open(file,O_RDONLY);
			fstat(fd,&buff);
			size = (int)buff.st_size;
			file_size = 0;
			if(size % 100 == 0)
				temp = (size / 100);
			else
				temp = (size / 100) + 1;

			temp1 = temp;
			printf("Uploading [");
			fflush(stdout);
			while(size > 0)
			{
				no_of_bytes = read(fd,data,MAXSZ);
				file_size += no_of_bytes;
				temp1 = temp * down;

				while(temp1 <= file_size)
				{
					printf("#");
					fflush(stdout);
					down += 2;
					temp1 = temp * down;
				}
				total = 0;
				while(total < no_of_bytes)
                		{
                		        p = send(newsockfd,data + total,no_of_bytes - total,0);
                       			 total += p;
				}
				size -= no_of_bytes;
			}

			printf("] 100%%\n");
			fflush(stdout);

			close(newsockfd);
			/* Set time boundation on receive buffer */
			setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO,(char *)&tm,sizeof(tm));
			while((no_of_bytes = recv(sockfd,message_from_server,MAXSZ,0)) > 0)
			{
				message_from_server[no_of_bytes] = '\0';
				printf("%s\n",message_from_server);
				fflush(stdout);
				if(strstr(message_from_server,"226 ") > 0)
					break;
			}
		}

	}
}