1 /*
2 * Copyright (C) 2011 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16 #ifndef __SOCKET_STREAM_H
17 #define __SOCKET_STREAM_H
18 
19 #include <stdlib.h>
20 #include "IOStream.h"
21 
22 class SocketStream : public IOStream {
23 public:
24     typedef enum { ERR_INVALID_SOCKET = -1000 } SocketStreamError;
25 
26     explicit SocketStream(size_t bufsize = 10000);
27     virtual ~SocketStream();
28 
29     virtual int listen(unsigned short port) = 0;
30     virtual SocketStream *accept() = 0;
31     virtual int connect(unsigned short port) = 0;
32 
33     virtual void *allocBuffer(size_t minSize);
34     virtual int commitBuffer(size_t size);
35     virtual const unsigned char *readFully(void *buf, size_t len);
36     virtual const unsigned char *commitBufferAndReadFully(size_t size, void *buf, size_t len);
37     virtual const unsigned char *read(void *buf, size_t *inout_len);
38 
valid()39     bool valid() { return m_sock >= 0; }
40     virtual int recv(void *buf, size_t len);
41     virtual int writeFully(const void *buf, size_t len);
42 
43 protected:
44     int            m_sock;
45     size_t         m_bufsize;
46     unsigned char *m_buf;
47 
48     SocketStream(int sock, size_t bufSize);
49 };
50 
51 #endif /* __SOCKET_STREAM_H */
52