fix auth flag in tunnel and redir

9 years ago · 656c97e64e
2 changed files with 2 additions and 2 deletions
--- a/src/redir.c
+++ b/src/redir.c
@ -378,9 +378,9 @@ static void remote_send_cb(EV_P_ ev_io *w, int revents)
            }

            if (auth) {
+                ss_addr_to_send[0] |= ONETIMEAUTH_FLAG;
                ss_onetimeauth(ss_addr_to_send + addr_len, ss_addr_to_send, addr_len);
                addr_len += ONETIMEAUTH_BYTES;
-                ss_addr_to_send[0] |= ONETIMEAUTH_FLAG;
            }

            int s = send(remote->fd, ss_addr_to_send, addr_len, 0);
--- a/src/tunnel.c
+++ b/src/tunnel.c
@ -420,9 +420,9 @@ static void remote_send_cb(EV_P_ ev_io *w, int revents)
            }

            if (auth) {
+                ss_addr_to_send[0] |= ONETIMEAUTH_FLAG;
                ss_onetimeauth(ss_addr_to_send + addr_len, ss_addr_to_send, addr_len);
                addr_len += ONETIMEAUTH_BYTES;
-                ss_addr_to_send[0] |= ONETIMEAUTH_FLAG;
            }

            int s = send(remote->fd, ss_addr_to_send, addr_len, 0);