Mercurial > dropbear
annotate cli-session.c @ 41:18eccbfb9641
added window-size change handling
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Sun, 01 Aug 2004 09:41:37 +0000 |
| parents | b4874d772210 |
| children | 942b22d7dd1c |
| rev | line source |
|---|---|
| 26 | 1 #include "includes.h" |
| 2 #include "session.h" | |
| 3 #include "dbutil.h" | |
| 4 #include "kex.h" | |
| 5 #include "ssh.h" | |
| 6 #include "packet.h" | |
| 7 #include "tcpfwd-direct.h" | |
| 8 #include "tcpfwd-remote.h" | |
| 9 #include "channel.h" | |
| 10 #include "random.h" | |
| 33 | 11 #include "service.h" |
|
40
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
12 #include "runopts.h" |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
13 #include "chansession.h" |
| 26 | 14 |
| 15 static void cli_remoteclosed(); | |
| 16 static void cli_sessionloop(); | |
| 33 | 17 static void cli_session_init(); |
|
40
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
18 static void cli_finished(); |
| 26 | 19 |
| 20 struct clientsession cli_ses; /* GLOBAL */ | |
| 21 | |
| 22 static const packettype cli_packettypes[] = { | |
| 23 /* TYPE, AUTHREQUIRED, FUNCTION */ | |
| 24 {SSH_MSG_KEXINIT, recv_msg_kexinit}, | |
| 25 {SSH_MSG_KEXDH_REPLY, recv_msg_kexdh_reply}, // client | |
| 26 {SSH_MSG_NEWKEYS, recv_msg_newkeys}, | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
27 {SSH_MSG_SERVICE_ACCEPT, recv_msg_service_accept}, // client |
| 26 | 28 {SSH_MSG_CHANNEL_DATA, recv_msg_channel_data}, |
| 29 {SSH_MSG_CHANNEL_WINDOW_ADJUST, recv_msg_channel_window_adjust}, | |
| 30 {SSH_MSG_GLOBAL_REQUEST, recv_msg_global_request_remotetcp}, | |
| 31 {SSH_MSG_CHANNEL_REQUEST, recv_msg_channel_request}, | |
| 32 {SSH_MSG_CHANNEL_OPEN, recv_msg_channel_open}, | |
| 33 {SSH_MSG_CHANNEL_EOF, recv_msg_channel_eof}, | |
| 34 {SSH_MSG_CHANNEL_CLOSE, recv_msg_channel_close}, | |
| 35 {SSH_MSG_CHANNEL_OPEN_CONFIRMATION, recv_msg_channel_open_confirmation}, | |
| 36 {SSH_MSG_CHANNEL_OPEN_FAILURE, recv_msg_channel_open_failure}, | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
37 {SSH_MSG_USERAUTH_FAILURE, recv_msg_userauth_failure}, // client |
|
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
38 {SSH_MSG_USERAUTH_SUCCESS, recv_msg_userauth_success}, // client |
| 26 | 39 {0, 0} /* End */ |
| 40 }; | |
| 41 | |
| 42 static const struct ChanType *cli_chantypes[] = { | |
| 43 /* &chan_tcpdirect etc, though need to only allow if we've requested | |
| 44 * that forwarding */ | |
| 45 NULL /* Null termination */ | |
| 46 }; | |
| 33 | 47 |
| 26 | 48 void cli_session(int sock, char* remotehost) { |
| 49 | |
| 50 crypto_init(); | |
| 51 common_session_init(sock, remotehost); | |
| 52 | |
| 53 chaninitialise(cli_chantypes); | |
| 54 | |
| 55 | |
| 33 | 56 /* Set up cli_ses vars */ |
| 57 cli_session_init(); | |
| 26 | 58 |
| 59 /* Ready to go */ | |
| 60 sessinitdone = 1; | |
| 61 | |
| 62 /* Exchange identification */ | |
| 63 session_identification(); | |
| 64 | |
| 65 seedrandom(); | |
| 66 | |
| 67 send_msg_kexinit(); | |
| 68 | |
| 69 /* XXX here we do stuff differently */ | |
| 70 | |
| 71 session_loop(cli_sessionloop); | |
| 72 | |
| 73 /* Not reached */ | |
| 74 | |
| 33 | 75 } |
| 26 | 76 |
| 33 | 77 static void cli_session_init() { |
| 78 | |
| 79 cli_ses.state = STATE_NOTHING; | |
| 80 cli_ses.kex_state = KEX_NOTHING; | |
| 81 | |
|
39
0883c0906870
tty raw mode support works mostly
Matt Johnston <matt@ucc.asn.au>
parents:
37
diff
changeset
|
82 cli_ses.tty_raw_mode = 0; |
|
41
18eccbfb9641
added window-size change handling
Matt Johnston <matt@ucc.asn.au>
parents:
40
diff
changeset
|
83 cli_ses.winchange = 0; |
|
39
0883c0906870
tty raw mode support works mostly
Matt Johnston <matt@ucc.asn.au>
parents:
37
diff
changeset
|
84 |
| 33 | 85 /* For printing "remote host closed" for the user */ |
| 86 ses.remoteclosed = cli_remoteclosed; | |
| 87 ses.buf_match_algo = cli_buf_match_algo; | |
| 88 | |
| 89 /* packet handlers */ | |
| 90 ses.packettypes = cli_packettypes; | |
|
35
0ad5fb979f42
set the isserver flag (oops)
Matt Johnston <matt@ucc.asn.au>
parents:
34
diff
changeset
|
91 |
|
0ad5fb979f42
set the isserver flag (oops)
Matt Johnston <matt@ucc.asn.au>
parents:
34
diff
changeset
|
92 ses.isserver = 0; |
| 26 | 93 } |
| 94 | |
| 33 | 95 /* This function drives the progress of the session - it initiates KEX, |
| 96 * service, userauth and channel requests */ | |
| 26 | 97 static void cli_sessionloop() { |
| 98 | |
| 33 | 99 TRACE(("enter cli_sessionloop")); |
| 100 | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
101 if (ses.lastpacket == SSH_MSG_KEXINIT && cli_ses.kex_state == KEX_NOTHING) { |
|
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
102 cli_ses.kex_state = KEXINIT_RCVD; |
| 33 | 103 } |
| 104 | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
105 if (cli_ses.kex_state == KEXINIT_RCVD) { |
| 33 | 106 |
| 107 /* We initiate the KEXDH. If DH wasn't the correct type, the KEXINIT | |
| 108 * negotiation would have failed. */ | |
| 109 send_msg_kexdh_init(); | |
| 110 cli_ses.kex_state = KEXDH_INIT_SENT; | |
| 111 TRACE(("leave cli_sessionloop: done with KEXINIT_RCVD")); | |
| 112 return; | |
| 113 } | |
| 114 | |
| 115 /* A KEX has finished, so we should go back to our KEX_NOTHING state */ | |
| 116 if (cli_ses.kex_state != KEX_NOTHING && ses.kexstate.recvkexinit == 0 | |
| 117 && ses.kexstate.sentkexinit == 0) { | |
| 118 cli_ses.kex_state = KEX_NOTHING; | |
| 119 } | |
| 120 | |
| 121 /* We shouldn't do anything else if a KEX is in progress */ | |
| 122 if (cli_ses.kex_state != KEX_NOTHING) { | |
| 123 TRACE(("leave cli_sessionloop: kex_state != KEX_NOTHING")); | |
| 124 return; | |
| 125 } | |
| 126 | |
| 127 /* We should exit if we haven't donefirstkex: we shouldn't reach here | |
| 128 * in normal operation */ | |
| 129 if (ses.kexstate.donefirstkex == 0) { | |
| 130 TRACE(("XXX XXX might be bad! leave cli_sessionloop: haven't donefirstkex")); | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
131 return; |
| 33 | 132 } |
| 133 | |
| 26 | 134 switch (cli_ses.state) { |
| 135 | |
| 33 | 136 case STATE_NOTHING: |
| 137 /* We've got the transport layer sorted, we now need to request | |
| 138 * userauth */ | |
| 139 send_msg_service_request(SSH_SERVICE_USERAUTH); | |
| 140 cli_ses.state = SERVICE_AUTH_REQ_SENT; | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
141 TRACE(("leave cli_sessionloop: sent userauth service req")); |
| 33 | 142 return; |
| 26 | 143 |
| 33 | 144 /* userauth code */ |
| 145 case SERVICE_AUTH_ACCEPT_RCVD: | |
| 146 cli_auth_getmethods(); | |
| 147 cli_ses.state = USERAUTH_METHODS_SENT; | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
148 TRACE(("leave cli_sessionloop: sent userauth methods req")); |
| 33 | 149 return; |
| 150 | |
| 151 case USERAUTH_FAIL_RCVD: | |
| 152 cli_auth_try(); | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
153 TRACE(("leave cli_sessionloop: cli_auth_try")); |
| 33 | 154 return; |
| 155 | |
| 37 | 156 /* |
| 157 case USERAUTH_SUCCESS_RCVD: | |
| 158 send_msg_service_request(SSH_SERVICE_CONNECTION); | |
| 159 cli_ses.state = SERVICE_CONN_REQ_SENT; | |
| 160 TRACE(("leave cli_sessionloop: sent ssh-connection service req")); | |
| 161 return; | |
| 162 */ | |
| 163 | |
| 164 case USERAUTH_SUCCESS_RCVD: | |
| 165 cli_send_chansess_request(); | |
| 166 TRACE(("leave cli_sessionloop: cli_send_chansess_request")); | |
| 167 cli_ses.state = SESSION_RUNNING; | |
| 168 return; | |
| 169 | |
|
40
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
170 case SESSION_RUNNING: |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
171 if (ses.chancount < 1) { |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
172 cli_finished(); |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
173 } |
|
41
18eccbfb9641
added window-size change handling
Matt Johnston <matt@ucc.asn.au>
parents:
40
diff
changeset
|
174 |
|
18eccbfb9641
added window-size change handling
Matt Johnston <matt@ucc.asn.au>
parents:
40
diff
changeset
|
175 if (cli_ses.winchange) { |
|
18eccbfb9641
added window-size change handling
Matt Johnston <matt@ucc.asn.au>
parents:
40
diff
changeset
|
176 cli_chansess_winchange(); |
|
18eccbfb9641
added window-size change handling
Matt Johnston <matt@ucc.asn.au>
parents:
40
diff
changeset
|
177 } |
|
40
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
178 return; |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
179 |
| 33 | 180 /* XXX more here needed */ |
| 181 | |
| 182 | |
| 183 default: | |
| 184 break; | |
| 26 | 185 } |
| 186 | |
|
34
e2a1eaa19f22
Client mostly works up to password auth
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
187 TRACE(("leave cli_sessionloop: fell out")); |
| 26 | 188 |
| 189 } | |
| 190 | |
|
40
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
191 void cli_session_cleanup() { |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
192 |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
193 if (!sessinitdone) { |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
194 return; |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
195 } |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
196 cli_tty_cleanup(); |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
197 |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
198 } |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
199 |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
200 static void cli_finished() { |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
201 |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
202 cli_session_cleanup(); |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
203 common_session_cleanup(); |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
204 fprintf(stderr, "Connection to %s@%s:%s closed.\n", cli_opts.username, |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
205 cli_opts.remotehost, cli_opts.remoteport); |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
206 exit(EXIT_SUCCESS); |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
207 } |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
208 |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
209 |
|
b4874d772210
- Added terminal mode handling etc for the client, and window change
Matt Johnston <matt@ucc.asn.au>
parents:
39
diff
changeset
|
210 |
| 26 | 211 /* called when the remote side closes the connection */ |
| 212 static void cli_remoteclosed() { | |
| 213 | |
| 214 /* XXX TODO perhaps print a friendlier message if we get this but have | |
| 215 * already sent/received disconnect message(s) ??? */ | |
| 216 close(ses.sock); | |
| 217 ses.sock = -1; | |
| 33 | 218 dropbear_exit("remote closed the connection"); |
| 26 | 219 } |