Changeset 7515 for lang/c

Timestamp:

03/05/08 22:25:56 (5 years ago)

Author:

frsyuki

Message:

lang/c/partty: fixed wrong MP_WRITE event tracing

Location:

lang/c/partty/trunk

Files:

• multiplexer.cc (modified) (14 diffs)
• multiplexer.h (modified) (2 diffs)

lang/c/partty/trunk/multiplexer.cc

@@ -22 +22 @@
                 emtelnet((void*)this),
                 m_host_telnet(host_telnet),
-                m_enable_ws(false)
+                m_enable_ws(false),
+                write_waiting(false)
 {
         // use these options
@@ -259 +260 @@
         if( fcntl(guest, F_SETFL, O_NONBLOCK) < 0 ||
                         mpev.add(guest, mp::EV_READ | mp::EV_WRITE) < 0 ) {
-                        // イベント待ちは readable or writable から始める
+                        // イベント待ちは書き込み待ち状態から始める
                 perror("guest connection failed");
                 close(guest);
@@ -267 +268 @@
         // 書き込み待ちバッファにSERVER_WELCOME_MESSAGEを加える
         guest_set.data(guest).send(SERVER_WELCOME_MESSAGE, strlen(SERVER_WELCOME_MESSAGE), NULL);
+        // 書き込み待ちから始める
+        guest_set.data(guest).write_waiting = true;
         num_guest++;
         return 0;
@@ -301 +304 @@
                 if( num_guest > 0 ) {
                         char sbbuf[4];
-                        *((short*)sbbuf) = htons(m_host_telnet.get_rows());
-                        *((short*)(sbbuf+2)) = htons(m_host_telnet.get_cols());
+                        *((short*)sbbuf) = htons(m_host_telnet.get_cols());
+                        *((short*)(sbbuf+2)) = htons(m_host_telnet.get_rows());
                         int n = 0;
                         for(int fd = 0; fd < INT_MAX; ++fd) {
@@ -312 +315 @@
                         }
                 }
+                m_host_telnet.ws_flush();
         }
         if( !m_host_telnet.ilength ) { return 0; }
@@ -354 +358 @@
                         remove_guest(fd);
                         return 0;
-                } else if( writable ) {
-                        // 書き込み可能guest -> host
+                } else {
+                        // guest -> host
                         filt_telnetd::buffer_t ibuf;
                         if( recv_filter(fd, shared_buffer, len, &ibuf) < 0 ) {
@@ -361 +365 @@
                                 return 0;
                         }
-                        // FIXME Hostが切断されたのかエラーが発生したのか区別できない
-                        if( continued_blocking_write_all(host, ibuf.buf, ibuf.len) != ibuf.len ) {
-                                throw io_error("host socket is broken");
-                        }
-                }
-                // read-only guestからの入力はそのまま捨てる
+                        if( writable ) {
+                                // FIXME Hostが切断されたのかエラーが発生したのか区別できない
+                                if( continued_blocking_write_all(host, ibuf.buf, ibuf.len) != ibuf.len ) {
+                                        throw io_error("host socket is broken");
+                                }
+                        }
+                        // read-only guestからの入力はそのまま捨てる
+                }
         }
         if( event & mp::EV_WRITE ) {
@@ -372 +378 @@
                 filt_telnetd& srv( guest_set.data(fd) );
                 ssize_t len = write(fd, srv.obuffer, srv.olength);
-                //if( srv.olength == 0 ) { return 0; }  // olengthは0の可能性がある？
                 if( len < 0 ) {
                         if( errno == EAGAIN || errno == EINTR ) { return 0; }
@@ -396 +401 @@
                                         return 0;
                                 }
+                                srv.write_waiting = false;
                                 srv.oclear();
                         }
@@ -406 +412 @@
 {
         filt_telnetd& srv( guest_set.data(fd) );
-        bool may_writable = srv.is_oempty();
         srv.recv(buf, len, ibuf, NULL);  // recvしたときにもobufが発生する
         srv.iclear();   // 呼び出し側はibufを確実に使い切らないといけない
-        if( may_writable ) {
-                // 書き込み待ちバッファが空だったので書き込めるかもしれない
+        if( !srv.write_waiting ) {
+                // 書き込み待ち状態ではないので今すぐ書き込めるかもしれない
                 if( guest_try_write(fd, srv) < 0 ) {
                         remove_guest(fd, srv);
@@ -422 +427 @@
 {
         filt_telnetd& srv( guest_set.data(fd) );
-        bool may_writable = srv.is_oempty();
         srv.send(buf, len, NULL);
-        if( may_writable ) {
-                // 書き込み待ちバッファが空だったので書き込めるかもしれない
+        if( !srv.write_waiting ) {
+                // 書き込み待ち状態ではないので今すぐ書き込めるかもしれない
                 if( guest_try_write(fd, srv) < 0 ) {
                         remove_guest(fd, srv);
@@ -438 +442 @@
         // 現在の書き込み待ちバッファにあるバッファを書き込んでみて、
         // 全部書き込めたらそのまま、書き込めなかったら書き込み待ちにする
-        // 書き込みバッファが無かった状態から書き込み待ちバッファがある状態に
-        // 遷移したときにのみ、この関数を呼べる
-        if( srv.olength <= 0 ) { return 0; }
+        if( srv.olength == 0 || srv.write_waiting ) { return 0; }
         ssize_t wlen = write(fd, srv.obuffer, srv.olength);
         if( wlen < 0 && errno != EAGAIN && errno != EINTR ) {
@@ -457 +459 @@
                         return -1;
                 }
+                srv.write_waiting = true;
         } else {
                 // 全部書き込めた
@@ -472 +475 @@
 void MultiplexerIMPL::remove_guest(int fd, filt_telnetd& srv)
 {
-        mpev.remove(fd, mp::EV_READ | (srv.is_oempty() ? 0 : mp::EV_WRITE));
+        mpev.remove(fd, mp::EV_READ | (srv.write_waiting ? mp::EV_READ : 0));
         guest_set.reset(fd);
         num_guest--;

lang/c/partty/trunk/multiplexer.h

@@ -46 +46 @@
         inline void iconsumed(size_t len);
         inline void oconsumed(size_t len);
-        inline bool is_oempty(void) { return olength == 0; }
-        inline bool is_iempty(void) { return ilength == 0; }
         inline void get_ibuffer(buffer_t* in);
         inline void get_obuffer(buffer_t* out);
@@ -57 +55 @@
         sender_telnetd& m_host_telnet;
         bool m_enable_ws;
+public:
+        bool write_waiting;
 };