record_tcp/main.c

// ////////////////////////////////////////////////////////////////////////////
//
// Sample program sends and receives four packets via two TCP sockets and 
// records the packets in two different record files. The record library
// device is configured using the record_tcp_config.txt file and the
// record_tcp_php_config.txt file. The record library device moves all 
// closed record files to a simulated CFDP dropbox.  Program uses 
// RegisterPrintMessage, StartLoggingMessages, StopLoggingMessages, 
// InitToolkitRecordDevice, CreateTCPClientSocketDevice, CreateDeviceKeyAlias, 
// ConnectSocketDevice, RegisterRecordDeviceData, SendPacket, 
// CloseAllRecordFiles and DSCleanUp API functions.
//
// ////////////////////////////////////////////////////////////////////////////
 
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <string.h>
#include "ds_shared.h"
#include "trek_error.h"
#include "toolkit_ds_api_error_codes.h"
#include "trek_toolkit_common_api_ansi_c.h"
#include "toolkit_ds_api_ansi_c.h"
#include "toolkit_record_device_api_ansi_c.h"
#include <sys/timeb.h>
#ifdef __linux__
#include <unistd.h>
#else
#include <windows.h>
#endif
 
// ////////////////////////////////////////////////////////////////////////////
//
// PrintTheMessage controls how messages are processed and displayed to 
// the user.
//
// ////////////////////////////////////////////////////////////////////////////
 
void PrintTheMessage(message_struct_type *mess_struct_ptr)
{
    // Do not print the debug messages
    if (mess_struct_ptr->category != MSG_CAT_DEBUG)
        printf("%s\t%s\n",GetMessageCategoryAsString(mess_struct_ptr->category),
               mess_struct_ptr->message);
}
 
// ////////////////////////////////////////////////////////////////////////////
//
// DeviceData's packet_buffer_ptr processes a record_device_struct ptr  
// defined as follows:
//
//  unsigned long ds_message_code; (e.g., DS_CLOSE)
//  char          record_library_device_name[256];  
//  char          record_file_base_name[256];       
//  char          record_path_and_filename[256];    
//
// The record_device_struct can be used to determine when the record library
// device closes a record file as well as the new name of the
// record file
//
// ////////////////////////////////////////////////////////////////////////////
 
void DeviceData(const char      *record_library_device_name, 
                int             packet_length, 
                unsigned char   *packet_buffer_ptr)
{
    record_device_struct_type *record_device_struct_ptr;
    
    record_device_struct_ptr = (record_device_struct_type *)packet_buffer_ptr;
    if (record_device_struct_ptr->ds_message_code == DS_CLOSE)
    {
        printf("Successfully closed record file with pathname: %s.\n",
            record_device_struct_ptr->record_path_and_filename);
    }
}
 
//
//  NAME:       ByteSwap4Bytes
//
//  INPUTS:     
//
//  OUTPUTS:    unsigned char *packet
//
//  RETURNS:    none
//
//  DESCRIPTION:
//          
 
void ByteSwap4Bytes(unsigned char *packet)
{
    unsigned char buffer[4];
 
    memmove(buffer, packet, 4);
    memmove(packet, &buffer[3], 1);
    memmove(packet + 1, &buffer[2], 1);
    memmove(packet + 2, &buffer[1], 1);
    memmove(packet + 3, &buffer[0], 1);
}
 
// ////////////////////////////////////////////////////////////////////////////
//
// CreateDataPackets creates four packets with an eight byte header. 
// The header is defined in the record_tcp_php_config.txt file. 
// Header definition: 2 byte big endian overall packet length
//                    4 byte big endian unix time stamp
//                    2 byte big endian mille time stamp
//
// ////////////////////////////////////////////////////////////////////////////
 
void CreateDataPackets(unsigned char *packet1_ptr,
                       unsigned char *packet2_ptr,
                       unsigned char *packet3_ptr,
                       unsigned char *packet4_ptr,
                       unsigned short *length1_ptr,
                       unsigned short *length2_ptr,
                       unsigned short *length3_ptr,
                       unsigned short *length4_ptr)
{
    // Determine if this is a little endian or big endian platform. 
    unsigned short tester= 1;
    unsigned char tester_byte[1];
    unsigned short big_endian_order = 1;
    struct timeb current_time;
    unsigned int tim;
    
    memcpy(tester_byte,&tester,1);
    if (tester_byte[0] == 0x01)
    {
        big_endian_order = 0;
    }
 
    // Zero fill the packets prior to populating.
    memset(packet1_ptr,0x00,50);
    memset(packet2_ptr,0x00,50);
    memset(packet3_ptr,0x00,50);
    memset(packet4_ptr,0x00,50);
 
    // Create an eight byte header for all the packets. The header is defined
    // in the record_tcp_php_config.txt file. 
    // Header definition: 2 byte big endian overall packet length
    //                    4 byte big endian unix time stamp
    //                    2 byte big endian mille time stamp
    
    // Populate the time zone
    if (big_endian_order == 1)
    {
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        memmove(packet1_ptr+2, (unsigned char*)&tim,4);
        memmove(packet1_ptr+6, (unsigned char*)&current_time.millitm,2);
        // 
        #ifdef __linux__
        sleep(1);
        #else
        Sleep(1000);
        #endif
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        memmove(packet2_ptr+2, (unsigned char*)&tim,4);
        memmove(packet2_ptr+6, (unsigned char*)&current_time.millitm,2);
        // 
        #ifdef __linux__
        sleep(1);
        #else
        Sleep(1000);
        #endif
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        memmove(packet3_ptr+2, (unsigned char*)&tim,4);
        memmove(packet3_ptr+6, (unsigned char*)&current_time.millitm,2);
        // 
        #ifdef __linux__
        sleep(1);
        #else
        Sleep(1000);
        #endif
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        memmove(packet4_ptr+2,(unsigned char*)&tim,4);
        memmove(packet4_ptr+6,(unsigned char*)&current_time.millitm,2);
    }
    else
    {
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        ByteSwap4Bytes((unsigned char*)&tim);
        memmove(packet1_ptr + 2, (unsigned char*)&tim, 4);
        memmove(packet1_ptr+6, (unsigned char*)&current_time.millitm+1,1);
        memmove(packet1_ptr+7, (unsigned char*)&current_time.millitm,1);
        // 
        #ifdef __linux__
        sleep(1);
        #else
        Sleep(1000);
        #endif
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        ByteSwap4Bytes((unsigned char*)&tim);
        memmove(packet2_ptr + 2, (unsigned char*)&tim, 4);
        memmove(packet2_ptr+6, (unsigned char*)&current_time.millitm+1,1);
        memmove(packet2_ptr+7, (unsigned char*)&current_time.millitm,1);
        // 
        #ifdef __linux__
        sleep(1);
        #else
        Sleep(1000);
        #endif
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        ByteSwap4Bytes((unsigned char*)&tim);
        memmove(packet3_ptr + 2, (unsigned char*)&tim, 4);
        memmove(packet3_ptr+6, (unsigned char*)&current_time.millitm+1,1);
        memmove(packet3_ptr+7, (unsigned char*)&current_time.millitm,1);
        // 
        #ifdef __linux__
        sleep(1);
        #else
        Sleep(1000);
        #endif
        ftime(&current_time);
        tim = (unsigned int)current_time.time;
        ByteSwap4Bytes((unsigned char*)&tim);
        memmove(packet4_ptr + 2, (unsigned char*)&tim, 4);
        memmove(packet4_ptr+6, (unsigned char*)&current_time.millitm+1,1);
        memmove(packet4_ptr+7, (unsigned char*)&current_time.millitm,1);
    }
 
    // Populate packet data zone 
    strcpy((char *)packet1_ptr+8,"Mary had a little lamb");
    strcpy((char *)packet2_ptr+8,"Its feet were black as soot");
    strcpy((char *)packet3_ptr+8,"And into Mary's bread and jam ");
    strcpy((char *)packet4_ptr+8,"Its sooty foot it put");
 
    // Get packet length including 5 byte header and null terminator.
    *length1_ptr = (unsigned short)strlen(packet1_ptr+8) + 8;
    *length2_ptr = (unsigned short)strlen(packet2_ptr+8) + 8;
    *length3_ptr = (unsigned short)strlen(packet3_ptr+8) + 8;
    *length4_ptr = (unsigned short)strlen(packet4_ptr+8) + 8;
 
    // Populate headers. 
 
    // Populate packet length field. 
    if (big_endian_order == 1)
    {
        memmove(packet1_ptr,length1_ptr,2);
        memmove(packet2_ptr,length2_ptr,2);
        memmove(packet3_ptr,length3_ptr,2);
        memmove(packet4_ptr,length4_ptr,2);
    }
    else
    {
        memmove(packet1_ptr+1,length1_ptr,1);
        memmove(packet2_ptr+1,length2_ptr,1);
        memmove(packet3_ptr+1,length3_ptr,1);
        memmove(packet4_ptr+1,length4_ptr,1);
    }
 
    return;
}
 
 
// ////////////////////////////////////////////////////////////////////////////
//
// Sample program sends and receives four packets via two TCP sockets and 
// records the packets in two different record files. The record library
// device is configured using the record_tcp_config.txt file and the
// record_tcp_php_config.txt file. The record library device moves all 
// closed record files to a simulated CFDP dropbox.  Program uses 
// RegisterPrintMessage, StartLoggingMessages, StopLoggingMessages, 
// InitToolkitRecordDevice, CreateTCPClientSocketDevice, CreateDeviceKeyAlias, 
// ConnectSocketDevice, RegisterRecordDeviceData, SendPacket, 
// CloseAllRecordFiles and DSCleanUp API functions.
//
// ////////////////////////////////////////////////////////////////////////////
 
int main(int argc, char *argv[])
{ 
    char ip_address[50];
    unsigned short source_port;
    char source1_device_key[50];
    char source1_device_name[50];
    char source2_device_key[50];
    char source2_device_name[50];
    unsigned int receive_queue_size;
    unsigned int receive_buffer_size;
    unsigned char packet1[50];
    unsigned char packet2[50];
    unsigned char packet3[50];
    unsigned char packet4[50];
    unsigned short length1;
    unsigned short length2;
    unsigned short length3;
    unsigned short length4;
    unsigned int key_buffer_size;
    char pathname[256];
    unsigned int connection_timeout_msec = 2100;
 
    // Set default record library config_pathname 
    strcpy (pathname, "./record_tcp_config.txt");
    if (argc == 2)
    {
        strcpy(pathname,argv[1]);
    }
    else if (argc > 2)
    {
        printf("Error    Command line contains too many arguments.\n");
        return 0;
    }
 
    // Register the PrintMessage callback function prior to InitToolkitCfdp 
    // to process error messages that may be generated during initialization.
    RegisterMessage(&PrintTheMessage);
 
    printf("\nRecord TCP\n\n");
 
    printf("Create four data packets.\n");
    
    // Create data packets.
    CreateDataPackets(packet1,
                      packet2,
                      packet3,
                      packet4,
                      &length1,
                      &length2,
                      &length3,
                      &length4);
 
    // Initialize the record library device. This funcition creates, 
    // connects and configures the sockets associated with 
    // record library
 
    if (InitToolkitRecordDevice(pathname) == SUCCESS)
    {
        // Create a TCP client 1 source socket device 
        strcpy(ip_address,"127.0.0.1");  // Use loopback address
        source_port = 0; 
        // A receive queue is not required since this socket is only 
        // sending packets 
        receive_queue_size = 0;
        receive_buffer_size = 0;
        key_buffer_size = sizeof(source1_device_key);
 
        if (CreateTCPClientSocketDevice(ip_address,
                                        source_port,
                                        receive_queue_size,
                                        receive_buffer_size,
                                        &key_buffer_size,
                                        source1_device_key) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
 
        // Define a user friendly device name and associate the name with a device 
        // key.  The device name is an alias for the device key and may 
        // be used in functions requiring a device key.  The name must be unique 
        // for all devices.  A device may not have multiple names.            
        strcpy(source1_device_name,"TCP_client1");
        if (CreateDeviceKeyAlias(source1_device_key,
                                 source1_device_name) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
 
        // Create a TCP client 2 source socket device 
        strcpy(ip_address,"127.0.0.1");  // Use loopback address
        source_port = 0; 
        // A receive queue is not required since this socket is only 
        // sending packets 
        receive_queue_size = 0;
        receive_buffer_size = 0;
        key_buffer_size = sizeof(source2_device_key);
 
        if (CreateTCPClientSocketDevice(ip_address,
                                        source_port,
                                        receive_queue_size,
                                        receive_buffer_size,
                                        &key_buffer_size,
                                        source2_device_key) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
 
        // Define a user friendly device name and associate the name with a device 
        // key.  The device name is an alias for the device key and may 
        // be used in functions requiring a device key.  The name must be unique 
        // for all devices.  A device may not have multiple names.            
        strcpy(source2_device_name,"TCP_client2");
        if (CreateDeviceKeyAlias(source2_device_key,
                                 source2_device_name) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
    
    
        // Connect the source or client socket to the recieve or listener socket.  
        // The listener socket will accept the client socket connection and create 
        // a server socket that is connected to the client socket. 
 
        // NOTE: The ip address and port number are found in the record
        // library device configuration file.
        if (ConnectSocketDevice(source1_device_name,
                                ip_address,
                                9100,
                                connection_timeout_msec) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
 
        // Connect the source or client socket to the recieve or listener socket.  
        // The listener socket will accept the client socket connection and create 
        // a server socket that is connected to the client socket.
 
        // NOTE: The ip address and port number are found in the record
        // library device configuration file.
        if (ConnectSocketDevice(source2_device_name,
                                ip_address,
                                9101,
                                connection_timeout_msec) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
 
        // Associate the DeviceData callback function with the record 
        // device library. The function will print a message when a 
        // record file is closed and renamed by the record library device.
 
        if (RegisterRecordDeviceData(&DeviceData) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
 
        // Send packet packet1 using TCP_client1 
        if (SendPacket (source1_device_name,length1,packet1) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
        // Send packet packet2 using TCP_client1
        if (SendPacket (source1_device_name,length2,packet2) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
        // Send packet packet3 using TCP_client2
        if (SendPacket (source2_device_name,length3,packet3) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
        // Send packet packet4 using TCP_client2
        if (SendPacket (source2_device_name,length4,packet4) != SUCCESS)
        {
            DSCleanUp();
            return 0;
        }
        // Allow time to print and log messages.
        #ifdef __linux__
        sleep(1);
        #else
        Sleep(1000);
        #endif
        // Close all record files using record device api
        if (CloseAllRecordFiles() != SUCCESS)
        {
            printf("Error    Closing all record files.\n");
        }
    }
 
    // Allow time to print and log messages.
    #ifdef __linux__
    sleep(1);
    #else
    Sleep(1000);
    #endif
 
    // Close the log file. A timetag is appended to the log filename. 
    StopLoggingMessages(); 
    DSCleanUp();
    return 0;
}