uint8_t* insert_connection_stats_payload(uint8_t* ptr, uint8_t* end, nabto_connect* con) {
nabto_stamp_t now;
uint32_t connectionAge;
UNABTO_ASSERT(ptr <= end);
if (end-ptr < NP_PAYLOAD_CONNECTION_STATS_BYTELENGTH) {
return NULL;
}
ptr = insert_payload(ptr, NP_PAYLOAD_TYPE_CONNECTION_STATS, 0, 21);
WRITE_FORWARD_U8(ptr, NP_PAYLOAD_CONNECTION_STATS_VERSION);
now = nabtoGetStamp();
connectionAge = nabtoStampDiff2ms(nabtoStampDiff(&now, &con->stats.connectionStart));
WRITE_FORWARD_U32(ptr, connectionAge);
WRITE_FORWARD_U32(ptr, con->stats.packetsReceived);
WRITE_FORWARD_U32(ptr, con->stats.packetsSent);
WRITE_FORWARD_U32(ptr, con->stats.bytesReceived);
WRITE_FORWARD_U32(ptr, con->stats.bytesSent);
return ptr;
}
bool build_and_send_packet(struct nabto_stream_s* stream, uint8_t type, uint32_t seq, const uint8_t* winInfoData, size_t winInfoSize, uint8_t* data, uint16_t size, struct nabto_stream_sack_data* sackData)
{
enum { l_win = NP_PAYLOAD_WINDOW_SYN_BYTELENGTH - NP_PAYLOAD_WINDOW_BYTELENGTH };
uint8_t* ptr;
nabto_connect* con = stream->connection;
uint8_t* buf = nabtoCommunicationBuffer;
struct nabto_stream_tcb* tcb = &stream->u.tcb;
uint32_t ackToSend = unabto_stream_ack_number_to_send(tcb);
uint16_t recvWinSize = unabto_stream_advertised_window_size(tcb);
if (con == NULL) {
return false;
}
nabtoSetFutureStamp(&tcb->ackStamp, 2*tcb->cfg.timeoutMsec);
ptr = insert_data_header(buf, con->spnsi, con->nsico, stream->streamTag);
ptr = insert_payload(ptr, NP_PAYLOAD_TYPE_WINDOW, 0, l_win + winInfoSize + (type == NP_PAYLOAD_WINDOW_FLAG_ACK ? 2 : 0));
WRITE_FORWARD_U8 (ptr, type);
WRITE_FORWARD_U8 (ptr, NP_STREAM_VERSION);
WRITE_FORWARD_U16(ptr, stream->idCP);
WRITE_FORWARD_U16(ptr, stream->idSP);
WRITE_FORWARD_U32(ptr, seq);
WRITE_FORWARD_U32(ptr, ackToSend);
if (type == NP_PAYLOAD_WINDOW_FLAG_ACK) {
WRITE_FORWARD_U16(ptr, recvWinSize);
}
if (winInfoSize) {
memcpy(ptr, (const void*) winInfoData, winInfoSize); ptr += winInfoSize;
}
if (sackData && sackData->nPairs > 0) {
uint8_t i;
ptr = insert_payload(ptr, NP_PAYLOAD_TYPE_SACK, 0, 8 * sackData->nPairs);
for (i = 0; i < sackData->nPairs; i++) {
WRITE_FORWARD_U32(ptr, sackData->pairs[i].start);
WRITE_FORWARD_U32(ptr, sackData->pairs[i].end);
}
}
ptr = insert_payload(ptr, NP_PAYLOAD_TYPE_CRYPTO, 0, 0);
if (send_and_encrypt_packet_con(con, data, size, ptr)) {
tcb->ackSent = ackToSend;
tcb->lastSentAdvertisedWindow = recvWinSize;
stream->stats.sentPackets++;
return true;
} else {
return false;
}
}
bool unabto_dns_fallback_data_send_packet_start_request(unabto_dns_fallback_session* session, uint16_t chunks, nabto_endpoint* ep, uint16_t totalLength)
{
uint8_t requestBuffer[19];
uint8_t* ptr = requestBuffer;
uint8_t* buffer;
uint8_t* end;
uint16_t id;
size_t length;
WRITE_FORWARD_U8(ptr, UDF_TYPE_SEND_START);
WRITE_FORWARD_U16(ptr, 19);
WRITE_FORWARD_U16(ptr, session->sessionId);
WRITE_FORWARD_U16(ptr, session->random++);
WRITE_FORWARD_U16(ptr, session->packetId);
WRITE_FORWARD_U16(ptr, chunks);
WRITE_FORWARD_U32(ptr, ep->addr);
WRITE_FORWARD_U16(ptr, ep->port);
WRITE_FORWARD_U16(ptr, totalLength);
buffer = requestEncoded;
end = buffer + ENCODED_BUFFER_SIZE;
buffer = unabto_dns_fallback_encode_request_as_question(buffer, end, requestBuffer, 19, session->domain, &id);
if (buffer == NULL) {
return false;
}
length = buffer - requestEncoded;
NABTO_LOG_INFO(("Sending DNS start request id: %" PRIu16, id));
nabto_write(session->dnsClient.sockets[(session->dnsClient.lastUsedSocket++) % DNS_CLIENT_SOCKETS], requestEncoded, length, session->dnsClient.dnsServer.addr, session->dnsClient.dnsServer.port);
return true;
}
