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
|
#include "rpc.h"
#include "server.h"
#include "net.h"
#include "uplink.h"
#include "log.h"
#include "locks.h"
#include "sockhelper.h"
#include "helper.h"
#include "image.h"
#include <unistd.h>
#include <string.h>
#include <jansson.h>
#include <inttypes.h>
static void clientsToJson(json_t *jsonClients);
void rpc_sendStatsJson(int sock)
{
const uint64_t bytesReceived = uplink_getTotalBytesReceived();
const uint64_t bytesSent = net_getTotalBytesSent();
const int uptime = dnbd3_serverUptime();
json_t *jsonClients = json_array();
clientsToJson( jsonClients );
json_t *statisticsJson = json_pack( "{sIsI}", "bytesReceived", (json_int_t) bytesReceived, "bytesSent", (json_int_t) bytesSent );
json_object_set_new( statisticsJson, "clients", jsonClients );
json_object_set_new( statisticsJson, "images", image_fillJson() );
json_object_set_new( statisticsJson, "uptime", json_integer( uptime ) );
char *jsonString = json_dumps( statisticsJson, 0 );
char buffer[500];
snprintf(buffer, sizeof buffer , "HTTP/1.1 200 OK\r\nConnection: Close\r\nContent-Length: %d\r\nContent-Type: application/json\r\n\r\n",
(int) strlen( jsonString ) );
write( sock, buffer, strlen( buffer ) );
sock_sendAll( sock, jsonString, strlen( jsonString ), 10 );
json_decref( statisticsJson );
// Wait for flush
shutdown( sock, SHUT_WR );
while ( read( sock, buffer, sizeof buffer ) > 0 );
free( jsonString );
}
static void clientsToJson(json_t *jsonClients)
{
json_t *clientStats;
int i;
char clientName[100];
spin_lock( &_clients_lock );
for (i = 0; i < _num_clients; ++i) {
if ( _clients[i] == NULL ) continue;
spin_lock( &_clients[i]->lock );
if ( _clients[i]->image != NULL ) {
if ( !host_to_string( &_clients[i]->host, clientName, sizeof(clientName) ) ) {
strcpy( clientName, "???" );
}
clientStats = json_pack( "{sssisI}", "address", clientName, "imageId", _clients[i]->image->id, "bytesSent", (json_int_t)_clients[i]->bytesSent );
json_array_append_new( jsonClients, clientStats );
}
spin_unlock( &_clients[i]->lock );
}
spin_unlock( &_clients_lock );
}
|
