bacnet_stack/apps/writebdt/main.c

/**************************************************************************
 *
 * Copyright (C) 2020 Steve Karg <skarg@users.sourceforge.net>
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 *********************************************************************/

/* command line tool that sends a BACnet BVLC message, and displays the reply */
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h> /* for time */
#include <errno.h>
#include <ctype.h> /* for toupper */
#include "bacnet/bactext.h"
#include "bacnet/iam.h"
#include "bacnet/basic/binding/address.h"
#include "bacnet/config.h"
#include "bacnet/bacdef.h"
#include "bacnet/npdu.h"
#include "bacnet/apdu.h"
#include "bacnet/basic/object/device.h"
#include "bacnet/datalink/datalink.h"
#include "bacnet/datalink/bvlc.h"
/* some demo stuff needed */
#ifndef DEBUG_ENABLED
#define DEBUG_ENABLED 0
#endif
#include "bacnet/basic/sys/debug.h"
#include "bacnet/basic/sys/filename.h"
#include "bacnet/basic/services.h"
#include "bacnet/basic/services.h"
#include "bacnet/basic/tsm/tsm.h"
#include "bacnet/datalink/dlenv.h"

/* buffer used for receive */
static uint8_t Rx_Buf[MAX_MPDU] = { 0 };

/* targets interpreted from the command line options */
static BACNET_IP_ADDRESS Target_BBMD_Address;
#ifndef MAX_BBMD_ENTRIES
#define MAX_BBMD_ENTRIES 128
#endif
static BACNET_IP_BROADCAST_DISTRIBUTION_TABLE_ENTRY
    BBMD_Table_Entry[MAX_BBMD_ENTRIES];

static bool Error_Detected = false;

static void MyAbortHandler(
    BACNET_ADDRESS *src, uint8_t invoke_id, uint8_t abort_reason, bool server)
{
    /* FIXME: verify src and invoke id */
    (void)src;
    (void)invoke_id;
    (void)server;
    printf("BACnet Abort: %s\r\n", bactext_abort_reason_name(abort_reason));
    Error_Detected = true;
}

static void MyRejectHandler(
    BACNET_ADDRESS *src, uint8_t invoke_id, uint8_t reject_reason)
{
    /* FIXME: verify src and invoke id */
    (void)src;
    (void)invoke_id;
    printf("BACnet Reject: %s\r\n", bactext_reject_reason_name(reject_reason));
    Error_Detected = true;
}

static void Init_Service_Handlers(void)
{
    Device_Init(NULL);
    /* we need to handle who-is
       to support dynamic device binding to us */
    apdu_set_unconfirmed_handler(SERVICE_UNCONFIRMED_WHO_IS, handler_who_is);
    /* set the handler for all the services we don't implement
       It is required to send the proper reject message... */
    apdu_set_unrecognized_service_handler_handler(handler_unrecognized_service);
    /* we must implement read property - it's required! */
    apdu_set_confirmed_handler(
        SERVICE_CONFIRMED_READ_PROPERTY, handler_read_property);
    /* handle the reply (request) coming back */
    apdu_set_unconfirmed_handler(SERVICE_UNCONFIRMED_I_AM, handler_i_am_add);
    /* handle any errors coming back */
    apdu_set_abort_handler(MyAbortHandler);
    apdu_set_reject_handler(MyRejectHandler);
}

int main(int argc, char *argv[])
{
    BACNET_ADDRESS src = { 0 }; /* address where message came from */
    uint16_t pdu_len = 0;
    unsigned timeout = 100; /* milliseconds */
    time_t total_seconds = 0;
    time_t elapsed_seconds = 0;
    time_t last_seconds = 0;
    time_t current_seconds = 0;
    time_t timeout_seconds = 0;
    long port = 0;
    int argi = 0;
    unsigned bdti = 0;
    BACNET_IP_BROADCAST_DISTRIBUTION_TABLE_ENTRY *bdt_entry;
    unsigned a[4] = { 0 }, p = 0, m[4] = { 0 };
    int c = 0;
    uint16_t result_code = 0;

    if (argc < 2) {
        printf("Usage: %s IP port <IP:port[:mask]> [<IP:port[:mask]>]\r\n",
            filename_remove_path(argv[0]));
        return 0;
    }
    if ((argc > 1) && (strcmp(argv[1], "--help") == 0)) {
        printf(
            "Send a Write-Broadcast-Distribution-Table message to a BBMD.\r\n"
            "\r\n"
            "IP:\r\n"
            "IP address of the BBMD in dotted decimal notation\r\n"
            "[port]\r\n"
            "optional BACnet/IP port number (default=47808=0xBAC0)\r\n"
            "\r\n"
            "To send a Write-Broadcast-Distribution-Table message to a BBMD\r\n"
            "at 192.168.0.1 using port 47808 table entry 10.0.0.1:47808\r\n"
            "%s 192.168.0.1 47808 10.0.0.1:47808\r\n",
            filename_remove_path(argv[0]));
        return 0;
    }
    /* decode the command line parameters */
    if (argc > 1) {
        argi = 1;
        if (!bip_get_addr_by_name(argv[argi], &Target_BBMD_Address)) {
            fprintf(
                stderr, "IP=%s - failed to convert address.\r\n", argv[argi]);
            return 1;
        }
    }
    if (argc > 2) {
        argi = 2;
        port = strtol(argv[argi], NULL, 0);
        if ((port > 0) && (port <= 65535)) {
            Target_BBMD_Address.port = (uint16_t)port;
        } else {
            fprintf(
                stderr, "port=%ld - port must be between 0-65535.\r\n", port);
            return 1;
        }
    } else {
        Target_BBMD_Address.port = 0xBAC0U;
    }
    bvlc_broadcast_distribution_table_link_array(
        &BBMD_Table_Entry[0], MAX_BBMD_ENTRIES);
    bdti = 0;
    argi = 3;
    while (argc > argi) {
        bdt_entry = &BBMD_Table_Entry[bdti];
        c = sscanf(argv[argi], "%3u.%3u.%3u.%3u:%5u:%3u.%3u.%3u.%3u", &a[0],
            &a[1], &a[2], &a[3], &p, &m[0], &m[1], &m[2], &m[3]);
        if ((c == 4) || (c == 5) || (c == 9)) {
            bvlc_address_set(&bdt_entry->dest_address, a[0], a[1], a[2], a[3]);
            if ((c == 5) || (c == 9)) {
                bdt_entry->dest_address.port = (uint16_t)p;
            } else {
                bdt_entry->dest_address.port = 0xBAC0U;
            }
            if (c == 9) {
                bvlc_broadcast_distribution_mask_set(
                    &bdt_entry->broadcast_mask, m[0], m[1], m[2], m[3]);
            } else {
                bvlc_broadcast_distribution_mask_set(
                    &bdt_entry->broadcast_mask, 255, 255, 255, 255);
            }
            bdt_entry->valid = true;
            bdti++;
            if (bdti >= MAX_BBMD_ENTRIES) {
                break;
            }
        }
        argi++;
    }
    /* setup my info */
    Device_Set_Object_Instance_Number(BACNET_MAX_INSTANCE);
    Init_Service_Handlers();
    address_init();
    dlenv_init();
    atexit(datalink_cleanup);
    /* configure the timeout values */
    last_seconds = time(NULL);
    timeout_seconds = apdu_timeout() / 1000;
    /* send the request */
    bvlc_bbmd_write_bdt(&Target_BBMD_Address, &BBMD_Table_Entry[0]);
    /* loop forever */
    for (;;) {
        /* increment timer - exit if timed out */
        current_seconds = time(NULL);
        /* returns 0 bytes on timeout */
        pdu_len = datalink_receive(&src, &Rx_Buf[0], MAX_MPDU, timeout);
        if (bvlc_get_function_code() != BVLC_INVALID) {
            if (bvlc_get_function_code() == BVLC_RESULT) {
                result_code = bvlc_get_last_result();
                printf("BVLC Result: %s\n", bvlc_result_code_name(result_code));
                break;
            }
            bvlc_set_function_code(BVLC_INVALID);
        }
        /* process */
        if (pdu_len) {
            npdu_handler(&src, &Rx_Buf[0], pdu_len);
        }
        if (Error_Detected) {
            break;
        }
        /* increment timer - exit if timed out */
        elapsed_seconds = current_seconds - last_seconds;
        if (elapsed_seconds) {
            datalink_maintenance_timer(elapsed_seconds);
        }
        total_seconds += elapsed_seconds;
        if (total_seconds > timeout_seconds) {
            break;
        }
        /* keep track of time for next check */
        last_seconds = current_seconds;
    }

    return 0;
}