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 THE COPYRIGHT OWNER 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 Connection_h 27 #define Connection_h 28 29 #include <unistd.h> 30 #include <sys/socket.h> 31 32 namespace android { 33 34 namespace WDS { 35 36 class Socket { 37 public: Socket()38 Socket(): m_fd(-1) {} Socket(int fd)39 Socket(int fd): m_fd(fd) {} ~Socket()40 ~Socket() { 41 if (m_fd != -1) { 42 shutdown(m_fd, SHUT_RDWR); 43 close(m_fd); 44 } 45 } 46 // Open a new socket using PF_INET and SOCK_STREAM 47 bool open(); fd()48 int fd() const { return m_fd; } 49 private: 50 int m_fd; 51 }; 52 53 class Connection { 54 public: Connection(int conn)55 Connection(int conn): m_socket(conn) {} read(char buf[],size_t length)56 int read(char buf[], size_t length) const { 57 return recv(m_socket.fd(), buf, length, 0); 58 } write(const char buf[],size_t length)59 int write(const char buf[], size_t length) const { 60 return send(m_socket.fd(), buf, length, 0); 61 } write(const char buf[])62 int write(const char buf[]) const { 63 return write(buf, strlen(buf)); 64 } 65 private: 66 Socket m_socket; 67 }; 68 69 class ConnectionServer { 70 public: ConnectionServer()71 ConnectionServer() {} 72 73 // Establish a connection to the local host on the given port. 74 bool connect(short port); 75 76 // Blocks on the established socket until a new connection arrives. 77 Connection* accept() const; 78 private: 79 Socket m_socket; 80 }; 81 82 } // end namespace WDS 83 84 } // end namespace android 85 86 #endif 87