/*
 ============================================================================
 Name		 : Dummy3rdPartyApp.cpp
 Author	     : Gurpreet Singh Nagi
 Copyright   : Your copyright notice
 Description : Exe source file
 ============================================================================
 */

//  Include Files  

#include "Dummy3rdPartyApp.h"
#include <e32base.h>
#include <e32std.h>
#include <e32cons.h>			// Console

//  Constants
_LIT(KTextConsoleTitle, "Dummy3rdPartyApp");
_LIT(KTextFailed, " failed, leave code = %d");
_LIT(KTextPressAnyKey, " [press any key]\n");

#define BUF_SIZE 2048
//  Global Variables
LOCAL_D CConsoleBase* console; // write all messages to this


//  Local Functions
void Dummy3rdPartyApp::InitConn()
    {
    if((iStatus = iVirtSocketServ.Connect()) == KErrNone )
        console->Printf(_L("Opening Socket Server on Virtual Interface.\n"));

    if((iStatus = iRealSocketServ.Connect()) == KErrNone )
        console->Printf(_L("Opening Socket Server on Real Interface.\n"));
    
    if((iStatus = iHookSocketServ.Connect()) == KErrNone )
            console->Printf(_L("Opening Socket Server for Hook.\n"));

    if((iStatus = iVirtConnection.Open(iVirtSocketServ, KAfInet)) == KErrNone )
        console->Printf(_L("\nOpening Virtual Connection."));

    if((iStatus = iRealConnection.Open(iRealSocketServ, KAfInet)) == KErrNone )
        console->Printf(_L("\nOpening Real Connection.\n"));
    
    iHookSocket.Open(iHookSocketServ,_L("tun"));
    
    }

void Dummy3rdPartyApp::StartConn()
    {
    iVirtCommDbPref.SetIapId(14);             // Dummy Interface
    iVirtCommDbPref.SetBearerSet(KCommDbBearerVirtual);
    iVirtCommDbPref.SetDialogPreference(ECommDbDialogPrefDoNotPrompt);

    TInt err = KErrNone;
    iVirtConnection.Start(iVirtCommDbPref, iStatus);
    User::WaitForRequest(iStatus);
    err = iStatus.Int();
    if(iStatus == KErrNone)
        console->Printf(_L("Connection Started on Virtual Interface.\n"));
    User::LeaveIfError(err);
    err = KErrNone;
    //open socket on Virtual interface.
    iStatus = iVirtSocket.Open(iVirtSocketServ, KAfInet, KSockDatagram, KProtocolInetUdp, iVirtConnection);
    //if((iStatus = iVirtSocket.Open(iVirtSocketServ, KAfInet, KSockDatagram, KProtocolInetUdp, iVirtConnection)) == KErrNone )
    if(iStatus == KErrNone)
        console->Printf(_L("Socket Opened on Virtual Interface.\n"));

    iRealCommDbPref.SetIapId(9);            //Real Interface i.e. ethernet
    iRealCommDbPref.SetBearerSet(KCommDbBearerLAN);
    iRealCommDbPref.SetDialogPreference(ECommDbDialogPrefDoNotPrompt);

    iRealConnection.Start(iRealCommDbPref, iStatus);
    User::WaitForRequest(iStatus);
    if(iStatus == KErrNone)
        console->Printf(_L("Connection Started on Real Interface.\n"));
    err = iStatus.Int();
    //open socket on real interface
    if((iStatus = iRealSocket.Open(iRealSocketServ, KAfInet, KSockDatagram, KProtocolInetUdp, iRealConnection)) == KErrNone )
        console->Printf(_L("Socket Opened on Real Interface.\n"));
    User::LeaveIfError(err);

    //set destination port and IP ie. OpenVPN gateway address
    iRealAddr.SetV4MappedAddress(INET_ADDR(10,225,171,71));        
    iRealAddr.SetPort(1194);
    err = KErrNone;
    //bind local port to socket
    // To Get any available port from the kernel
    
    TInetAddr inetAddrBind;
    inetAddrBind.SetPort(KInetPortAny);
    inetAddrBind.SetAddress(KInetAddrAny);
    iVirtSocket.Bind(inetAddrBind);
    TUint port1  = iVirtSocket.LocalPort();
      
    TPckgC <TUint> portPckg (port1);
    iHookSocket.SetOpt(KSolInetIp,KSoTunnelPort,portPckg);

    console->Printf(_L("Socket bound on Virtual Interface.\n"), err);

    err = iRealSocket.SetLocalPort(1194); 
    console->Printf(_L("Socket bound on Real Interface.\n"), err);
     
    }


void Dummy3rdPartyApp::ProcessData()
    {
    //Receive data from client on virtual interface
    iVirtSocket.RecvFrom(iSendBuf, iVirtAddr, NULL, iStatus);
    User::WaitForRequest(iStatus);

    if(iStatus == KErrNone)
        {
        console->Printf(_L("\nData Received from Virtual Interface:  "));
        //copy received data to temp buffer and print it
        //TBuf<BUF_SIZE> buf;
        //buf.Copy(iSendBuf);
        //console->Printf(buf);
        }
    else
        console->Printf(_L("\nData receive failed on Virtual Interface."));     

    //Send data on real interface to OpenVPN Gateway.
    iRealSocket.SendTo(iSendBuf, iRealAddr, NULL, iStatus);
    User::WaitForRequest(iStatus);  

    if( iStatus == KErrNone)
        {
        console->Printf(_L("\nData Sent on Real interface : "));

        //display sent data
        //TBuf<BUF_SIZE> buf;
        //buf.Copy(iSendBuf);
        //console->Printf(buf);
        }
    else
        console->Printf(_L("\nData send failed on Real Interface"));   

    //Receive data from from real interface
    iRealSocket.RecvFrom(iRecvBuf, iRealAddr, NULL, iStatus);
    User::WaitForRequest(iStatus);

    if(iStatus == KErrNone)
        {
        console->Printf(_L("\nData Received from Real Interface:  "));
        //copy received data to temp buffer and print it
        //TBuf<BUF_SIZE> buf;
        //buf.Copy(iRecvBuf);
        //console->Printf(buf);
        }
    else
        console->Printf(_L("\nData receive failed on Real Interface"));     

    //Send data to virtual interface.
    iVirtSocket.SendTo(iRecvBuf, iVirtAddr, NULL, iStatus);
    User::WaitForRequest(iStatus);  

    if( iStatus == KErrNone)
        {
        console->Printf(_L("\nData Sent on Virtual interface : "));

        //display sent data
        //TBuf<BUF_SIZE> buf;
        //buf.Copy(iRecvBuf);
        //console->Printf(buf);
        }
    else
        console->Printf(_L("\nData send failed on Virtual Interface"));   
    }

void Dummy3rdPartyApp::CloseConn()
    {
    //Close all handles.
    iVirtSocket.Close();
    iRealSocket.Close();
    iHookSocket.Close();

    iVirtConnection.Close();
    iRealConnection.Close();
    iVirtSocketServ.Close();
    iRealSocketServ.Close();
    iHookSocketServ.Close();
    }

LOCAL_C void MainL()
    {
    Dummy3rdPartyApp obj;

    //intializing connection and socket
    obj.InitConn();
    //start the connection
    obj.StartConn();

    while(1)
        {
        obj.ProcessData();
        }
    //close connection, subconnection, socket
    //TODO: how to explicitly close the loop? 
    // One possible solution is running a timer.
    obj.CloseConn();
    }

LOCAL_C void DoStartL()
    {
    // Create active scheduler (to run active objects)
    CActiveScheduler* scheduler = new (ELeave) CActiveScheduler();
    CleanupStack::PushL(scheduler);
    CActiveScheduler::Install(scheduler);

    MainL();
    // Delete active scheduler
    CleanupStack::PopAndDestroy(scheduler);
    }

//  Global Functions

GLDEF_C TInt E32Main()
    {
    // Create cleanup stack
    __UHEAP_MARK;
    CTrapCleanup* cleanup = CTrapCleanup::New();

    // Create output console
    TRAPD(createError, console = Console::NewL(KTextConsoleTitle, TSize(
            KConsFullScreen, KConsFullScreen)));
    if (createError)
        return createError;

    // Run application code inside TRAP harness, wait keypress when terminated
    TRAPD(mainError, DoStartL());
    if (mainError)
        console->Printf(KTextFailed, mainError);
    console->Printf(KTextPressAnyKey);
    console->Getch();

    delete console;
    delete cleanup;
    __UHEAP_MARKEND;
    return KErrNone;
    }