00001 /*************************************************************************** 00002 ** 00003 ** This file is part of GpCoreTools. 00004 ** 00005 ** This file may be distributed and/or modified under the terms of the 00006 ** GNU General Public License version 2 or 3 as published by the Free 00007 ** Software Foundation and appearing in the file LICENSE.GPL included 00008 ** in the packaging of this file. 00009 ** 00010 ** This file is distributed in the hope that it will be useful, but WITHOUT 00011 ** ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 00012 ** FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for 00013 ** more details. 00014 ** 00015 ** You should have received a copy of the GNU General Public License 00016 ** along with this program. If not, see <http://www.gnu.org/licenses/>. 00017 ** 00018 ** See http://www.geopsy.org for more information. 00019 ** 00020 ** Created: 2009-08-19 00021 ** Authors: 00022 ** Marc Wathelet 00023 ** Marc Wathelet (LGIT, Grenoble, France) 00024 ** 00025 ***************************************************************************/ 00026 00027 #ifndef TCPCLIENTSTREAM_H 00028 #define TCPCLIENTSTREAM_H 00029 00030 #include <poll.h> 00031 #include <unistd.h> 00032 #include <stdlib.h> 00033 00034 #include "Stream.h" 00035 #include "Address.h" 00036 #include "GpCoreToolsDLLExport.h" 00037 00038 namespace GpCoreTools { 00039 00040 class DynamicBuffer; 00041 00042 class GPCORETOOLS_EXPORT TcpClientStream : public Stream 00043 { 00044 public: 00045 TcpClientStream(int fileDescriptor, const Address& peer); 00046 TcpClientStream(const Address& peer); 00047 virtual ~TcpClientStream(); 00048 00049 virtual bool connect(uint16_t port); 00050 00051 virtual short eventTypes() {return POLLIN | POLLRDHUP;} 00052 virtual void event(short type); 00053 virtual void hungUp() {} 00054 00055 const Address& peer() const {return _peer;} 00056 DynamicBuffer * buffer() const {return _buffer;} 00057 00058 bool writePartial(const char * bytes, int byteCount) const { 00059 return ::send(fileDescriptor(), bytes, byteCount, MSG_MORE)==byteCount; 00060 } 00061 bool writePartialNoGateway(const char * bytes, int byteCount) const { 00062 return ::send(fileDescriptor(), bytes, byteCount, MSG_DONTROUTE | MSG_MORE)==byteCount; 00063 } 00064 bool write(const char * bytes, int byteCount) const { 00065 return ::write(fileDescriptor(), bytes, byteCount)==byteCount; 00066 } 00067 bool writeNoGateway(const char * bytes, int byteCount) const { 00068 return ::send(fileDescriptor(), bytes, byteCount, MSG_DONTROUTE)==byteCount; 00069 } 00070 protected: 00071 void setBuffer(DynamicBuffer * buffer) {_buffer=buffer;} 00072 private: 00073 Address _peer; 00074 DynamicBuffer * _buffer; 00075 }; 00076 00077 } // namespace GpCoreTools 00078 00079 #endif // TCPCLIENTSTREAM_H
