#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>

#include "modbus.h"

#define SLAVE_1         0x01
#define DEFAULT_BAUD    57600L
#define TOTAL_COILS     32

int main(int argc, char* argv[])
{
    int ret;
    int start_addr;
    uint16_t dest;

    char buff[32];
    modbus_param_t mb_param;
    long baud;

    if ( argc < 3)
    {
        printf("Usage: rd_reg port address [baud]\n");
        printf("Default baud is %ld\n\n", DEFAULT_BAUD);
        printf("Examples:\n");
        printf("./rd_reg ttyS0 0x1FF       #for Linux\n");
        printf("./rd_reg com1 0x1FF        #for Cygwin\n");
        printf("\n");
        printf("./rd_reg ttyS0 0x1FF 38400 #for Linux with non default baud\n");
        return -1;
    }

    /* set device name */
    sprintf(buff, "/dev/%s", argv[1]);

    /* get the reg address */
    start_addr = (int)strtol(argv[2], NULL, 16);
    
    baud = DEFAULT_BAUD;
    if ( argc == 4 )
    {
        baud = atol(argv[3]);
        switch ( baud )
        {
            case 9600:   break;
            case 19200:  break;
            case 38400:  break;
            case 57600:  break;
            case 115200: break;
            default: 
                printf("Invalid baud: %ld\n", baud);
                printf("Supported bauds: 9600, 19200, 38400, 57600, 115200\n");
                return -1;
        }
    }

    modbus_init_rtu(&mb_param, buff, baud, "none", 8, 1);

    if (modbus_connect(&mb_param) == -1) 
    {
        printf("ERROR Connection failed\n");
        exit(1);
    }
    
    /* read the holding register */
    printf("read_holding_registers(ID 0x01, ADDRESS 0x%04X): ", start_addr);
    
    ret = read_holding_registers(&mb_param, SLAVE_1, (start_addr - 1), 1, &dest);

    if ( ret != 1 ) 
    {
        printf("FAILED\n");
        exit(1);
    }
        
    printf("OK\n");
    
    printf("ADDRESS 0x%04X, REG 0x%04X\n", start_addr, dest);
 
    return 0;
}