D:/Zythum/DinoKod/Network/FileTransfer.cpp

//---------------------------------------------------------------------------------------------
//      This file is a part of "DinoKod".
//      Copyright © 2003 Dino Productions. All Rights Reserved.
//      
//      File                    : FileTransfer.cpp
//      Author                  : Sebastien LEIX        sebastien.leix@wanadoo.fr
//      Date                    : 09/09/2002
//      Modification    :
//
//---------------------------------------------------------------------------------------------
#include "Common/Game.h"
#include "Common/Ini.h"
#include "Common/File.h"
#include "Common/Console.h"
#include "Common/Error.h"
#include "Common/Time.h"

#include "Network/FileTransfer.h"

//---------------------------------------------------------------------------------------------------------------------
KFTClientConnection::KFTClientConnection()
: KUDPClientSocket()
{
        m_pFile = NULL;
        m_Size  = 0;
}

//---------------------------------------------------------------------------------------------------------------------
KFTClientConnection::~KFTClientConnection()
{
        if( m_pFile )
        {
                m_pFile->Close();
                Deletep( m_pFile );
        }
}

//---------------------------------------------------------------------------------------------------------------------
void KFTClientConnection::OnMessage( KInetAddr& Addr, KNetMsg& Msg )
{
        KNFILETRANSFER          MsgType;

        Msg >> MsgType;

        switch( MsgType )
        {
        case KNFT_NEW:
                {
                        KStr    sString;

                        if( m_pFile )
                        {
                                m_pFile->Close();
                                Deletep( m_pFile );
                                m_Size = 0;
                        }

                        Msg.ReadString( sString );
                        g_Console << "FT : NEW FILE " << sString.GetpString() << KENDL;

                        sString += ".jpg" ;
                        m_pFile = new KFile();
                        if( !m_pFile->Create( sString.GetpString() ) )
                        {
                                KError::Warning( NULL, "KFTClientConnection::OnMessage() : Cannot create file [%s]", sString.GetpString() );
                                return;
                        }
                        return;
                }
        case KNFT_DATA:
                {
                        char    pBuffer[1024];
                        u32             Size;

                        Msg >> Size;
                        KASSERT( Size <= sizeof( pBuffer ) );
                        Msg.ReadBuffer( pBuffer, Size );
                        m_Size += Size;
                        g_Console << "FT : DATA " << m_Size << KENDL;
                        m_pFile->Write( (u8*)pBuffer, Size );
                        return;
                }
        case KNFT_CLOSE:
                {
                        if( m_pFile )
                        {
                                m_pFile->Close();
                                Deletep( m_pFile );
                                m_Size = 0;
                        }

                        g_Console << "FT : CLOSE " << KENDL;
                        
                        return;
                }
        default:
                KError::Warning( NULL, "KFTClientConnection::OnMessage() : Unknown _KNFILETRANSFER type %i", MsgType );
                return;
        }
}

//---------------------------------------------------------------------------------------------------------------------
KFileTransfer::KFileTransfer()
: KUDPServerSocket()
{
        m_bServer       = false;
        m_pFile         = NULL;
        m_LastTime      = 0;
}

//---------------------------------------------------------------------------------------------------------------------
KFileTransfer::~KFileTransfer()
{
}

//---------------------------------------------------------------------------------------------------------------------
bool KFileTransfer::Init( bool bServer )
{
        m_bServer = bServer;

        // Demarrage de  WINSOCK
        if( !KSocket::StartNetwork() )
                return false;

        // Server
        if( m_bServer )
        {
                u32             MaxClients      = g_pGame->GetpIniFile()->ReadInt( "NETWORK", "FTMaxClients", 32 );
                u16             ListenPort      = (u16)g_pGame->GetpIniFile()->ReadInt( "NETWORK", "FTListenPort", 28002 );
                
                SetMaxClients( MaxClients );

                if( !Create() )
                        return false;
                if( !Listen( KInetAddr( (char*)NULL, ListenPort ) ) )
                        return false;
        }

        // Client
        if( !m_ClientSocket.Create() )
                return false;

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
void KFileTransfer::End()
{
        // Server
        if( m_bServer )
        {
                FlushAllClients();
                Close();
        }

        // Client
        m_ClientSocket.Close();

        // Arret de WINSOCK
        KSocket::StopNetwork();
}

//---------------------------------------------------------------------------------------------------------------------
bool KFileTransfer::Manage()
{
        if( m_bServer )
        {
                if( !KUDPServerSocket::Manage() )
                        return false;
        }

        if( !m_ClientSocket.Manage() )
                return false;


        if( m_LastTime < g_Time.GetTime() )
        {
                m_LastTime = g_Time.GetTime() + 100;
                if( m_pFile )
                {
                        KNetMsg Msg;
                        char    pBuffer[240];
                        u32             Size = 0;

                        m_pFile->Read( (u8*)pBuffer, sizeof( pBuffer ), &Size );

                        if( Size )
                        {
                                Msg << KNFT_DATA << Size;
                                Msg.WriteBuffer( pBuffer, Size );
                                m_ClientSocket.SendMessage( Msg );
                        }

                        if( Size != sizeof( pBuffer ) )
                        {
                                Msg.Flush();
                                Msg << KNFT_CLOSE;
                                m_ClientSocket.SendMessage( Msg );

                                m_pFile->Close();
                                Deletep( m_pFile );
                        }
                }
        }

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
bool KFileTransfer::Connect( KStr pHostName )
{
        u16             ListenPort      = (u16)g_pGame->GetpIniFile()->ReadInt( "NETWORK", "FTListenPort", 28002 );

        if( !m_ClientSocket.Connect( KInetAddr( pHostName.GetpString(), ListenPort ) ) )
                return false;

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
bool KFileTransfer::PutFile( KStr sFileName )
{
        if( m_pFile )
        {
                m_pFile->Close();
                Deletep( m_pFile );
        }

        m_pFile = new KFile();
        if( !m_pFile->Open( sFileName ) )
        {
                Deletep( m_pFile );
                return false;
        }

        KNetMsg Msg;

        Msg << KNFT_NEW;
        Msg.WriteString( sFileName );
        m_ClientSocket.SendMessage( Msg );

        return true;
}

---