Changeset 2159 for trunk/tools/neercs/term/pty.cppTweet

Timestamp:

Dec 22, 2012, 5:19:41 PM (10 years ago)

Author:

sam

Message:

neercs: fix our unread() function so that it stops losing characters.

File:

• trunk/tools/neercs/term/pty.cpp (modified) (3 diffs)

trunk/tools/neercs/term/pty.cpp

@@ -114 +114 @@
 {
 #if defined HAVE_FORKPTY
+    size_t sent = 0;
+
     /* Do we have data from previous call? */
     if (m_unread_len)
     {
-        /* FIXME: check that m_unread_len < maxlen */
-        memcpy(data, m_unread_data, m_unread_len);
-
-        data += m_unread_len;
-        maxlen -= m_unread_len;
-
-        delete[] m_unread_data;
-        m_unread_data = 0;
-        m_unread_len = 0;
+        size_t tocopy = min(maxlen, m_unread_len);
+
+        memcpy(data, m_unread_data, tocopy);
+
+        data += tocopy;
+        sent += tocopy;
+        maxlen -= tocopy;
+
+        if (tocopy < m_unread_len)
+        {
+            m_unread_len -= tocopy;
+            memmove(m_unread_data, m_unread_data + tocopy, m_unread_len);
+        }
+        else
+        {
+            delete[] m_unread_data;
+            m_unread_data = 0;
+            m_unread_len = 0;
+        }
     }
 
@@ -160 +172 @@
                 if (nr <= 0)
                     m_eof = true;
-
-                if (nr >= 0)
-                    return nr;
+                else
+                    sent += nr;
+
+                if (sent >= 0)
+                    return sent;
             }
         }
@@ -174 +188 @@
 {
 #if defined HAVE_FORKPTY
-    char *new_data;
-
+    /* Prepare unread buffer */
     if (m_unread_data)
     {
-        new_data = new char[m_unread_len + len];
-        memcpy(new_data + len, m_unread_data, m_unread_len);
+        char *tmp = new char[m_unread_len + len];
+        memcpy(tmp + len, m_unread_data, m_unread_len);
         delete[] m_unread_data;
+        m_unread_data = tmp;
+        m_unread_len += len;
     }
     else
     {
-        new_data = new char[len];
-    }
-
-    memcpy(new_data, data, len);
-    m_unread_data = new_data;
+        m_unread_data = new char[len];
+        m_unread_len = len;
+    }
+
+    /* Copy data to the unread buffer */
+    memcpy(m_unread_data, data, len);
 #endif
 }