int bs_stats_tube(int fd, char *tube, char **yaml) {
char command[512];
snprintf(command, 512, "stats-tube %s\r\n", tube);
return bs_get_info(fd, command, yaml);
}
int bs_stats_job(int fd, int id, char **yaml) {
char command[128];
snprintf(command, 128, "stats-job %d\r\n", id);
return bs_get_info(fd, command, yaml);
}
int bs_list_tubes_watched(int fd, char **yaml) {
char command[64];
snprintf(command, 64, "list-tubes-watched\r\n");
return bs_get_info(fd, command, yaml);
}
int bs_stats(int fd, char **yaml) {
char command[64];
snprintf(command, 64, "stats\r\n");
return bs_get_info(fd, command, yaml);
}
int main(int argc, char *argv[])
{
fd_set readfds;
struct timeval tv;
int result;
char * port = "5685";
internal_data_t data;
char * filename = "bootstrap_server.ini";
int opt;
FILE * fd;
command_desc_t commands[] =
{
{"boot", "Bootstrap a client (Server Initiated).", " boot URI [NAME]\r\n"
" URI: uri of the client to bootstrap\r\n"
" NAME: endpoint name of the client as in the .ini file (optionnal)\r\n"
"Example: boot coap://[::1]:56830 testlwm2mclient", prv_bootstrap_client, &data},
{"q", "Quit the server.", NULL, prv_quit, NULL},
COMMAND_END_LIST
};
while ((opt = getopt(argc, argv, "f:p:")) != -1)
{
switch (opt)
{
case 'f':
filename = optarg;
break;
case 'p':
port = optarg;
break;
default:
print_usage(filename, port);
return 0;
}
}
memset(&data, 0, sizeof(internal_data_t));
data.sock = create_socket(port);
if (data.sock < 0)
{
fprintf(stderr, "Error opening socket: %d\r\n", errno);
return -1;
}
data.lwm2mH = lwm2m_init(NULL, prv_buffer_send, NULL);
if (NULL == data.lwm2mH)
{
fprintf(stderr, "lwm2m_init() failed\r\n");
return -1;
}
signal(SIGINT, handle_sigint);
fd = fopen(filename, "r");
if (fd == NULL)
{
fprintf(stderr, "Opening file %s failed.\r\n", filename);
return -1;
}
data.bsInfo = bs_get_info(fd);
fclose(fd);
if (data.bsInfo == NULL)
{
fprintf(stderr, "Reading Bootsrap Info from file %s failed.\r\n", filename);
return -1;
}
lwm2m_set_bootstrap_callback(data.lwm2mH, prv_bootstrap_callback, (void *)&data);
fprintf(stdout, "LWM2M Bootstrap Server now listening on port %s.\r\n\n", port);
fprintf(stdout, "> "); fflush(stdout);
while (0 == g_quit)
{
endpoint_t * endP;
FD_ZERO(&readfds);
FD_SET(data.sock, &readfds);
FD_SET(STDIN_FILENO, &readfds);
tv.tv_sec = 60;
tv.tv_usec = 0;
result = lwm2m_step(data.lwm2mH, &(tv.tv_sec));
if (result != 0)
{
fprintf(stderr, "lwm2m_step() failed: 0x%X\r\n", result);
return -1;
}
result = select(FD_SETSIZE, &readfds, 0, 0, &tv);
if ( result < 0 )
{
if (errno != EINTR)
{
fprintf(stderr, "Error in select(): %d\r\n", errno);
}
}
else if (result >= 0)
{
uint8_t buffer[MAX_PACKET_SIZE];
int numBytes;
// Packet received
if (FD_ISSET(data.sock, &readfds))
{
struct sockaddr_storage addr;
socklen_t addrLen;
addrLen = sizeof(addr);
numBytes = recvfrom(data.sock, buffer, MAX_PACKET_SIZE, 0, (struct sockaddr *)&addr, &addrLen);
if (numBytes == -1)
{
fprintf(stderr, "Error in recvfrom(): %d\r\n", errno);
}
else
{
char s[INET6_ADDRSTRLEN];
in_port_t port;
connection_t * connP;
s[0] = 0;
if (AF_INET == addr.ss_family)
{
struct sockaddr_in *saddr = (struct sockaddr_in *)&addr;
inet_ntop(saddr->sin_family, &saddr->sin_addr, s, INET6_ADDRSTRLEN);
port = saddr->sin_port;
}
else if (AF_INET6 == addr.ss_family)
{
struct sockaddr_in6 *saddr = (struct sockaddr_in6 *)&addr;
inet_ntop(saddr->sin6_family, &saddr->sin6_addr, s, INET6_ADDRSTRLEN);
port = saddr->sin6_port;
}
fprintf(stderr, "%d bytes received from [%s]:%hu\r\n", numBytes, s, ntohs(port));
output_buffer(stderr, buffer, numBytes, 0);
connP = connection_find(data.connList, &addr, addrLen);
if (connP == NULL)
{
connP = connection_new_incoming(data.connList, data.sock, (struct sockaddr *)&addr, addrLen);
if (connP != NULL)
{
data.connList = connP;
}
}
if (connP != NULL)
{
lwm2m_handle_packet(data.lwm2mH, buffer, numBytes, connP);
}
}
}
// command line input
else if (FD_ISSET(STDIN_FILENO, &readfds))
{
numBytes = read(STDIN_FILENO, buffer, MAX_PACKET_SIZE - 1);
if (numBytes > 1)
{
buffer[numBytes] = 0;
handle_command(commands, (char*)buffer);
}
if (g_quit == 0)
{
fprintf(stdout, "\r\n> ");
fflush(stdout);
}
else
{
fprintf(stdout, "\r\n");
}
}
// Do operations on endpoints
prv_endpoint_clean(&data);
endP = data.endpointList;
while (endP != NULL)
{
switch(endP->status)
{
case CMD_STATUS_OK:
endP->cmdList = endP->cmdList->next;
endP->status = CMD_STATUS_NEW;
// fall through
case CMD_STATUS_NEW:
prv_send_command(&data, endP);
break;
default:
break;
}
endP = endP->next;
}
}
}
lwm2m_close(data.lwm2mH);
bs_free_info(data.bsInfo);
while (data.endpointList != NULL)
{
endpoint_t * endP;
endP = data.endpointList;
data.endpointList = data.endpointList->next;
prv_endpoint_free(endP);
}
close(data.sock);
connection_free(data.connList);
return 0;
}