RNBSocket.h revision a619e8ec81ff9c838285105c034ee70a290be904
1d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org//===-- RNBSocket.h ---------------------------------------------*- C++ -*-===// 2d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// 3d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// The LLVM Compiler Infrastructure 4d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// 5d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// This file is distributed under the University of Illinois Open Source 6d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// License. See LICENSE.TXT for details. 7d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// 87f9bebe3906fb30f0632f9d010754d12dd6c02d7yusukes@chromium.org//===----------------------------------------------------------------------===// 9d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// 10cd35d0324e9cb7d46625ea1c0990dc75c69fc835bashi@chromium.org// Created by Greg Clayton on 12/12/07. 11d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org// 12d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org//===----------------------------------------------------------------------===// 13d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 14d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#ifndef __RNBSocket_h__ 15d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#define __RNBSocket_h__ 16d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 17d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#include "RNBDefs.h" 18d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#include <sys/socket.h> 19d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#include <sys/types.h> 20d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#include <string> 21d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#include "DNBTimer.h" 22d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 23d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#ifdef WITH_LOCKDOWN 24d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#include "lockdown.h" 25d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#endif 26d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 27d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.orgclass RNBSocket 28d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org{ 29d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.orgpublic: 30d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org typedef void (*PortBoundCallback) (const void *baton, in_port_t port); 31d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 32d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org RNBSocket () : 33d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org m_fd (-1), 34d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#ifdef WITH_LOCKDOWN 35d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org m_fd_from_lockdown (false), 36d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org m_ld_conn (), 37d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#endif 38d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org m_timer (true) // Make a thread safe timer 39d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org { 40d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org } 41d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org ~RNBSocket (void) 42d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org { 43d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org Disconnect (false); 44d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org } 45d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 46d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t Listen (const char *listen_host, 47d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org in_port_t port, 48d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org PortBoundCallback callback, 49d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org const void *callback_baton); 50d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t Connect (const char *host, uint16_t port); 51d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 52d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t useFD(int fd); 53d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 54d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#ifdef WITH_LOCKDOWN 55d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t ConnectToService(); 56d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#endif 57d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t OpenFile (const char *path); 58d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t Disconnect (bool save_errno); 59d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t Read (std::string &p); 60d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t Write (const void *buffer, size_t length); 61d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 62d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org bool IsConnected () const { return m_fd != -1; } 63d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org void SaveErrno (int curr_errno); 64d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org DNBTimer& Timer() { return m_timer; } 65d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 66d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org static int SetSocketOption(int fd, int level, int option_name, int option_value); 67d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.orgprivate: 68d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org // Outlaw some constructors 69d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org RNBSocket (const RNBSocket &); 70d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 71d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.orgprotected: 72d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org rnb_err_t ClosePort (int& fd, bool save_errno); 73d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 74d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org int m_fd; // Socket we use to communicate once conn established 75d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 76d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#ifdef WITH_LOCKDOWN 77d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org bool m_fd_from_lockdown; 78d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org lockdown_connection m_ld_conn; 79d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#endif 80d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 81d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org DNBTimer m_timer; 82d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org}; 83d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 84d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org 85d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org#endif // #ifndef __RNBSocket_h__ 86d257d186ae2a08042a412824678f98241a1a4f3cyusukes@chromium.org