refine srs buffer, min interfaces

trunk/src/kernel/srs_kernel_buffer.cpp

@@ -44,43 +44,38 @@ SrsBuffer::~SrsBuffer()
 {
 }
 
-int SrsBuffer::size()
+int SrsBuffer::length()
 {
-    return (int)data.size();
-}
-
-bool SrsBuffer::empty()
-{
-    return size() <= 0;
+    int len = (int)data.size();
+    srs_assert(len >= 0);
+    return len;
 }
 
 char* SrsBuffer::bytes()
 {
-    return &data.at(0);
+    return (length() == 0)? NULL : &data.at(0);
 }
 
-void SrsBuffer::erase(int _size)
+void SrsBuffer::erase(int size)
 {
-    if (_size == size()) {
-        clear();
+    srs_assert(size > 0);
+    
+    if (size == length()) {
+        data.clear();
         return;
     }
     
-    data.erase(data.begin(), data.begin() + _size);
-}
-
-void SrsBuffer::clear()
-{
-    data.clear();
+    data.erase(data.begin(), data.begin() + size);
 }
 
 void SrsBuffer::append(const char* bytes, int size)
 {
     srs_assert(size > 0);
+
     data.insert(data.end(), bytes, bytes + size);
 }
 
-int SrsBuffer::ensure_buffer_bytes(ISrsBufferReader* skt, int required_size)
+int SrsBuffer::grow(ISrsBufferReader* reader, int required_size)
 {
     int ret = ERROR_SUCCESS;
 
@@ -90,11 +85,11 @@ int SrsBuffer::ensure_buffer_bytes(ISrsBufferReader* skt, int required_size)
         return ret;
     }
 
-    while (size() < required_size) {
+    while (length() < required_size) {
         char buffer[SOCKET_READ_SIZE];
         
         ssize_t nread;
-        if ((ret = skt->read(buffer, SOCKET_READ_SIZE, &nread)) != ERROR_SUCCESS) {
+        if ((ret = reader->read(buffer, SOCKET_READ_SIZE, &nread)) != ERROR_SUCCESS) {
             return ret;
         }
         

trunk/src/kernel/srs_kernel_buffer.hpp

@@ -50,7 +50,7 @@ public:
 * protocol recv data from socket, put into buffer, decode to RTMP message.
 * protocol encode RTMP message to bytes, put into buffer, send to socket.
 */
-class SrsBuffer
+ class SrsBuffer
 {
 private:
     std::vector<char> data;
@@ -58,14 +58,34 @@ public:
     SrsBuffer();
     virtual ~SrsBuffer();
 public:
-    virtual int size();
-    virtual bool empty();
+    /**
+    * get the length of buffer.
+    * never negative, empty if zero.
+    */
+    virtual int length();
+    /**
+    * get the buffer bytes.
+    * @return the bytes, NULL if empty.
+    */
     virtual char* bytes();
+    /**
+    * erase size of bytes from begin.
+    * if size equals to length(), clear buffer.
+    * @param size
+    */
     virtual void erase(int size);
-    virtual void clear();
+    /**
+    * append specified bytes to buffer.
+    * @param size the size of bytes, assert positive.
+    */
     virtual void append(const char* bytes, int size);
 public:
-    virtual int ensure_buffer_bytes(ISrsBufferReader* skt, int required_size);
+    /**
+    * grow buffer to the required size, loop to read from skt to fill.
+    * @param reader, read more bytes from reader to fill the buffer to required size.
+    * @param required_size, loop to fill to ensure buffer size to required.
+    */
+    virtual int grow(ISrsBufferReader* reader, int required_size);
 };
 
 #endif