diff --git a/trunk/src/libs/srs_lib_simple_socket.cpp b/trunk/src/libs/srs_lib_simple_socket.cpp index 9a0f03d05..5d53d0a8d 100644 --- a/trunk/src/libs/srs_lib_simple_socket.cpp +++ b/trunk/src/libs/srs_lib_simple_socket.cpp @@ -23,9 +23,32 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. #include -SimpleSocketStream::SimpleSocketStream(int fd) { +#include + +#include +#include +#include + +SimpleSocketStream::SimpleSocketStream() +{ + fd = -1; } -SimpleSocketStream::~SimpleSocketStream() { +SimpleSocketStream::~SimpleSocketStream() +{ + if (fd != -1) { + ::close(fd); + fd = -1; + } +} + +int SimpleSocketStream::create_socket() +{ + fd = ::socket(AF_INET, SOCK_STREAM, 0); + if(fd == -1){ + return -1; + } + + return ERROR_SUCCESS; } diff --git a/trunk/src/libs/srs_lib_simple_socket.hpp b/trunk/src/libs/srs_lib_simple_socket.hpp index 4da261ae0..dd4ac73e8 100644 --- a/trunk/src/libs/srs_lib_simple_socket.hpp +++ b/trunk/src/libs/srs_lib_simple_socket.hpp @@ -33,10 +33,13 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ class SimpleSocketStream { +private: + int fd; public: - SimpleSocketStream(int fd); + SimpleSocketStream(); virtual ~SimpleSocketStream(); public: + virtual int create_socket(); }; #endif