C++程序  |  86行  |  2.09 KB

//===-- RNBSocket.h ---------------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
//  Created by Greg Clayton on 12/12/07.
//
//===----------------------------------------------------------------------===//

#ifndef __RNBSocket_h__
#define __RNBSocket_h__

#include "RNBDefs.h"
#include <sys/socket.h>
#include <sys/types.h>
#include <string>
#include "DNBTimer.h"

#ifdef WITH_LOCKDOWN
#include "lockdown.h"
#endif

class RNBSocket
{
public:
    typedef void (*PortBoundCallback) (const void *baton, in_port_t port);

    RNBSocket () :
        m_fd (-1),
#ifdef WITH_LOCKDOWN
        m_fd_from_lockdown (false),
        m_ld_conn (),
#endif
        m_timer (true)      // Make a thread safe timer
    {
    }
    ~RNBSocket (void)
    {
        Disconnect (false);
    }

    rnb_err_t Listen (const char *listen_host,
                      in_port_t port,
                      PortBoundCallback callback,
                      const void *callback_baton);
    rnb_err_t Connect (const char *host, uint16_t port);

    rnb_err_t useFD(int fd);

#ifdef WITH_LOCKDOWN
    rnb_err_t ConnectToService();
#endif
    rnb_err_t OpenFile (const char *path);
    rnb_err_t Disconnect (bool save_errno);
    rnb_err_t Read (std::string &p);
    rnb_err_t Write (const void *buffer, size_t length);

    bool IsConnected () const { return m_fd != -1; }
    void SaveErrno (int curr_errno);
    DNBTimer& Timer() { return m_timer; }

    static int SetSocketOption(int fd, int level, int option_name, int option_value);
private:
    // Outlaw some constructors
    RNBSocket (const RNBSocket &);

protected:
    rnb_err_t ClosePort (int& fd, bool save_errno);

    int m_fd;    // Socket we use to communicate once conn established

#ifdef WITH_LOCKDOWN
    bool m_fd_from_lockdown;
    lockdown_connection m_ld_conn;
#endif

    DNBTimer m_timer;
};


#endif // #ifndef __RNBSocket_h__