|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Copyright (c) 2010-2011, Red Hat, Inc.
|
|
Packit |
61cb5a |
*
|
|
Packit |
61cb5a |
* Permission to use, copy, modify, and/or distribute this software for any
|
|
Packit |
61cb5a |
* purpose with or without fee is hereby granted, provided that the above
|
|
Packit |
61cb5a |
* copyright notice and this permission notice appear in all copies.
|
|
Packit |
61cb5a |
*
|
|
Packit |
61cb5a |
* THE SOFTWARE IS PROVIDED "AS IS" AND RED HAT, INC. DISCLAIMS ALL WARRANTIES
|
|
Packit |
61cb5a |
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
|
|
Packit |
61cb5a |
* OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL RED HAT, INC. BE LIABLE
|
|
Packit |
61cb5a |
* FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
Packit |
61cb5a |
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
|
|
Packit |
61cb5a |
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
|
|
Packit |
61cb5a |
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Author: Jan Friesse <jfriesse@redhat.com>
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
#include <err.h>
|
|
Packit |
61cb5a |
#include <stdio.h>
|
|
Packit |
61cb5a |
#include <string.h>
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
#include "logging.h"
|
|
Packit |
61cb5a |
#include "msg.h"
|
|
Packit |
61cb5a |
#include "omping.h"
|
|
Packit |
61cb5a |
#include "tlv.h"
|
|
Packit |
61cb5a |
#include "util.h"
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Create answer message from query message. orig_msg is pointer to buffer with query message
|
|
Packit |
61cb5a |
* with orig_msg_len length (only used bytes, not buffer size). new_msg is pointer to buffer where
|
|
Packit |
61cb5a |
* to store result message. new_msg_len is size of buffer. ttl is value of TTL option. server_tstamp
|
|
Packit |
61cb5a |
* is boolean variable and if set, server timestamp option is added to message.
|
|
Packit |
61cb5a |
*
|
|
Packit |
61cb5a |
* All options from original messages are copied without changing order. Only exceptions are Server
|
|
Packit |
61cb5a |
* Info, Multicast Prefix, Session ID, TTL and Server Timestamp, which are not copied.
|
|
Packit |
61cb5a |
*
|
|
Packit |
61cb5a |
* Returned value is size of new message or 0 on fail (mostly because msg_len
|
|
Packit |
61cb5a |
* is smaller then needed). If success, new message is always at least 1 bytes long.
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
size_t
|
|
Packit |
61cb5a |
msg_answer_create(const char *orig_msg, size_t orig_msg_len, char *new_msg, size_t new_msg_len,
|
|
Packit |
61cb5a |
uint8_t ttl, int server_tstamp)
|
|
Packit |
61cb5a |
{
|
|
Packit |
61cb5a |
struct tlv_iterator tlv_iter;
|
|
Packit |
61cb5a |
enum tlv_opt_type opt_type;
|
|
Packit |
61cb5a |
size_t pos;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
pos = 0;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
new_msg[pos++] = (unsigned char)MSG_TYPE_ANSWER;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
memset(&tlv_iter, 0, sizeof(tlv_iter));
|
|
Packit |
61cb5a |
tlv_iter_init(orig_msg, orig_msg_len, &tlv_iter);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
while (tlv_iter_next(&tlv_iter) != -1) {
|
|
Packit |
61cb5a |
opt_type = tlv_iter_get_type(&tlv_iter);
|
|
Packit |
61cb5a |
if (opt_type != TLV_OPT_TYPE_SERVER_INFO &&
|
|
Packit |
61cb5a |
opt_type != TLV_OPT_TYPE_MCAST_PREFIX &&
|
|
Packit |
61cb5a |
opt_type != TLV_OPT_TYPE_SES_ID &&
|
|
Packit |
61cb5a |
opt_type != TLV_OPT_TYPE_TTL &&
|
|
Packit |
61cb5a |
opt_type != TLV_OPT_TYPE_SERVER_TSTAMP) {
|
|
Packit |
61cb5a |
tlv_iter_item_copy(&tlv_iter, new_msg, new_msg_len, &pos;;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_ttl(new_msg, new_msg_len, &pos, ttl) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (server_tstamp) {
|
|
Packit |
61cb5a |
if (tlv_add_server_tstamp(new_msg, new_msg_len, &pos) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
return (pos);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
small_buf_err:
|
|
Packit |
61cb5a |
return (0);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Decode message. Decoded message is stored in msg_decoded structure.
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
void
|
|
Packit |
61cb5a |
msg_decode(const char *msg, size_t msg_len, struct msg_decoded *decoded)
|
|
Packit |
61cb5a |
{
|
|
Packit |
61cb5a |
char debug_str[128];
|
|
Packit |
61cb5a |
struct tlv_iterator tlv_iter;
|
|
Packit |
61cb5a |
size_t pos;
|
|
Packit |
61cb5a |
uint32_t u32, u32_2;
|
|
Packit |
61cb5a |
uint16_t tlv_len;
|
|
Packit |
61cb5a |
uint16_t u16;
|
|
Packit |
61cb5a |
uint8_t u8;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
memset(decoded, 0, sizeof(struct msg_decoded));
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
decoded->msg_type = (enum msg_type)msg[0];
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("Message type %c (0x%X)", decoded->msg_type, decoded->msg_type);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
tlv_iter_init(msg, msg_len, &tlv_iter);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
while (tlv_iter_next(&tlv_iter) == 0) {
|
|
Packit |
61cb5a |
tlv_len = tlv_iter_get_len(&tlv_iter);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (logging_get_verbose() >= LOGGING_LEVEL_DEBUG2) {
|
|
Packit |
61cb5a |
snprintf(debug_str, sizeof(debug_str), "%u:%s:%u:",
|
|
Packit |
61cb5a |
tlv_iter_get_type(&tlv_iter),
|
|
Packit |
61cb5a |
tlv_opt_type_to_str(tlv_iter_get_type(&tlv_iter)), tlv_len);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
switch (tlv_iter_get_type(&tlv_iter)) {
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_VERSION:
|
|
Packit |
61cb5a |
if (tlv_len == 1) {
|
|
Packit |
61cb5a |
memcpy(&u8, tlv_iter_get_data(&tlv_iter), sizeof(u8));
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
decoded->version = u8;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s%u", debug_str, u8);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen != 1", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_CLIENT_ID:
|
|
Packit |
61cb5a |
if (tlv_len > 0) {
|
|
Packit |
61cb5a |
decoded->client_id_len = tlv_len;
|
|
Packit |
61cb5a |
decoded->client_id = tlv_iter_get_data(&tlv_iter);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_HEXDUMP(debug_str, decoded->client_id, tlv_len);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen <= 0", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_SEQ_NUM:
|
|
Packit |
61cb5a |
if (tlv_len == 4) {
|
|
Packit |
61cb5a |
decoded->seq_num_isset = 1;
|
|
Packit |
61cb5a |
memcpy(&u32, tlv_iter_get_data(&tlv_iter), sizeof(u32));
|
|
Packit |
61cb5a |
u32 = ntohl(u32);
|
|
Packit |
61cb5a |
decoded->seq_num = u32;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s%u", debug_str, u32);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen != 4", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_CLIENT_TSTAMP:
|
|
Packit |
61cb5a |
if (tlv_len == 8) {
|
|
Packit |
61cb5a |
memcpy(&u32, tlv_iter_get_data(&tlv_iter), sizeof(u32));
|
|
Packit |
61cb5a |
u32 = ntohl(u32);
|
|
Packit |
61cb5a |
decoded->client_tstamp.tv_sec = u32;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
memcpy(&u32_2, tlv_iter_get_data(&tlv_iter) + sizeof(u32),
|
|
Packit |
61cb5a |
sizeof(u32_2));
|
|
Packit |
61cb5a |
u32_2 = ntohl(u32_2);
|
|
Packit |
61cb5a |
decoded->client_tstamp.tv_usec = u32_2;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
decoded->client_tstamp_isset = 1;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s(%u,%u)", debug_str, u32, u32_2);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen != 8", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_MCAST_GRP:
|
|
Packit |
61cb5a |
if (tlv_len > 2) {
|
|
Packit |
61cb5a |
memcpy(&u16, tlv_iter_get_data(&tlv_iter), sizeof(u16));
|
|
Packit |
61cb5a |
u16 = ntohs(u16);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if ((u16 == AF_IANA_IP && tlv_len == 6) ||
|
|
Packit |
61cb5a |
(u16 == AF_IANA_IP6 && tlv_len == 18)) {
|
|
Packit |
61cb5a |
decoded->mcast_grp_len = tlv_len;
|
|
Packit |
61cb5a |
decoded->mcast_grp = tlv_iter_get_data(&tlv_iter);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%sAF %u", debug_str, u16);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%sbad AF %u or len", debug_str, u16);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen <= 2", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_OPT_REQUEST:
|
|
Packit |
61cb5a |
if (tlv_len > 1 && (tlv_len % 2 == 0)) {
|
|
Packit |
61cb5a |
for (pos = 0; pos < (uint16_t)(tlv_len / 2); pos++) {
|
|
Packit |
61cb5a |
memcpy(&u16, tlv_iter_get_data(&tlv_iter) + pos * 2,
|
|
Packit |
61cb5a |
sizeof(u16));
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
u16 = ntohs(u16);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
switch (u16) {
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_SERVER_INFO:
|
|
Packit |
61cb5a |
decoded->request_opt_server_info = 1;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s%zu opt %u", debug_str, pos, u16);
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_SERVER_TSTAMP:
|
|
Packit |
61cb5a |
decoded->request_opt_server_tstamp = 1;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s%zu opt %u", debug_str, pos, u16);
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
default:
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s%zu unknown opt %u", debug_str,
|
|
Packit |
61cb5a |
pos, u16);
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen <= 1 || (tlv_len %%2 != 0)", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_SERVER_INFO:
|
|
Packit |
61cb5a |
if (tlv_len > 0) {
|
|
Packit |
61cb5a |
decoded->server_info = tlv_iter_get_data(&tlv_iter);
|
|
Packit |
61cb5a |
decoded->server_info_len = tlv_len;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_HEXDUMP(debug_str, decoded->server_info, tlv_len);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen <= 0", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_TTL:
|
|
Packit |
61cb5a |
if (tlv_len == 1) {
|
|
Packit |
61cb5a |
memcpy(&u8, tlv_iter_get_data(&tlv_iter), sizeof(u8));
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
decoded->ttl = u8;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s%u", debug_str, u8);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen != 1", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_MCAST_PREFIX:
|
|
Packit |
61cb5a |
if (tlv_len > 2) {
|
|
Packit |
61cb5a |
memcpy(&u16, tlv_iter_get_data(&tlv_iter), sizeof(u16));
|
|
Packit |
61cb5a |
u16 = ntohs(u16);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (u16 == AF_IANA_IP || u16 == AF_IANA_IP6) {
|
|
Packit |
61cb5a |
decoded->mcast_prefix_isset = 1;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%sAF %u", debug_str, u16);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%sbad AF %u", debug_str, u16);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen <= 2", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_SES_ID:
|
|
Packit |
61cb5a |
if (tlv_len > 0) {
|
|
Packit |
61cb5a |
decoded->ses_id_len = tlv_len;
|
|
Packit |
61cb5a |
decoded->ses_id = tlv_iter_get_data(&tlv_iter);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_HEXDUMP(debug_str, decoded->ses_id, tlv_len);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen <= 0", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
case TLV_OPT_TYPE_SERVER_TSTAMP:
|
|
Packit |
61cb5a |
if (tlv_len == 8) {
|
|
Packit |
61cb5a |
memcpy(&u32, tlv_iter_get_data(&tlv_iter), sizeof(u32));
|
|
Packit |
61cb5a |
u32 = ntohl(u32);
|
|
Packit |
61cb5a |
decoded->server_tstamp.tv_sec = u32;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
memcpy(&u32_2, tlv_iter_get_data(&tlv_iter) + sizeof(u32),
|
|
Packit |
61cb5a |
sizeof(u32_2));
|
|
Packit |
61cb5a |
u32_2 = ntohl(u32_2);
|
|
Packit |
61cb5a |
decoded->server_tstamp.tv_usec = u32_2;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
decoded->server_tstamp_isset = 1;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s(%u,%u)", debug_str, u32, u32_2);
|
|
Packit |
61cb5a |
} else {
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%slen != 8", debug_str);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
default:
|
|
Packit |
61cb5a |
DEBUG2_PRINTF("%s", debug_str);
|
|
Packit |
61cb5a |
break;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Create init message. msg is pointer to buffer where to store result message. msg_len is size of
|
|
Packit |
61cb5a |
* buffer. mcast_addr is required multicast address. client_id is client ID to store in message with
|
|
Packit |
61cb5a |
* length client_id_len.
|
|
Packit |
61cb5a |
*
|
|
Packit |
61cb5a |
* Returned value is size of new message or 0 on fail (mostly because msg_len
|
|
Packit |
61cb5a |
* is smaller then needed). If success, new message is always at least 1 bytes long.
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
size_t
|
|
Packit |
61cb5a |
msg_init_create(char *msg, size_t msg_len, int req_si, const struct sockaddr_storage *mcast_addr,
|
|
Packit |
61cb5a |
const char *client_id, size_t client_id_len)
|
|
Packit |
61cb5a |
{
|
|
Packit |
61cb5a |
size_t pos;
|
|
Packit |
61cb5a |
uint16_t u16;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
pos = 0;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (client_id == NULL) {
|
|
Packit |
61cb5a |
return (0);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
msg[pos++] = (unsigned char)MSG_TYPE_INIT;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_version(msg, msg_len, &pos) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add(msg, msg_len, &pos, TLV_OPT_TYPE_CLIENT_ID, client_id_len, client_id) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (req_si) {
|
|
Packit |
61cb5a |
u16 = TLV_OPT_TYPE_SERVER_INFO;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_opt_request(msg, msg_len, &pos, &u16, 1) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_mcast_prefix(msg, msg_len, &pos, mcast_addr) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
return (pos);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
small_buf_err:
|
|
Packit |
61cb5a |
return (0);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Create query message. msg is pointer to buffer where to store result message. msg_len is size
|
|
Packit |
61cb5a |
* of buffer. mcast_addr is required multicast group address. server_tstamp is boolean to decide if
|
|
Packit |
61cb5a |
* to include Option request option with server time stamp. client_id is Client ID with length
|
|
Packit |
61cb5a |
* client_id_len. session_id with session_id_len is similar, but for Session ID.
|
|
Packit |
61cb5a |
*
|
|
Packit |
61cb5a |
* Returned value is size of new message or 0 on fail (mostly because msg_len
|
|
Packit |
61cb5a |
* is smaller then needed). If success, new message is always at least 1 bytes long.
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
size_t
|
|
Packit |
61cb5a |
msg_query_create(char *msg, size_t msg_len, const struct sockaddr_storage *mcast_addr,
|
|
Packit |
61cb5a |
uint32_t seq_num, int server_tstamp, const char *client_id, size_t client_id_len,
|
|
Packit |
61cb5a |
const char *session_id, size_t session_id_len)
|
|
Packit |
61cb5a |
{
|
|
Packit |
61cb5a |
size_t pos;
|
|
Packit |
61cb5a |
uint16_t u16;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
pos = 0;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
msg[pos++] = (unsigned char)MSG_TYPE_QUERY;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_version(msg, msg_len, &pos) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (client_id) {
|
|
Packit |
61cb5a |
if (tlv_add(msg, msg_len, &pos, TLV_OPT_TYPE_CLIENT_ID, client_id_len,
|
|
Packit |
61cb5a |
client_id) == -1) {
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_seq_num(msg, msg_len, &pos, seq_num) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_client_tstamp(msg, msg_len, &pos) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_mcast_grp(msg, msg_len, &pos, mcast_addr) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (server_tstamp) {
|
|
Packit |
61cb5a |
u16 = TLV_OPT_TYPE_SERVER_TSTAMP;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_opt_request(msg, msg_len, &pos, &u16, 1) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add(msg, msg_len, &pos, TLV_OPT_TYPE_SES_ID, session_id_len, session_id) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
return (pos);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
small_buf_err:
|
|
Packit |
61cb5a |
return (0);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Create response message. msg is pointer to buffer where to store result message. msg_len is size
|
|
Packit |
61cb5a |
* of buffer. msg_decoded is decoded init message used for some informations (like client id, ...).
|
|
Packit |
61cb5a |
* mcast_grp and mcast_prefix are booleans used for decision, if to add Multicast Group and/or
|
|
Packit |
61cb5a |
* Multicast Prefix options. mcast_addr is value for mcast_grp and/or mcast_prefix. If none of this
|
|
Packit |
61cb5a |
* options is/are required, mcasr_addr can be set to NULL. ulticast address. session_id is
|
|
Packit |
61cb5a |
* session ID of client.
|
|
Packit |
61cb5a |
*
|
|
Packit |
61cb5a |
* Returned value is size of new message or 0 on fail (mostly because msg_len
|
|
Packit |
61cb5a |
* is smaller then needed). If success, new message is always at least 1 bytes long.
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
size_t
|
|
Packit |
61cb5a |
msg_response_create(char *msg, size_t msg_len, const struct msg_decoded *msg_decoded,
|
|
Packit |
61cb5a |
int mcast_grp, int mcast_prefix, const struct sockaddr_storage *mcast_addr,
|
|
Packit |
61cb5a |
const char *session_id, size_t session_id_len)
|
|
Packit |
61cb5a |
{
|
|
Packit |
61cb5a |
size_t pos;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
pos = 0;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
msg[pos++] = (unsigned char)MSG_TYPE_RESPONSE;
|
|
Packit |
61cb5a |
if (tlv_add_version(msg, msg_len, &pos) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (msg_decoded->client_id) {
|
|
Packit |
61cb5a |
if (tlv_add(msg, msg_len, &pos, TLV_OPT_TYPE_CLIENT_ID, msg_decoded->client_id_len,
|
|
Packit |
61cb5a |
msg_decoded->client_id) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (msg_decoded->seq_num_isset) {
|
|
Packit |
61cb5a |
if (tlv_add_seq_num(msg, msg_len, &pos, msg_decoded->seq_num) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (mcast_grp) {
|
|
Packit |
61cb5a |
if (tlv_add_mcast_grp(msg, msg_len, &pos, mcast_addr) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (msg_decoded->request_opt_server_info) {
|
|
Packit |
61cb5a |
if (tlv_add_server_info(msg, msg_len, &pos, PROGRAM_SERVER_INFO) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (mcast_prefix) {
|
|
Packit |
61cb5a |
if (tlv_add_mcast_prefix(msg, msg_len, &pos, mcast_addr) == -1)
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (session_id) {
|
|
Packit |
61cb5a |
if (tlv_add(msg, msg_len, &pos, TLV_OPT_TYPE_SES_ID, session_id_len,
|
|
Packit |
61cb5a |
session_id) == -1) {
|
|
Packit |
61cb5a |
goto small_buf_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
return (pos);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
small_buf_err:
|
|
Packit |
61cb5a |
return (0);
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
/*
|
|
Packit |
61cb5a |
* Update Server Timestamp option in message to current time stamp. msg is pointer to buffer with
|
|
Packit |
61cb5a |
* message and msg_len is length of message (without unused space).
|
|
Packit |
61cb5a |
* Function returns 0 on success, otherwise -1.
|
|
Packit |
61cb5a |
*/
|
|
Packit |
61cb5a |
int
|
|
Packit |
61cb5a |
msg_update_server_tstamp(char *msg, size_t msg_len)
|
|
Packit |
61cb5a |
{
|
|
Packit |
61cb5a |
struct tlv_iterator tlv_iter;
|
|
Packit |
61cb5a |
size_t pos;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
memset(&tlv_iter, 0, sizeof(tlv_iter));
|
|
Packit |
61cb5a |
tlv_iter_init(msg, msg_len, &tlv_iter);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
while (tlv_iter_next(&tlv_iter) != -1) {
|
|
Packit |
61cb5a |
if (tlv_iter_get_type(&tlv_iter) == TLV_OPT_TYPE_SERVER_TSTAMP) {
|
|
Packit |
61cb5a |
pos = tlv_iter.pos;
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
if (tlv_add_server_tstamp(msg, msg_len, &pos) == -1)
|
|
Packit |
61cb5a |
goto add_tstamp_err;
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
}
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
return (0);
|
|
Packit |
61cb5a |
|
|
Packit |
61cb5a |
add_tstamp_err:
|
|
Packit |
61cb5a |
return (-1);
|
|
Packit |
61cb5a |
}
|