Feature/length check use pointers (#66)
* Added received bytes length check and some comments. * Added more check for MAX_PDU
This commit is contained in:
@@ -19,6 +19,13 @@ BACNET_BVLC_RESULT BVLC_Result_Code = BVLC_RESULT_SUCCESSFUL_COMPLETION;
|
||||
/** The current BVLC Function Code being handled. */
|
||||
BACNET_BVLC_FUNCTION BVLC_Function_Code = BVLC_RESULT; /* A safe default */
|
||||
|
||||
/** Encode the BVLC Result message
|
||||
*
|
||||
* @param pdu - buffer to store the encoding
|
||||
* @param result_code - BVLC result code
|
||||
*
|
||||
* @return number of bytes encoded
|
||||
*/
|
||||
static int bvlc_encode_bvlc_result(uint8_t * pdu,
|
||||
BACNET_BVLC_RESULT result_code)
|
||||
{
|
||||
@@ -35,6 +42,19 @@ static int bvlc_encode_bvlc_result(uint8_t * pdu,
|
||||
return 6;
|
||||
}
|
||||
|
||||
/**
|
||||
* The common send function for bvlc functions, using b/ip.
|
||||
*
|
||||
* @param dest_addr - Points to a sockaddr_in structure containing the
|
||||
* destination address. The length and format of the address depend
|
||||
* on the address family of the socket (AF_INET).
|
||||
* The address is in network byte order.
|
||||
* @param dest_port - Destination port number
|
||||
* @param mtu - the bytes of data to send
|
||||
* @param mtu_len - the number of bytes of data to send
|
||||
* @return Upon successful completion, returns the number of bytes sent.
|
||||
* Otherwise, -1 shall be returned and errno set to indicate the error.
|
||||
*/
|
||||
static int bvlc_send_mpdu(uint8_t * dest_addr, /* the destination address */
|
||||
|
||||
uint16_t * dest_port, /* the destination port */
|
||||
@@ -51,6 +71,11 @@ static int bvlc_send_mpdu(uint8_t * dest_addr, /* the destination address */
|
||||
return sendto_func(bip_socket(), mtu, mtu_len, dest_addr, *dest_port);
|
||||
}
|
||||
|
||||
/** Sends a BVLC Result
|
||||
*
|
||||
* @param dest_addr - destination address
|
||||
* @param dest_port - destination port
|
||||
*/
|
||||
static void bvlc_send_result(uint8_t * dest_addr,
|
||||
uint16_t * dest_port, /* the destination address */
|
||||
|
||||
@@ -65,50 +90,65 @@ static void bvlc_send_result(uint8_t * dest_addr,
|
||||
return;
|
||||
}
|
||||
|
||||
/** Note any BVLC_RESULT code, or NAK the BVLL message in the unsupported cases.
|
||||
* Use this handler when you are not a BBMD.
|
||||
* Sets the BVLC_Function_Code in case it is needed later.
|
||||
*
|
||||
* @param addr [in] Socket address to send any NAK back to.
|
||||
* @param port [in] Socket port
|
||||
* @param npdu [in] The received buffer.
|
||||
* @param received_bytes [in] How many bytes in npdu[].
|
||||
* @return Non-zero BVLC_RESULT_ code if we sent a response (NAK) to this
|
||||
* BVLC message. If zero, may need further processing.
|
||||
*/
|
||||
uint16_t bvlc_for_non_bbmd(uint8_t * addr,
|
||||
uint16_t * port,
|
||||
uint8_t * npdu,
|
||||
uint16_t received_bytes)
|
||||
{
|
||||
|
||||
uint16_t result_code = 0; /* aka, BVLC_RESULT_SUCCESSFUL_COMPLETION */
|
||||
BVLC_Function_Code = npdu[1]; /* The BVLC function */
|
||||
switch (BVLC_Function_Code) {
|
||||
case BVLC_RESULT:
|
||||
if (received_bytes >= 6) {
|
||||
/* This is the result of our foreign device registration */
|
||||
(void) decode_unsigned16(&npdu[4], &result_code);
|
||||
BVLC_Result_Code = (BACNET_BVLC_RESULT) result_code;
|
||||
fprintf(stderr, "BVLC: Result Code=%d\n", BVLC_Result_Code);
|
||||
/* But don't send any response */
|
||||
result_code = 0;
|
||||
}
|
||||
break;
|
||||
case BVLC_WRITE_BROADCAST_DISTRIBUTION_TABLE:
|
||||
result_code = BVLC_RESULT_WRITE_BROADCAST_DISTRIBUTION_TABLE_NAK;
|
||||
break;
|
||||
case BVLC_READ_BROADCAST_DIST_TABLE:
|
||||
result_code = BVLC_RESULT_READ_BROADCAST_DISTRIBUTION_TABLE_NAK;
|
||||
break;
|
||||
/* case BVLC_READ_BROADCAST_DIST_TABLE_ACK: */
|
||||
case BVLC_REGISTER_FOREIGN_DEVICE:
|
||||
result_code = BVLC_RESULT_REGISTER_FOREIGN_DEVICE_NAK;
|
||||
break;
|
||||
case BVLC_READ_FOREIGN_DEVICE_TABLE:
|
||||
result_code = BVLC_RESULT_READ_FOREIGN_DEVICE_TABLE_NAK;
|
||||
break;
|
||||
/* case BVLC_READ_FOREIGN_DEVICE_TABLE_ACK: */
|
||||
case BVLC_DELETE_FOREIGN_DEVICE_TABLE_ENTRY:
|
||||
result_code = BVLC_RESULT_DELETE_FOREIGN_DEVICE_TABLE_ENTRY_NAK;
|
||||
break;
|
||||
case BVLC_DISTRIBUTE_BROADCAST_TO_NETWORK:
|
||||
result_code = BVLC_RESULT_DISTRIBUTE_BROADCAST_TO_NETWORK_NAK;
|
||||
break;
|
||||
/* case BVLC_FORWARDED_NPDU: */
|
||||
/* case BVLC_ORIGINAL_UNICAST_NPDU: */
|
||||
/* case BVLC_ORIGINAL_BROADCAST_NPDU: */
|
||||
default:
|
||||
break;
|
||||
|
||||
/* To check the BVLC-function code, the buffer of received
|
||||
* bytes has to be at least one byte long. */
|
||||
if (received_bytes >= 1) {
|
||||
BVLC_Function_Code = npdu[1]; /* The BVLC function */
|
||||
switch (BVLC_Function_Code) {
|
||||
case BVLC_RESULT:
|
||||
if (received_bytes >= 6) {
|
||||
/* This is the result of our foreign device registration */
|
||||
(void) decode_unsigned16(&npdu[4], &result_code);
|
||||
BVLC_Result_Code = (BACNET_BVLC_RESULT) result_code;
|
||||
fprintf(stderr, "BVLC: Result Code=%d\n", BVLC_Result_Code);
|
||||
/* But don't send any response */
|
||||
result_code = 0;
|
||||
}
|
||||
break;
|
||||
case BVLC_WRITE_BROADCAST_DISTRIBUTION_TABLE:
|
||||
result_code = BVLC_RESULT_WRITE_BROADCAST_DISTRIBUTION_TABLE_NAK;
|
||||
break;
|
||||
case BVLC_READ_BROADCAST_DIST_TABLE:
|
||||
result_code = BVLC_RESULT_READ_BROADCAST_DISTRIBUTION_TABLE_NAK;
|
||||
break;
|
||||
/* case BVLC_READ_BROADCAST_DIST_TABLE_ACK: */
|
||||
case BVLC_REGISTER_FOREIGN_DEVICE:
|
||||
result_code = BVLC_RESULT_REGISTER_FOREIGN_DEVICE_NAK;
|
||||
break;
|
||||
case BVLC_READ_FOREIGN_DEVICE_TABLE:
|
||||
result_code = BVLC_RESULT_READ_FOREIGN_DEVICE_TABLE_NAK;
|
||||
break;
|
||||
/* case BVLC_READ_FOREIGN_DEVICE_TABLE_ACK: */
|
||||
case BVLC_DELETE_FOREIGN_DEVICE_TABLE_ENTRY:
|
||||
result_code = BVLC_RESULT_DELETE_FOREIGN_DEVICE_TABLE_ENTRY_NAK;
|
||||
break;
|
||||
case BVLC_DISTRIBUTE_BROADCAST_TO_NETWORK:
|
||||
result_code = BVLC_RESULT_DISTRIBUTE_BROADCAST_TO_NETWORK_NAK;
|
||||
break;
|
||||
/* case BVLC_FORWARDED_NPDU: */
|
||||
/* case BVLC_ORIGINAL_UNICAST_NPDU: */
|
||||
/* case BVLC_ORIGINAL_BROADCAST_NPDU: */
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (result_code > 0) {
|
||||
@@ -118,6 +158,13 @@ uint16_t bvlc_for_non_bbmd(uint8_t * addr,
|
||||
return result_code;
|
||||
}
|
||||
|
||||
/** Returns the current BVLL Function Code we are processing.
|
||||
* We have to store this higher layer code for when the lower layers
|
||||
* need to know what it is, especially to differentiate between
|
||||
* BVLC_ORIGINAL_UNICAST_NPDU and BVLC_ORIGINAL_BROADCAST_NPDU.
|
||||
*
|
||||
* @return A BVLC_ function code, such as BVLC_ORIGINAL_UNICAST_NPDU.
|
||||
*/
|
||||
BACNET_BVLC_FUNCTION bvlc_get_function_code(void)
|
||||
{
|
||||
return BVLC_Function_Code;
|
||||
|
||||
Reference in New Issue
Block a user