-
Notifications
You must be signed in to change notification settings - Fork 0
/
client.c
120 lines (97 loc) · 3.03 KB
/
client.c
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
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <netinet/in.h>
#include <netdb.h>
#define BUFFER_LENGTH (10 * 1024)
void acceptFile(int socketHandle);
void fail(const char* message) {
fprintf(stderr, "%s\n", message);
exit(1);
}
int main(int argc, char *argv[])
{
int sockfd, portno;
int bytesWritten, bytesRead;
int fileHandle;
struct sockaddr_in serv_addr;
struct hostent *server;
char buffer[BUFFER_LENGTH + 1];
if (argc < 3) {
sprintf(buffer, "Usage: %s hostname port", argv[0]);
fail(buffer);
}
portno = atoi(argv[2]);
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
fail("Error while opening socket");
}
server = gethostbyname(argv[1]);
if (server == NULL) {
fail("Error: no such host");
}
bzero((char *) &serv_addr, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
bcopy((char *)server->h_addr, (char *)&serv_addr.sin_addr.s_addr, server->h_length);
serv_addr.sin_port = htons(portno);
if (connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr)) < 0) {
fail("Error while connecting");
}
acceptFile(sockfd);
close(sockfd);
return 0;
}
void acceptFile(int socketHandle)
{
int fileLen, nameLen;
int bytes_read, bytes_written;
int fileHandle;
char buffer[BUFFER_LENGTH + 1];
char filename[BUFFER_LENGTH + 1];
bytes_read = read(socketHandle, &nameLen, sizeof(nameLen));
if (bytes_read != sizeof(nameLen)) {
fail("Can't read filename length");
}
printf("Filename length = %d\n", nameLen);
bzero(buffer, sizeof(BUFFER_LENGTH));
bytes_read = read(socketHandle, buffer, nameLen);
if (bytes_read != nameLen) {
fail("Can't read filename");
}
printf("Filename = %s\n", buffer);
bzero(filename, sizeof(filename));
strcpy(filename, "accepted_files/");
strcat(filename, buffer);
bytes_read = read(socketHandle, &fileLen, sizeof(fileLen));
if (bytes_read != sizeof(fileLen)) {
fail("Can't read file length");
}
printf("File length = %d\n", fileLen);
fileHandle = open(filename, O_WRONLY | O_CREAT | O_TRUNC, S_IWRITE | S_IREAD);
if (fileHandle < 0) {
fail("Can't open file to write");
}
while (fileLen > 0) {
bzero(buffer, sizeof(buffer));
bytes_read = read(socketHandle, buffer, BUFFER_LENGTH);
if (bytes_read < 0) {
fail("Can't read file from socket");
}
bytes_written = write(fileHandle, buffer, bytes_read);
if (bytes_written < 0) {
fail("Can't write data to file");
}
fileLen -= bytes_read;
}
close(fileHandle);
printf("File %s is transmitted\n", filename);
bytes_written = write(socketHandle, "Transmission finished", 23);
if (bytes_written < 0) {
fail("Transmission is not finished really. Can't write final message");
}
}