|| || || || || || || ||

---

mainFileServer.cpp

Go to the documentation of this file.

#include "Basis/CmdOptions.h"
#include "Util/ChannelServer.h"
#include "Util/Channel.h"
#include "Util/IOBufferFile.h"

namespace Impala
{
namespace Application
{


using namespace Util;

/******************************************************************
 Instances of this class serve files to clients.
*******************************************************************/
class FileServer : public ChannelServer
{

public: 

    FileServer(int port, int nrPorts, CString passwordFile)
        : ChannelServer(port, nrPorts, false, passwordFile)
    {
        Init();
    }

protected:

    virtual int
    AcceptRequest(char* buf, int len, int bufSize, CString conn, int port)
    {
        String curCon = ConnectionDescr();

        int rcBase = ChannelServer::AcceptRequest(buf, len, bufSize, conn, port);
        if (rcBase >= 0)
            return rcBase;

        else if (strncmp(buf, "get ", 4) == 0)
        {
            
            ILOG_DEBUG("Doing get");
            String fileName(buf+4);
            
            ILOG_DEBUG("name : [" << fileName << "]");
            mIoBuf = new IOBufferFile(fileName, true, false);
            if (!mIoBuf->Valid())
            {
                delete mIoBuf;
                mIoBuf = 0;
                sprintf(buf, "ERROR : unable to open file %s\0",
                        fileName.c_str());
                return strlen(buf)+1;
            }
            int nrToSend = Min<Int64>(Channel::DATA_BUFFER_SIZE, mIoBuf->Size());
            mIoBuf->Read(buf, nrToSend);
            if (nrToSend != Channel::DATA_BUFFER_SIZE)
            {
                delete mIoBuf;
                mIoBuf = 0;
            }
            return nrToSend;
        }

        else if (strncmp(buf, "getmore", 7) == 0)
        {
            
            ILOG_DEBUG("Doing getmore");
            if (mIoBuf == 0)
            {
                sprintf(buf, "ERROR : no open file\0");
                return strlen(buf)+1;
            }
            int nrToSend = Min<Int64>(Channel::DATA_BUFFER_SIZE,
                                      mIoBuf->Available());
            mIoBuf->Read(buf, nrToSend);
            if (nrToSend != Channel::DATA_BUFFER_SIZE)
            {
                delete mIoBuf;
                mIoBuf = 0;
            }
            return nrToSend;
        }

        else if (!AllowWrite(port))
        {
            sprintf(buf, "ERROR : permission denied\0");
            return strlen(buf)+1;
        }

        else if (strncmp(buf, "put ", 4) == 0)
        {
            
            ILOG_DEBUG("Doing put");
            String fileName(buf+4);
            
            ILOG_DEBUG("name : [" << fileName << "]");
            mIoBuf = new IOBufferFile(fileName, false, false);
            if (!mIoBuf->Valid())
            {
                delete mIoBuf;
                mIoBuf = 0;
                sprintf(buf, "ERROR : unable to open file %s\0",
                        fileName.c_str());
                return strlen(buf)+1;
            }
            sprintf(buf, "OK\0");
            return strlen(buf)+1;
        }

        else if (strncmp(buf, "putmore", 7) == 0)
        {
            
            ILOG_DEBUG("Doing putmore");
            if (mIoBuf == 0)
            {
                sprintf(buf, "ERROR : no open file\0");
                return strlen(buf)+1;
            }
            mIoBuf->Write(buf+8, len-8);
            if (len != Channel::DATA_BUFFER_SIZE)
            {
                delete mIoBuf;
                mIoBuf = 0;
            }
            sprintf(buf, "OK\0");
            return strlen(buf)+1;
        }

        else
            return -1;
    }


private:

    void 
    Init()
    {
        mIoBuf = 0;
    }

    IOBufferFile* mIoBuf;

    ILOG_VAR_DEC;

}; // class

ILOG_VAR_INIT(FileServer, Impala.Application);

int
mainFileServer(int argc, char* argv[])
{
    CmdOptions& options = CmdOptions::GetInstance();
    options.Initialise(true, true, true);
    // multiple ports require a lot more bookkeeping for multipart messages
    //if (! options.ParseArgs(argc, argv, "port [nrPorts]", 1))
    if (! options.ParseArgs(argc, argv, "port", 1))
        return 1;

    int port = atol(options.GetArg(0));

    //if (options.GetNrArg() > 1)
    //    nrPorts = atol(options.GetArg(1));
    int nrPorts = 1;

    String passwordFile = options.GetString("passwordFile");
    Application::FileServer fileServer(port, nrPorts, passwordFile);
    bool doIdle = false;
    fileServer.Start(doIdle);

    return 0;
}

} // namespace Application
} // namespace Impala

int
main(int argc, char* argv[])
{
    return Impala::Application::mainFileServer(argc, argv);
}

---