1 /* 2 * Copyright 2008, The Android Open Source Project 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * * Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * * Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #ifndef WDS_CONNECTION_H 27 #define WDS_CONNECTION_H 28 29 #include <sys/socket.h> 30 31 namespace android { 32 33 namespace WDS { 34 35 class Socket { 36 public: Socket()37 Socket(): m_fd(-1) {} Socket(int fd)38 Socket(int fd): m_fd(fd) {} ~Socket()39 ~Socket() { 40 if (m_fd != -1) { 41 shutdown(m_fd, SHUT_RDWR); 42 close(m_fd); 43 } 44 } 45 // Open a new socket using PF_INET and SOCK_STREAM 46 bool open(); fd()47 int fd() const { return m_fd; } 48 private: 49 int m_fd; 50 }; 51 52 class Connection { 53 public: Connection(int conn)54 Connection(int conn): m_socket(conn) {} read(char buf[],size_t length)55 int read(char buf[], size_t length) const { 56 return recv(m_socket.fd(), buf, length, 0); 57 } write(const char buf[],size_t length)58 int write(const char buf[], size_t length) const { 59 return send(m_socket.fd(), buf, length, 0); 60 } write(const char buf[])61 int write(const char buf[]) const { 62 return write(buf, strlen(buf)); 63 } 64 private: 65 Socket m_socket; 66 }; 67 68 class ConnectionServer { 69 public: ConnectionServer()70 ConnectionServer() {} 71 72 // Establish a connection to the local host on the given port. 73 bool connect(short port); 74 75 // Blocks on the established socket until a new connection arrives. 76 Connection* accept() const; 77 private: 78 Socket m_socket; 79 }; 80 81 } // end namespace WDS 82 83 } // end namespace android 84 85 #endif 86