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
|
/*
* Helper functions for imageFuse
* by Stephan Schwaer, January 2014
*/
#include "imageHelper.h"
void printLog(log_info *info){
FILE *logFile;
// Create logfile
logFile = fopen("log.txt", "w+");
if ( logFile == NULL) {
printf("Error creating/opening log.txt\n");
return;
}
//rewind(file);
fprintf(logFile, "ImageSize: %"PRIu64" MiB\n", (uint64_t)(info->imageSize/(1024ll*1024ll)));
fprintf(logFile, "ReceivedMiB: %"PRIu64" MiB\n", (uint64_t)(info->receivedBytes/(1024ll*1024ll)));
fprintf(logFile, "imageBlockCount: %"PRIu64"\n", info->imageBlockCount);
fprintf(logFile, "Blocksize: 4KiB\n\n");
fprintf(logFile, "Block access count:\n");
uint64_t i = 0;
for (; i < info->imageBlockCount; i++){
if( i % 50 == 0 ) fprintf(logFile, "\n");
fprintf(logFile, "%i ",(int) info->blockRequestCount[i]);
}
fprintf(logFile, "\n");
fclose(logFile);
}
bool sock_printable(struct sockaddr *addr, socklen_t addrLen, char *output, int len)
{
char host[100], port[10];
int ret = getnameinfo( addr, addrLen, host, 100, port, 10, NI_NUMERICHOST | NI_NUMERICSERV );
if ( ret == 0 ) snprintf( output, len, "[%s]:%s", host, port );
return ret == 0;
}
int connect_to_server(char ** server_adress, int * port){
const int on = 1;
int sock = -1;
struct addrinfo hints, *res, *ptr;
char portStr[6];
// Set hints for local addresses.
memset( &hints, 0, sizeof(hints) );
hints.ai_flags = AI_PASSIVE;
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
snprintf( portStr, sizeof portStr, "%d", *port );
if ( getaddrinfo( *server_adress, portStr, &hints, &res ) != 0 || res == NULL ) return false;
// Attempt to bind to all of the addresses as long as there's room in the poll list
for( ptr = res; ptr != NULL; ptr = ptr->ai_next ) {
char bla[100];
if ( !sock_printable( (struct sockaddr*)ptr->ai_addr, ptr->ai_addrlen, bla, 100 ) ) snprintf( bla, 100, "[invalid]" );
printf( "Trying to connect to %s ", bla );
sock = socket( ptr->ai_family, ptr->ai_socktype, ptr->ai_protocol );
if ( sock < 0 ) {
printf( "...cannot create socket, errno=%d\n", errno );
sock = -1;
continue;
}
setsockopt( sock, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on) );
if ( ptr->ai_family == PF_INET6 ) setsockopt( sock, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on) );
if (connect(sock, ptr->ai_addr, ptr->ai_addrlen) < 0) {
// if ( bind( sock, ptr->ai_addr, ptr->ai_addrlen ) == -1 ) {
printf( "...socket Error, errno=%d\n", errno );
close( sock );
sock = -1;
continue;
} else {
printf( "... connecting successful!\n");
break;
}
}
freeaddrinfo( res );
return sock;
}
|
