portfwd.c 36 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175
  1. /*
  2. * SSH port forwarding.
  3. */
  4. #include <assert.h>
  5. #include <stdio.h>
  6. #include <stdlib.h>
  7. #include "putty.h"
  8. #include "ssh.h"
  9. #include "sshchan.h"
  10. /*
  11. * Enumeration of values that live in the 'socks_state' field of
  12. * struct PortForwarding.
  13. */
  14. typedef enum {
  15. SOCKS_NONE, /* direct connection (no SOCKS, or SOCKS already done) */
  16. SOCKS_INITIAL, /* don't know if we're SOCKS 4 or 5 yet */
  17. SOCKS_4, /* expect a SOCKS 4 (or 4A) connection message */
  18. SOCKS_5_INITIAL, /* expect a SOCKS 5 preliminary message */
  19. SOCKS_5_CONNECT /* expect a SOCKS 5 connection message */
  20. } SocksState;
  21. typedef struct PortForwarding {
  22. SshChannel *c; /* channel structure held by SSH connection layer */
  23. ConnectionLayer *cl; /* the connection layer itself */
  24. /* Note that ssh need not be filled in if c is non-NULL */
  25. Socket *s;
  26. bool input_wanted;
  27. bool ready;
  28. SocksState socks_state;
  29. /*
  30. * `hostname' and `port' are the real hostname and port, once
  31. * we know what we're connecting to.
  32. */
  33. char *hostname;
  34. int port;
  35. /*
  36. * `socksbuf' is the buffer we use to accumulate the initial SOCKS
  37. * segment of the incoming data, plus anything after that that we
  38. * receive before we're ready to send data to the SSH server.
  39. */
  40. strbuf *socksbuf;
  41. size_t socksbuf_consumed;
  42. Plug plug;
  43. Channel chan;
  44. } PortForwarding;
  45. struct PortListener {
  46. ConnectionLayer *cl;
  47. Socket *s;
  48. bool is_dynamic;
  49. /*
  50. * `hostname' and `port' are the real hostname and port, for
  51. * ordinary forwardings.
  52. */
  53. char *hostname;
  54. int port;
  55. Plug plug;
  56. };
  57. static struct PortForwarding *new_portfwd_state(void)
  58. {
  59. struct PortForwarding *pf = snew(struct PortForwarding);
  60. pf->hostname = NULL;
  61. pf->socksbuf = NULL;
  62. return pf;
  63. }
  64. static void free_portfwd_state(struct PortForwarding *pf)
  65. {
  66. if (!pf)
  67. return;
  68. sfree(pf->hostname);
  69. if (pf->socksbuf)
  70. strbuf_free(pf->socksbuf);
  71. sfree(pf);
  72. }
  73. static struct PortListener *new_portlistener_state(void)
  74. {
  75. struct PortListener *pl = snew(struct PortListener);
  76. pl->hostname = NULL;
  77. return pl;
  78. }
  79. static void free_portlistener_state(struct PortListener *pl)
  80. {
  81. if (!pl)
  82. return;
  83. sfree(pl->hostname);
  84. sfree(pl);
  85. }
  86. static void pfd_log(Plug *plug, PlugLogType type, SockAddr *addr, int port,
  87. const char *error_msg, int error_code)
  88. {
  89. /* we have to dump these since we have no interface to logging.c */
  90. }
  91. static void pfl_log(Plug *plug, PlugLogType type, SockAddr *addr, int port,
  92. const char *error_msg, int error_code)
  93. {
  94. /* we have to dump these since we have no interface to logging.c */
  95. }
  96. static void pfd_close(struct PortForwarding *pf);
  97. static void pfd_closing(Plug *plug, const char *error_msg, int error_code,
  98. bool calling_back)
  99. {
  100. struct PortForwarding *pf =
  101. container_of(plug, struct PortForwarding, plug);
  102. if (error_msg) {
  103. /*
  104. * Socket error. Slam the connection instantly shut.
  105. */
  106. if (pf->c) {
  107. sshfwd_initiate_close(pf->c, error_msg);
  108. } else {
  109. /*
  110. * We might not have an SSH channel, if a socket error
  111. * occurred during SOCKS negotiation. If not, we must
  112. * clean ourself up without sshfwd_initiate_close's call
  113. * back to pfd_close.
  114. */
  115. pfd_close(pf);
  116. }
  117. } else {
  118. /*
  119. * Ordinary EOF received on socket. Send an EOF on the SSH
  120. * channel.
  121. */
  122. if (pf->c)
  123. sshfwd_write_eof(pf->c);
  124. }
  125. }
  126. static void pfl_terminate(struct PortListener *pl);
  127. static void pfl_closing(Plug *plug, const char *error_msg, int error_code,
  128. bool calling_back)
  129. {
  130. struct PortListener *pl = (struct PortListener *) plug;
  131. pfl_terminate(pl);
  132. }
  133. static SshChannel *wrap_lportfwd_open(
  134. ConnectionLayer *cl, const char *hostname, int port,
  135. Socket *s, Channel *chan)
  136. {
  137. SocketPeerInfo *pi;
  138. char *description;
  139. SshChannel *toret;
  140. pi = sk_peer_info(s);
  141. if (pi && pi->log_text) {
  142. description = dupprintf("forwarding from %s", pi->log_text);
  143. } else {
  144. description = dupstr("forwarding");
  145. }
  146. toret = ssh_lportfwd_open(cl, hostname, port, description, pi, chan);
  147. sk_free_peer_info(pi);
  148. sfree(description);
  149. return toret;
  150. }
  151. static char *ipv4_to_string(unsigned ipv4)
  152. {
  153. return dupprintf("%u.%u.%u.%u",
  154. (ipv4 >> 24) & 0xFF, (ipv4 >> 16) & 0xFF,
  155. (ipv4 >> 8) & 0xFF, (ipv4 ) & 0xFF);
  156. }
  157. static char *ipv6_to_string(ptrlen ipv6)
  158. {
  159. const unsigned char *addr = ipv6.ptr;
  160. assert(ipv6.len == 16);
  161. return dupprintf("%04x:%04x:%04x:%04x:%04x:%04x:%04x:%04x",
  162. (unsigned)GET_16BIT_MSB_FIRST(addr + 0),
  163. (unsigned)GET_16BIT_MSB_FIRST(addr + 2),
  164. (unsigned)GET_16BIT_MSB_FIRST(addr + 4),
  165. (unsigned)GET_16BIT_MSB_FIRST(addr + 6),
  166. (unsigned)GET_16BIT_MSB_FIRST(addr + 8),
  167. (unsigned)GET_16BIT_MSB_FIRST(addr + 10),
  168. (unsigned)GET_16BIT_MSB_FIRST(addr + 12),
  169. (unsigned)GET_16BIT_MSB_FIRST(addr + 14));
  170. }
  171. static void pfd_receive(Plug *plug, int urgent, const char *data, size_t len)
  172. {
  173. struct PortForwarding *pf =
  174. container_of(plug, struct PortForwarding, plug);
  175. if (len == 0)
  176. return;
  177. if (pf->socks_state != SOCKS_NONE) {
  178. BinarySource src[1];
  179. /*
  180. * Store all the data we've got in socksbuf.
  181. */
  182. put_data(pf->socksbuf, data, len);
  183. /*
  184. * Check the start of socksbuf to see if it's a valid and
  185. * complete message in the SOCKS exchange.
  186. */
  187. if (pf->socks_state == SOCKS_INITIAL) {
  188. /* Preliminary: check the first byte of the data (which we
  189. * _must_ have by now) to find out which SOCKS major
  190. * version we're speaking. */
  191. switch (pf->socksbuf->u[0]) {
  192. case 4:
  193. pf->socks_state = SOCKS_4;
  194. break;
  195. case 5:
  196. pf->socks_state = SOCKS_5_INITIAL;
  197. break;
  198. default:
  199. pfd_close(pf); /* unrecognised version */
  200. return;
  201. }
  202. }
  203. BinarySource_BARE_INIT(src, pf->socksbuf->u, pf->socksbuf->len);
  204. get_data(src, pf->socksbuf_consumed);
  205. while (pf->socks_state != SOCKS_NONE) {
  206. unsigned socks_version, message_type, reserved_byte;
  207. unsigned reply_code, port, ipv4, method;
  208. ptrlen methods;
  209. const char *socks4_hostname;
  210. strbuf *output;
  211. switch (pf->socks_state) {
  212. case SOCKS_INITIAL:
  213. case SOCKS_NONE:
  214. unreachable("These case values cannot appear");
  215. case SOCKS_4:
  216. /* SOCKS 4/4A connect message */
  217. socks_version = get_byte(src);
  218. message_type = get_byte(src);
  219. if (get_err(src) == BSE_OUT_OF_DATA)
  220. return;
  221. if (socks_version == 4 && message_type == 1) {
  222. /* CONNECT message */
  223. bool name_based = false;
  224. port = get_uint16(src);
  225. ipv4 = get_uint32(src);
  226. if (ipv4 > 0x00000000 && ipv4 < 0x00000100) {
  227. /*
  228. * Addresses in this range indicate the SOCKS 4A
  229. * extension to specify a hostname, which comes
  230. * after the username.
  231. */
  232. name_based = true;
  233. }
  234. get_asciz(src); /* skip username */
  235. socks4_hostname = name_based ? get_asciz(src) : NULL;
  236. if (get_err(src) == BSE_OUT_OF_DATA)
  237. return;
  238. if (get_err(src))
  239. goto socks4_reject;
  240. pf->port = port;
  241. if (name_based) {
  242. pf->hostname = dupstr(socks4_hostname);
  243. } else {
  244. pf->hostname = ipv4_to_string(ipv4);
  245. }
  246. output = strbuf_new();
  247. put_byte(output, 0); /* reply version */
  248. put_byte(output, 90); /* SOCKS 4 'request granted' */
  249. put_uint16(output, 0); /* null port field */
  250. put_uint32(output, 0); /* null address field */
  251. sk_write(pf->s, output->u, output->len);
  252. strbuf_free(output);
  253. pf->socks_state = SOCKS_NONE;
  254. pf->socksbuf_consumed = src->pos;
  255. break;
  256. }
  257. socks4_reject:
  258. output = strbuf_new();
  259. put_byte(output, 0); /* reply version */
  260. put_byte(output, 91); /* SOCKS 4 'request rejected' */
  261. put_uint16(output, 0); /* null port field */
  262. put_uint32(output, 0); /* null address field */
  263. sk_write(pf->s, output->u, output->len);
  264. strbuf_free(output);
  265. pfd_close(pf);
  266. return;
  267. case SOCKS_5_INITIAL:
  268. /* SOCKS 5 initial method list */
  269. socks_version = get_byte(src);
  270. methods = get_pstring(src);
  271. method = 0xFF; /* means 'no usable method found' */
  272. {
  273. int i;
  274. for (i = 0; i < methods.len; i++) {
  275. if (((const unsigned char *)methods.ptr)[i] == 0 ) {
  276. method = 0; /* no auth */
  277. break;
  278. }
  279. }
  280. }
  281. if (get_err(src) == BSE_OUT_OF_DATA)
  282. return;
  283. if (get_err(src))
  284. method = 0xFF;
  285. output = strbuf_new();
  286. put_byte(output, 5); /* SOCKS version */
  287. put_byte(output, method); /* selected auth method */
  288. sk_write(pf->s, output->u, output->len);
  289. strbuf_free(output);
  290. if (method == 0xFF) {
  291. pfd_close(pf);
  292. return;
  293. }
  294. pf->socks_state = SOCKS_5_CONNECT;
  295. pf->socksbuf_consumed = src->pos;
  296. break;
  297. case SOCKS_5_CONNECT:
  298. /* SOCKS 5 connect message */
  299. socks_version = get_byte(src);
  300. message_type = get_byte(src);
  301. reserved_byte = get_byte(src);
  302. if (socks_version == 5 && message_type == 1 &&
  303. reserved_byte == 0) {
  304. reply_code = 0; /* success */
  305. switch (get_byte(src)) {
  306. case 1: /* IPv4 */
  307. pf->hostname = ipv4_to_string(get_uint32(src));
  308. break;
  309. case 4: /* IPv6 */
  310. pf->hostname = ipv6_to_string(get_data(src, 16));
  311. break;
  312. case 3: /* unresolved domain name */
  313. pf->hostname = mkstr(get_pstring(src));
  314. break;
  315. default:
  316. pf->hostname = NULL;
  317. reply_code = 8; /* address type not supported */
  318. break;
  319. }
  320. pf->port = get_uint16(src);
  321. } else {
  322. reply_code = 7; /* command not supported */
  323. }
  324. if (get_err(src) == BSE_OUT_OF_DATA)
  325. return;
  326. if (get_err(src))
  327. reply_code = 1; /* general server failure */
  328. output = strbuf_new();
  329. put_byte(output, 5); /* SOCKS version */
  330. put_byte(output, reply_code);
  331. put_byte(output, 0); /* reserved */
  332. put_byte(output, 1); /* IPv4 address follows */
  333. put_uint32(output, 0); /* bound IPv4 address (unused) */
  334. put_uint16(output, 0); /* bound port number (unused) */
  335. sk_write(pf->s, output->u, output->len);
  336. strbuf_free(output);
  337. if (reply_code != 0) {
  338. pfd_close(pf);
  339. return;
  340. }
  341. pf->socks_state = SOCKS_NONE;
  342. pf->socksbuf_consumed = src->pos;
  343. break;
  344. }
  345. }
  346. /*
  347. * We come here when we're ready to make an actual
  348. * connection.
  349. */
  350. /*
  351. * Freeze the socket until the SSH server confirms the
  352. * connection.
  353. */
  354. sk_set_frozen(pf->s, true);
  355. pf->c = wrap_lportfwd_open(pf->cl, pf->hostname, pf->port, pf->s,
  356. &pf->chan);
  357. }
  358. if (pf->ready)
  359. sshfwd_write(pf->c, data, len);
  360. }
  361. static void pfd_sent(Plug *plug, size_t bufsize)
  362. {
  363. struct PortForwarding *pf =
  364. container_of(plug, struct PortForwarding, plug);
  365. if (pf->c)
  366. sshfwd_unthrottle(pf->c, bufsize);
  367. }
  368. static const PlugVtable PortForwarding_plugvt = {
  369. .log = pfd_log,
  370. .closing = pfd_closing,
  371. .receive = pfd_receive,
  372. .sent = pfd_sent,
  373. };
  374. static void pfd_chan_free(Channel *chan);
  375. static void pfd_open_confirmation(Channel *chan);
  376. static void pfd_open_failure(Channel *chan, const char *errtext);
  377. static size_t pfd_send(
  378. Channel *chan, bool is_stderr, const void *data, size_t len);
  379. static void pfd_send_eof(Channel *chan);
  380. static void pfd_set_input_wanted(Channel *chan, bool wanted);
  381. static char *pfd_log_close_msg(Channel *chan);
  382. static const ChannelVtable PortForwarding_channelvt = {
  383. .free = pfd_chan_free,
  384. .open_confirmation = pfd_open_confirmation,
  385. .open_failed = pfd_open_failure,
  386. .send = pfd_send,
  387. .send_eof = pfd_send_eof,
  388. .set_input_wanted = pfd_set_input_wanted,
  389. .log_close_msg = pfd_log_close_msg,
  390. .want_close = chan_default_want_close,
  391. .rcvd_exit_status = chan_no_exit_status,
  392. .rcvd_exit_signal = chan_no_exit_signal,
  393. .rcvd_exit_signal_numeric = chan_no_exit_signal_numeric,
  394. .run_shell = chan_no_run_shell,
  395. .run_command = chan_no_run_command,
  396. .run_subsystem = chan_no_run_subsystem,
  397. .enable_x11_forwarding = chan_no_enable_x11_forwarding,
  398. .enable_agent_forwarding = chan_no_enable_agent_forwarding,
  399. .allocate_pty = chan_no_allocate_pty,
  400. .set_env = chan_no_set_env,
  401. .send_break = chan_no_send_break,
  402. .send_signal = chan_no_send_signal,
  403. .change_window_size = chan_no_change_window_size,
  404. .request_response = chan_no_request_response,
  405. };
  406. Channel *portfwd_raw_new(ConnectionLayer *cl, Plug **plug, bool start_ready)
  407. {
  408. struct PortForwarding *pf;
  409. pf = new_portfwd_state();
  410. pf->plug.vt = &PortForwarding_plugvt;
  411. pf->chan.initial_fixed_window_size = 0;
  412. pf->chan.vt = &PortForwarding_channelvt;
  413. pf->input_wanted = true;
  414. pf->c = NULL;
  415. pf->cl = cl;
  416. pf->input_wanted = true;
  417. pf->ready = start_ready;
  418. pf->socks_state = SOCKS_NONE;
  419. pf->hostname = NULL;
  420. pf->port = 0;
  421. *plug = &pf->plug;
  422. return &pf->chan;
  423. }
  424. void portfwd_raw_free(Channel *pfchan)
  425. {
  426. struct PortForwarding *pf;
  427. assert(pfchan->vt == &PortForwarding_channelvt);
  428. pf = container_of(pfchan, struct PortForwarding, chan);
  429. free_portfwd_state(pf);
  430. }
  431. void portfwd_raw_setup(Channel *pfchan, Socket *s, SshChannel *sc)
  432. {
  433. struct PortForwarding *pf;
  434. assert(pfchan->vt == &PortForwarding_channelvt);
  435. pf = container_of(pfchan, struct PortForwarding, chan);
  436. pf->s = s;
  437. pf->c = sc;
  438. }
  439. /*
  440. called when someone connects to the local port
  441. */
  442. static int pfl_accepting(Plug *p, accept_fn_t constructor, accept_ctx_t ctx)
  443. {
  444. struct PortListener *pl = container_of(p, struct PortListener, plug);
  445. struct PortForwarding *pf;
  446. Channel *chan;
  447. Plug *plug;
  448. Socket *s;
  449. const char *err;
  450. chan = portfwd_raw_new(pl->cl, &plug, false);
  451. s = constructor(ctx, plug);
  452. if ((err = sk_socket_error(s)) != NULL) {
  453. portfwd_raw_free(chan);
  454. return 1;
  455. }
  456. pf = container_of(chan, struct PortForwarding, chan);
  457. if (pl->is_dynamic) {
  458. pf->s = s;
  459. pf->socks_state = SOCKS_INITIAL;
  460. pf->socksbuf = strbuf_new();
  461. pf->socksbuf_consumed = 0;
  462. pf->port = 0; /* "hostname" buffer is so far empty */
  463. sk_set_frozen(s, false); /* we want to receive SOCKS _now_! */
  464. } else {
  465. pf->hostname = dupstr(pl->hostname);
  466. pf->port = pl->port;
  467. portfwd_raw_setup(
  468. chan, s,
  469. wrap_lportfwd_open(pl->cl, pf->hostname, pf->port, s, &pf->chan));
  470. }
  471. return 0;
  472. }
  473. static const PlugVtable PortListener_plugvt = {
  474. .log = pfl_log,
  475. .closing = pfl_closing,
  476. .accepting = pfl_accepting,
  477. };
  478. /*
  479. * Add a new port-forwarding listener from srcaddr:port -> desthost:destport.
  480. *
  481. * desthost == NULL indicates dynamic SOCKS port forwarding.
  482. *
  483. * On success, returns NULL and fills in *pl_ret. On error, returns a
  484. * dynamically allocated error message string.
  485. */
  486. static char *pfl_listen(const char *desthost, int destport,
  487. const char *srcaddr, int port,
  488. ConnectionLayer *cl, Conf *conf,
  489. struct PortListener **pl_ret, int address_family)
  490. {
  491. const char *err;
  492. struct PortListener *pl;
  493. /*
  494. * Open socket.
  495. */
  496. pl = *pl_ret = new_portlistener_state();
  497. pl->plug.vt = &PortListener_plugvt;
  498. if (desthost) {
  499. pl->hostname = dupstr(desthost);
  500. pl->port = destport;
  501. pl->is_dynamic = false;
  502. } else
  503. pl->is_dynamic = true;
  504. pl->cl = cl;
  505. pl->s = new_listener(srcaddr, port, &pl->plug,
  506. !conf_get_bool(conf, CONF_lport_acceptall),
  507. conf, address_family);
  508. if ((err = sk_socket_error(pl->s)) != NULL) {
  509. char *err_ret = dupstr(err);
  510. sk_close(pl->s);
  511. free_portlistener_state(pl);
  512. *pl_ret = NULL;
  513. return err_ret;
  514. }
  515. return NULL;
  516. }
  517. static char *pfd_log_close_msg(Channel *chan)
  518. {
  519. return dupstr("Forwarded port closed");
  520. }
  521. static void pfd_close(struct PortForwarding *pf)
  522. {
  523. if (!pf)
  524. return;
  525. sk_close(pf->s);
  526. free_portfwd_state(pf);
  527. }
  528. /*
  529. * Terminate a listener.
  530. */
  531. static void pfl_terminate(struct PortListener *pl)
  532. {
  533. if (!pl)
  534. return;
  535. sk_close(pl->s);
  536. free_portlistener_state(pl);
  537. }
  538. static void pfd_set_input_wanted(Channel *chan, bool wanted)
  539. {
  540. assert(chan->vt == &PortForwarding_channelvt);
  541. PortForwarding *pf = container_of(chan, PortForwarding, chan);
  542. pf->input_wanted = wanted;
  543. sk_set_frozen(pf->s, !pf->input_wanted);
  544. }
  545. static void pfd_chan_free(Channel *chan)
  546. {
  547. assert(chan->vt == &PortForwarding_channelvt);
  548. PortForwarding *pf = container_of(chan, PortForwarding, chan);
  549. pfd_close(pf);
  550. }
  551. /*
  552. * Called to send data down the raw connection.
  553. */
  554. static size_t pfd_send(
  555. Channel *chan, bool is_stderr, const void *data, size_t len)
  556. {
  557. assert(chan->vt == &PortForwarding_channelvt);
  558. PortForwarding *pf = container_of(chan, PortForwarding, chan);
  559. return sk_write(pf->s, data, len);
  560. }
  561. static void pfd_send_eof(Channel *chan)
  562. {
  563. assert(chan->vt == &PortForwarding_channelvt);
  564. PortForwarding *pf = container_of(chan, PortForwarding, chan);
  565. sk_write_eof(pf->s);
  566. }
  567. static void pfd_open_confirmation(Channel *chan)
  568. {
  569. assert(chan->vt == &PortForwarding_channelvt);
  570. PortForwarding *pf = container_of(chan, PortForwarding, chan);
  571. pf->ready = true;
  572. sk_set_frozen(pf->s, false);
  573. sk_write(pf->s, NULL, 0);
  574. if (pf->socksbuf) {
  575. sshfwd_write(pf->c, pf->socksbuf->u + pf->socksbuf_consumed,
  576. pf->socksbuf->len - pf->socksbuf_consumed);
  577. strbuf_free(pf->socksbuf);
  578. pf->socksbuf = NULL;
  579. }
  580. }
  581. static void pfd_open_failure(Channel *chan, const char *errtext)
  582. {
  583. assert(chan->vt == &PortForwarding_channelvt);
  584. PortForwarding *pf = container_of(chan, PortForwarding, chan);
  585. logeventf(pf->cl->logctx,
  586. "Forwarded connection refused by remote%s%s",
  587. errtext ? ": " : "", errtext ? errtext : "");
  588. }
  589. /* ----------------------------------------------------------------------
  590. * Code to manage the complete set of currently active port
  591. * forwardings, and update it from Conf.
  592. */
  593. struct PortFwdRecord {
  594. enum { DESTROY, KEEP, CREATE } status;
  595. int type;
  596. unsigned sport, dport;
  597. char *saddr, *daddr;
  598. char *sserv, *dserv;
  599. struct ssh_rportfwd *remote;
  600. int addressfamily;
  601. struct PortListener *local;
  602. };
  603. static int pfr_cmp(void *av, void *bv)
  604. {
  605. PortFwdRecord *a = (PortFwdRecord *) av;
  606. PortFwdRecord *b = (PortFwdRecord *) bv;
  607. int i;
  608. if (a->type > b->type)
  609. return +1;
  610. if (a->type < b->type)
  611. return -1;
  612. if (a->addressfamily > b->addressfamily)
  613. return +1;
  614. if (a->addressfamily < b->addressfamily)
  615. return -1;
  616. if ( (i = nullstrcmp(a->saddr, b->saddr)) != 0)
  617. return i < 0 ? -1 : +1;
  618. if (a->sport > b->sport)
  619. return +1;
  620. if (a->sport < b->sport)
  621. return -1;
  622. if (a->type != 'D') {
  623. if ( (i = nullstrcmp(a->daddr, b->daddr)) != 0)
  624. return i < 0 ? -1 : +1;
  625. if (a->dport > b->dport)
  626. return +1;
  627. if (a->dport < b->dport)
  628. return -1;
  629. }
  630. return 0;
  631. }
  632. static void pfr_free(PortFwdRecord *pfr)
  633. {
  634. /* Dispose of any listening socket. */
  635. if (pfr->local)
  636. pfl_terminate(pfr->local);
  637. sfree(pfr->saddr);
  638. sfree(pfr->daddr);
  639. sfree(pfr->sserv);
  640. sfree(pfr->dserv);
  641. sfree(pfr);
  642. }
  643. struct PortFwdManager {
  644. ConnectionLayer *cl;
  645. Conf *conf;
  646. tree234 *forwardings;
  647. };
  648. PortFwdManager *portfwdmgr_new(ConnectionLayer *cl)
  649. {
  650. PortFwdManager *mgr = snew(PortFwdManager);
  651. mgr->cl = cl;
  652. mgr->conf = NULL;
  653. mgr->forwardings = newtree234(pfr_cmp);
  654. return mgr;
  655. }
  656. void portfwdmgr_close(PortFwdManager *mgr, PortFwdRecord *pfr)
  657. {
  658. PortFwdRecord *realpfr = del234(mgr->forwardings, pfr);
  659. if (realpfr == pfr)
  660. pfr_free(pfr);
  661. }
  662. void portfwdmgr_close_all(PortFwdManager *mgr)
  663. {
  664. PortFwdRecord *pfr;
  665. while ((pfr = delpos234(mgr->forwardings, 0)) != NULL)
  666. pfr_free(pfr);
  667. }
  668. void portfwdmgr_free(PortFwdManager *mgr)
  669. {
  670. portfwdmgr_close_all(mgr);
  671. freetree234(mgr->forwardings);
  672. if (mgr->conf)
  673. conf_free(mgr->conf);
  674. sfree(mgr);
  675. }
  676. void portfwdmgr_config(PortFwdManager *mgr, Conf *conf)
  677. {
  678. PortFwdRecord *pfr;
  679. int i;
  680. char *key, *val;
  681. if (mgr->conf)
  682. conf_free(mgr->conf);
  683. mgr->conf = conf_copy(conf);
  684. /*
  685. * Go through the existing port forwardings and tag them
  686. * with status==DESTROY. Any that we want to keep will be
  687. * re-enabled (status==KEEP) as we go through the
  688. * configuration and find out which bits are the same as
  689. * they were before.
  690. */
  691. for (i = 0; (pfr = index234(mgr->forwardings, i)) != NULL; i++)
  692. pfr->status = DESTROY;
  693. for (val = conf_get_str_strs(conf, CONF_portfwd, NULL, &key);
  694. val != NULL;
  695. val = conf_get_str_strs(conf, CONF_portfwd, key, &key)) {
  696. char *kp, *kp2, *vp, *vp2;
  697. char address_family, type;
  698. int sport, dport, sserv, dserv;
  699. char *sports, *dports, *saddr, *host;
  700. kp = key;
  701. address_family = 'A';
  702. type = 'L';
  703. if (*kp == 'A' || *kp == '4' || *kp == '6')
  704. address_family = *kp++;
  705. if (*kp == 'L' || *kp == 'R')
  706. type = *kp++;
  707. if ((kp2 = host_strchr(kp, ':')) != NULL) {
  708. /*
  709. * There's a colon in the middle of the source port
  710. * string, which means that the part before it is
  711. * actually a source address.
  712. */
  713. char *saddr_tmp = dupprintf("%.*s", (int)(kp2 - kp), kp);
  714. saddr = host_strduptrim(saddr_tmp);
  715. sfree(saddr_tmp);
  716. sports = kp2+1;
  717. } else {
  718. saddr = NULL;
  719. sports = kp;
  720. }
  721. sport = atoi(sports);
  722. sserv = 0;
  723. if (sport == 0) {
  724. sserv = 1;
  725. sport = net_service_lookup(sports);
  726. if (!sport) {
  727. logeventf(mgr->cl->logctx, "Service lookup failed for source"
  728. " port \"%s\"", sports);
  729. }
  730. }
  731. if (type == 'L' && !strcmp(val, "D")) {
  732. /* dynamic forwarding */
  733. host = NULL;
  734. dports = NULL;
  735. dport = -1;
  736. dserv = 0;
  737. type = 'D';
  738. } else {
  739. /* ordinary forwarding */
  740. vp = val;
  741. vp2 = vp + host_strcspn(vp, ":");
  742. host = dupprintf("%.*s", (int)(vp2 - vp), vp);
  743. if (*vp2)
  744. vp2++;
  745. dports = vp2;
  746. dport = atoi(dports);
  747. dserv = 0;
  748. if (dport == 0) {
  749. dserv = 1;
  750. dport = net_service_lookup(dports);
  751. if (!dport) {
  752. logeventf(mgr->cl->logctx,
  753. "Service lookup failed for destination"
  754. " port \"%s\"", dports);
  755. }
  756. }
  757. }
  758. if (sport && dport) {
  759. /* Set up a description of the source port. */
  760. pfr = snew(PortFwdRecord);
  761. pfr->type = type;
  762. pfr->saddr = saddr;
  763. pfr->sserv = sserv ? dupstr(sports) : NULL;
  764. pfr->sport = sport;
  765. pfr->daddr = host;
  766. pfr->dserv = dserv ? dupstr(dports) : NULL;
  767. pfr->dport = dport;
  768. pfr->local = NULL;
  769. pfr->remote = NULL;
  770. pfr->addressfamily = (address_family == '4' ? ADDRTYPE_IPV4 :
  771. address_family == '6' ? ADDRTYPE_IPV6 :
  772. ADDRTYPE_UNSPEC);
  773. PortFwdRecord *existing = add234(mgr->forwardings, pfr);
  774. if (existing != pfr) {
  775. if (existing->status == DESTROY) {
  776. /*
  777. * We already have a port forwarding up and running
  778. * with precisely these parameters. Hence, no need
  779. * to do anything; simply re-tag the existing one
  780. * as KEEP.
  781. */
  782. existing->status = KEEP;
  783. }
  784. /*
  785. * Anything else indicates that there was a duplicate
  786. * in our input, which we'll silently ignore.
  787. */
  788. pfr_free(pfr);
  789. } else {
  790. pfr->status = CREATE;
  791. }
  792. } else {
  793. sfree(saddr);
  794. sfree(host);
  795. }
  796. }
  797. /*
  798. * Now go through and destroy any port forwardings which were
  799. * not re-enabled.
  800. */
  801. for (i = 0; (pfr = index234(mgr->forwardings, i)) != NULL; i++) {
  802. if (pfr->status == DESTROY) {
  803. char *message;
  804. message = dupprintf("%s port forwarding from %s%s%d",
  805. pfr->type == 'L' ? "local" :
  806. pfr->type == 'R' ? "remote" : "dynamic",
  807. pfr->saddr ? pfr->saddr : "",
  808. pfr->saddr ? ":" : "",
  809. pfr->sport);
  810. if (pfr->type != 'D') {
  811. char *msg2 = dupprintf("%s to %s:%d", message,
  812. pfr->daddr, pfr->dport);
  813. sfree(message);
  814. message = msg2;
  815. }
  816. logeventf(mgr->cl->logctx, "Cancelling %s", message);
  817. sfree(message);
  818. /* pfr->remote or pfr->local may be NULL if setting up a
  819. * forwarding failed. */
  820. if (pfr->remote) {
  821. /*
  822. * Cancel the port forwarding at the server
  823. * end.
  824. *
  825. * Actually closing the listening port on the server
  826. * side may fail - because in SSH-1 there's no message
  827. * in the protocol to request it!
  828. *
  829. * Instead, we simply remove the record of the
  830. * forwarding from our local end, so that any
  831. * connections the server tries to make on it are
  832. * rejected.
  833. */
  834. ssh_rportfwd_remove(mgr->cl, pfr->remote);
  835. pfr->remote = NULL;
  836. } else if (pfr->local) {
  837. pfl_terminate(pfr->local);
  838. pfr->local = NULL;
  839. }
  840. delpos234(mgr->forwardings, i);
  841. pfr_free(pfr);
  842. i--; /* so we don't skip one in the list */
  843. }
  844. }
  845. /*
  846. * And finally, set up any new port forwardings (status==CREATE).
  847. */
  848. for (i = 0; (pfr = index234(mgr->forwardings, i)) != NULL; i++) {
  849. if (pfr->status == CREATE) {
  850. char *sportdesc, *dportdesc;
  851. sportdesc = dupprintf("%s%s%s%s%d%s",
  852. pfr->saddr ? pfr->saddr : "",
  853. pfr->saddr ? ":" : "",
  854. pfr->sserv ? pfr->sserv : "",
  855. pfr->sserv ? "(" : "",
  856. pfr->sport,
  857. pfr->sserv ? ")" : "");
  858. if (pfr->type == 'D') {
  859. dportdesc = NULL;
  860. } else {
  861. dportdesc = dupprintf("%s:%s%s%d%s",
  862. pfr->daddr,
  863. pfr->dserv ? pfr->dserv : "",
  864. pfr->dserv ? "(" : "",
  865. pfr->dport,
  866. pfr->dserv ? ")" : "");
  867. }
  868. if (pfr->type == 'L') {
  869. char *err = pfl_listen(pfr->daddr, pfr->dport,
  870. pfr->saddr, pfr->sport,
  871. mgr->cl, conf, &pfr->local,
  872. pfr->addressfamily);
  873. logeventf(mgr->cl->logctx,
  874. "Local %sport %s forwarding to %s%s%s",
  875. pfr->addressfamily == ADDRTYPE_IPV4 ? "IPv4 " :
  876. pfr->addressfamily == ADDRTYPE_IPV6 ? "IPv6 " : "",
  877. sportdesc, dportdesc,
  878. err ? " failed: " : "", err ? err : "");
  879. if (err)
  880. sfree(err);
  881. } else if (pfr->type == 'D') {
  882. char *err = pfl_listen(NULL, -1, pfr->saddr, pfr->sport,
  883. mgr->cl, conf, &pfr->local,
  884. pfr->addressfamily);
  885. logeventf(mgr->cl->logctx,
  886. "Local %sport %s SOCKS dynamic forwarding%s%s",
  887. pfr->addressfamily == ADDRTYPE_IPV4 ? "IPv4 " :
  888. pfr->addressfamily == ADDRTYPE_IPV6 ? "IPv6 " : "",
  889. sportdesc,
  890. err ? " failed: " : "", err ? err : "");
  891. if (err)
  892. sfree(err);
  893. } else {
  894. const char *shost;
  895. if (pfr->saddr) {
  896. shost = pfr->saddr;
  897. } else if (conf_get_bool(conf, CONF_rport_acceptall)) {
  898. shost = "";
  899. } else {
  900. shost = "localhost";
  901. }
  902. pfr->remote = ssh_rportfwd_alloc(
  903. mgr->cl, shost, pfr->sport, pfr->daddr, pfr->dport,
  904. pfr->addressfamily, sportdesc, pfr, NULL);
  905. if (!pfr->remote) {
  906. logeventf(mgr->cl->logctx,
  907. "Duplicate remote port forwarding to %s:%d",
  908. pfr->daddr, pfr->dport);
  909. pfr_free(pfr);
  910. } else {
  911. logeventf(mgr->cl->logctx, "Requesting remote port %s"
  912. " forward to %s", sportdesc, dportdesc);
  913. }
  914. }
  915. sfree(sportdesc);
  916. sfree(dportdesc);
  917. }
  918. }
  919. }
  920. bool portfwdmgr_listen(PortFwdManager *mgr, const char *host, int port,
  921. const char *keyhost, int keyport, Conf *conf)
  922. {
  923. PortFwdRecord *pfr;
  924. pfr = snew(PortFwdRecord);
  925. pfr->type = 'L';
  926. pfr->saddr = host ? dupstr(host) : NULL;
  927. pfr->daddr = keyhost ? dupstr(keyhost) : NULL;
  928. pfr->sserv = pfr->dserv = NULL;
  929. pfr->sport = port;
  930. pfr->dport = keyport;
  931. pfr->local = NULL;
  932. pfr->remote = NULL;
  933. pfr->addressfamily = ADDRTYPE_UNSPEC;
  934. PortFwdRecord *existing = add234(mgr->forwardings, pfr);
  935. if (existing != pfr) {
  936. /*
  937. * We had this record already. Return failure.
  938. */
  939. pfr_free(pfr);
  940. return false;
  941. }
  942. char *err = pfl_listen(keyhost, keyport, host, port,
  943. mgr->cl, conf, &pfr->local, pfr->addressfamily);
  944. logeventf(mgr->cl->logctx,
  945. "%s on port %s:%d to forward to client%s%s",
  946. err ? "Failed to listen" : "Listening", host, port,
  947. err ? ": " : "", err ? err : "");
  948. if (err) {
  949. sfree(err);
  950. del234(mgr->forwardings, pfr);
  951. pfr_free(pfr);
  952. return false;
  953. }
  954. return true;
  955. }
  956. bool portfwdmgr_unlisten(PortFwdManager *mgr, const char *host, int port)
  957. {
  958. PortFwdRecord pfr_key;
  959. pfr_key.type = 'L';
  960. /* Safe to cast the const away here, because it will only be used
  961. * by pfr_cmp, which won't write to the string */
  962. pfr_key.saddr = pfr_key.daddr = (char *)host;
  963. pfr_key.sserv = pfr_key.dserv = NULL;
  964. pfr_key.sport = pfr_key.dport = port;
  965. pfr_key.local = NULL;
  966. pfr_key.remote = NULL;
  967. pfr_key.addressfamily = ADDRTYPE_UNSPEC;
  968. PortFwdRecord *pfr = del234(mgr->forwardings, &pfr_key);
  969. if (!pfr)
  970. return false;
  971. logeventf(mgr->cl->logctx, "Closing listening port %s:%d", host, port);
  972. pfr_free(pfr);
  973. return true;
  974. }
  975. /*
  976. * Called when receiving a PORT OPEN from the server to make a
  977. * connection to a destination host.
  978. *
  979. * On success, returns NULL and fills in *pf_ret. On error, returns a
  980. * dynamically allocated error message string.
  981. */
  982. char *portfwdmgr_connect(PortFwdManager *mgr, Channel **chan_ret,
  983. char *hostname, int port, SshChannel *c,
  984. int addressfamily)
  985. {
  986. SockAddr *addr;
  987. const char *err;
  988. char *dummy_realhost = NULL;
  989. struct PortForwarding *pf;
  990. /*
  991. * Try to find host.
  992. */
  993. addr = name_lookup(hostname, port, &dummy_realhost, mgr->conf,
  994. addressfamily, NULL, NULL);
  995. if ((err = sk_addr_error(addr)) != NULL) {
  996. char *err_ret = dupstr(err);
  997. sk_addr_free(addr);
  998. sfree(dummy_realhost);
  999. return err_ret;
  1000. }
  1001. /*
  1002. * Open socket.
  1003. */
  1004. pf = new_portfwd_state();
  1005. *chan_ret = &pf->chan;
  1006. pf->plug.vt = &PortForwarding_plugvt;
  1007. pf->chan.initial_fixed_window_size = 0;
  1008. pf->chan.vt = &PortForwarding_channelvt;
  1009. pf->input_wanted = true;
  1010. pf->ready = true;
  1011. pf->c = c;
  1012. pf->cl = mgr->cl;
  1013. pf->socks_state = SOCKS_NONE;
  1014. pf->s = new_connection(addr, dummy_realhost, port,
  1015. false, true, false, false, &pf->plug, mgr->conf);
  1016. sfree(dummy_realhost);
  1017. if ((err = sk_socket_error(pf->s)) != NULL) {
  1018. char *err_ret = dupstr(err);
  1019. sk_close(pf->s);
  1020. free_portfwd_state(pf);
  1021. *chan_ret = NULL;
  1022. return err_ret;
  1023. }
  1024. return NULL;
  1025. }