Subversion Repositories seema-scanner

/*****************************************************************************
**             TEXAS INSTRUMENTS PROPRIETARY INFORMATION
**
**  (c) Copyright, Texas Instruments Incorporated, 2012-2013.
**      All Rights Reserved.
**
**  Property of Texas Instruments Incorporated. Restricted Rights -
**  Use, duplication, or disclosure is subject to restrictions set
**  forth in TI's program license agreement and associated documentation.
******************************************************************************/
/**
*
* @file    tcp_client.c
*
* @brief        TCP implementation using <system supported function > APIs.
**/
/*****************************************************************************/


/*Include header files related to TCP/IP communication */
//#ifdef __WIN32__
//#include "stdafx.h"
//#endif

#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif


#include <stdio.h>
#include <stdlib.h>

#ifdef WIN32
    #include <windows.h>
    #include <winsock2.h>
    #include <ws2tcpip.h>
    #include <iphlpapi.h>
    #pragma comment(lib, "Ws2_32.lib")
#else

    #include <netdb.h>
    #include <unistd.h>
    typedef int SOCKET;
    #define INVALID_SOCKET -1
    #define SOCKET_ERROR   -1
    #define closesocket(s) close(s)
#endif


#include "tcp_client.h"


int TCP_Connect(char *host, unsigned long int port)
{

        //int socket = 0;
        //1. Declare and Initialize TCP/IP module
        //2. Create a socket for connecting to server
        //3. Connect to server.
        //return socket;

        int iResult;
        char charPort[33];

        //Convert the port number into ascii char 
    //_ultoa(port,&charPort[0],10);
    sprintf(charPort, "%ld", port);

    #ifdef WIN32
        WSADATA wsaData;
        // Initialize Winsock
        iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
        if (iResult != 0) {
            printf("WSAStartup failed: %d\n", iResult);
            return -1;
        }
    #endif

        //Declare addrinfo object that contains a socketaddr strcuture and initialize these values
        struct addrinfo *result = NULL,
                *ptr = NULL,
        hints = {};
    //ZeroMemory( &hints, sizeof(hints) );
        hints.ai_family = AF_UNSPEC;
        hints.ai_socktype = SOCK_STREAM;
        hints.ai_protocol = IPPROTO_TCP;

        // Resolve the server address and port
    iResult = getaddrinfo(host, charPort, &hints, &result);
        if (iResult != 0) {
                printf("getaddrinfo failed: %d\n", iResult);
        #ifdef WIN32
            WSACleanup();
            return -1;
        #endif
        }

        // Attempt to connect to the first address returned by
        // the call to getaddrinfo
        ptr=result;

        // Create a SOCKET for connecting to server
        SOCKET ConnectSocket = INVALID_SOCKET;

        ConnectSocket = socket(ptr->ai_family, ptr->ai_socktype, 
                ptr->ai_protocol);

        if (ConnectSocket == INVALID_SOCKET) {
        #ifdef WIN32
            printf("Error at socket(): %ld\n", WSAGetLastError());
            WSACleanup();
        #endif
        freeaddrinfo(result);
                return -1;
        }


        // Connect to server.
        iResult = connect( ConnectSocket, ptr->ai_addr, (int)ptr->ai_addrlen);
        if (iResult == SOCKET_ERROR) {
                closesocket(ConnectSocket);
                ConnectSocket = INVALID_SOCKET;
        }

        // Should really try the next address returned by getaddrinfo
        // if the connect call failed
        // But for this simple example we just free the resources
        // returned by getaddrinfo and print an error message

        freeaddrinfo(result);

        if (ConnectSocket == INVALID_SOCKET) {
                printf("Unable to connect to server!\n");
        #ifdef WIN32
            WSACleanup();
        #endif
                return -1;
        }

        return ConnectSocket;
}

int TCP_Send(int sock, unsigned char *buffer, int length)
{
        // Send 'length' number of bytes from buffer via provided
        // socket <sock> address
        int iResult;
        iResult = send(sock,(char*) buffer,length,0);
        if (iResult == SOCKET_ERROR) {
        #ifdef WIN32
            printf("send failed with error: %d\n", WSAGetLastError());
        #endif
                closesocket(sock);
        #ifdef WIN32
            WSACleanup();
        #endif
                return 1;
        }

        //printf("Bytes Sent: %ld\n", iResult);
        return 0;
}

int TCP_Receive(int sock, unsigned char *buffer, int length)
{
        //Retrieve 'length' number of bytes into 'buffer' from the socket <sock> address
        int iResult = 0;

        while(length)
        {
                iResult = recv(sock, (char*) buffer, length, 0);

                if ( iResult > 0 )
                        ; //printf("Bytes received: %d\n", iResult);
                else if ( iResult == 0 )
                        printf("Connection closed\n");
        else {
            #ifdef WIN32
                printf("recv failed with error: %d\n", WSAGetLastError());
            #endif
        }
                length -= iResult;
                buffer += iResult;
        }

        return 0;
}

int TCP_Disconnect(int sock)
{
        //free and cleanup socket occupied memory
        closesocket(sock);
    #ifdef WIN32
        WSACleanup();
    #endif
        return 0;
}