Du kannst nicht mehr als 25 Themen auswählen Themen müssen mit entweder einem Buchstaben oder einer Ziffer beginnen. Sie können Bindestriche („-“) enthalten und bis zu 35 Zeichen lang sein.

slam.c 21KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810
  1. /*
  2. * Copyright (C) 2008 Michael Brown <mbrown@fensystems.co.uk>.
  3. *
  4. * This program is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU General Public License as
  6. * published by the Free Software Foundation; either version 2 of the
  7. * License, or any later version.
  8. *
  9. * This program is distributed in the hope that it will be useful, but
  10. * WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU General Public License
  15. * along with this program; if not, write to the Free Software
  16. * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17. */
  18. #include <stdint.h>
  19. #include <stdlib.h>
  20. #include <string.h>
  21. #include <strings.h>
  22. #include <errno.h>
  23. #include <assert.h>
  24. #include <byteswap.h>
  25. #include <gpxe/features.h>
  26. #include <gpxe/iobuf.h>
  27. #include <gpxe/bitmap.h>
  28. #include <gpxe/xfer.h>
  29. #include <gpxe/open.h>
  30. #include <gpxe/uri.h>
  31. #include <gpxe/tcpip.h>
  32. #include <gpxe/timer.h>
  33. #include <gpxe/retry.h>
  34. /** @file
  35. *
  36. * Scalable Local Area Multicast protocol
  37. *
  38. * The SLAM protocol is supported only by Etherboot; it was designed
  39. * and implemented by Eric Biederman. A server implementation is
  40. * available in contrib/mini-slamd. There does not appear to be any
  41. * documentation beyond a few sparse comments in Etherboot's
  42. * proto_slam.c.
  43. *
  44. * SLAM packets use three types of data field:
  45. *
  46. * Nul : A single NUL (0) byte, used as a list terminator
  47. *
  48. * Raw : A block of raw data
  49. *
  50. * Int : A variable-length integer, in big-endian order. The length
  51. * of the integer is encoded in the most significant three bits.
  52. *
  53. * Packets received by the client have the following layout:
  54. *
  55. * Int : Transaction identifier. This is an opaque value.
  56. *
  57. * Int : Total number of bytes in the transfer.
  58. *
  59. * Int : Block size, in bytes.
  60. *
  61. * Int : Packet sequence number within the transfer (if this packet
  62. * contains data).
  63. *
  64. * Raw : Packet data (if this packet contains data).
  65. *
  66. * Packets transmitted by the client consist of a run-length-encoded
  67. * representation of the received-blocks bitmap, looking something
  68. * like:
  69. *
  70. * Int : Number of consecutive successfully-received packets
  71. * Int : Number of consecutive missing packets
  72. * Int : Number of consecutive successfully-received packets
  73. * Int : Number of consecutive missing packets
  74. * ....
  75. * Nul
  76. *
  77. */
  78. FEATURE ( FEATURE_PROTOCOL, "SLAM", DHCP_EB_FEATURE_SLAM, 1 );
  79. /** Default SLAM server port */
  80. #define SLAM_DEFAULT_PORT 10000
  81. /** Default SLAM multicast IP address */
  82. #define SLAM_DEFAULT_MULTICAST_IP \
  83. ( ( 239 << 24 ) | ( 255 << 16 ) | ( 1 << 8 ) | ( 1 << 0 ) )
  84. /** Default SLAM multicast port */
  85. #define SLAM_DEFAULT_MULTICAST_PORT 10000
  86. /** Maximum SLAM header length */
  87. #define SLAM_MAX_HEADER_LEN ( 7 /* transaction id */ + 7 /* total_bytes */ + \
  88. 7 /* block_size */ )
  89. /** Maximum number of blocks to request per NACK
  90. *
  91. * This is a policy decision equivalent to selecting a TCP window
  92. * size.
  93. */
  94. #define SLAM_MAX_BLOCKS_PER_NACK 4
  95. /** Maximum SLAM NACK length
  96. *
  97. * We only ever send a NACK for a single range of up to @c
  98. * SLAM_MAX_BLOCKS_PER_NACK blocks.
  99. */
  100. #define SLAM_MAX_NACK_LEN ( 7 /* block */ + 7 /* #blocks */ + 1 /* NUL */ )
  101. /** SLAM slave timeout */
  102. #define SLAM_SLAVE_TIMEOUT ( 1 * TICKS_PER_SEC )
  103. /** A SLAM request */
  104. struct slam_request {
  105. /** Reference counter */
  106. struct refcnt refcnt;
  107. /** Data transfer interface */
  108. struct xfer_interface xfer;
  109. /** Unicast socket */
  110. struct xfer_interface socket;
  111. /** Multicast socket */
  112. struct xfer_interface mc_socket;
  113. /** Master client retry timer */
  114. struct retry_timer master_timer;
  115. /** Slave client retry timer */
  116. struct retry_timer slave_timer;
  117. /** Cached header */
  118. uint8_t header[SLAM_MAX_HEADER_LEN];
  119. /** Size of cached header */
  120. size_t header_len;
  121. /** Total number of bytes in transfer */
  122. unsigned long total_bytes;
  123. /** Transfer block size */
  124. unsigned long block_size;
  125. /** Number of blocks in transfer */
  126. unsigned long num_blocks;
  127. /** Block bitmap */
  128. struct bitmap bitmap;
  129. /** NACK sent flag */
  130. int nack_sent;
  131. };
  132. /**
  133. * Free a SLAM request
  134. *
  135. * @v refcnt Reference counter
  136. */
  137. static void slam_free ( struct refcnt *refcnt ) {
  138. struct slam_request *slam =
  139. container_of ( refcnt, struct slam_request, refcnt );
  140. bitmap_free ( &slam->bitmap );
  141. free ( slam );
  142. }
  143. /**
  144. * Mark SLAM request as complete
  145. *
  146. * @v slam SLAM request
  147. * @v rc Return status code
  148. */
  149. static void slam_finished ( struct slam_request *slam, int rc ) {
  150. static const uint8_t slam_disconnect[] = { 0 };
  151. DBGC ( slam, "SLAM %p finished with status code %d (%s)\n",
  152. slam, rc, strerror ( rc ) );
  153. /* Send a disconnect message if we ever sent anything to the
  154. * server.
  155. */
  156. if ( slam->nack_sent ) {
  157. xfer_deliver_raw ( &slam->socket, slam_disconnect,
  158. sizeof ( slam_disconnect ) );
  159. }
  160. /* Stop the retry timers */
  161. stop_timer ( &slam->master_timer );
  162. stop_timer ( &slam->slave_timer );
  163. /* Close all data transfer interfaces */
  164. xfer_nullify ( &slam->socket );
  165. xfer_close ( &slam->socket, rc );
  166. xfer_nullify ( &slam->mc_socket );
  167. xfer_close ( &slam->mc_socket, rc );
  168. xfer_nullify ( &slam->xfer );
  169. xfer_close ( &slam->xfer, rc );
  170. }
  171. /****************************************************************************
  172. *
  173. * TX datapath
  174. *
  175. */
  176. /**
  177. * Add a variable-length value to a SLAM packet
  178. *
  179. * @v slam SLAM request
  180. * @v iobuf I/O buffer
  181. * @v value Value to add
  182. * @ret rc Return status code
  183. *
  184. * Adds a variable-length value to the end of an I/O buffer. Will
  185. * always leave at least one byte of tailroom in the I/O buffer (to
  186. * allow space for the terminating NUL).
  187. */
  188. static int slam_put_value ( struct slam_request *slam,
  189. struct io_buffer *iobuf, unsigned long value ) {
  190. uint8_t *data;
  191. size_t len;
  192. unsigned int i;
  193. /* Calculate variable length required to store value. Always
  194. * leave at least one byte in the I/O buffer.
  195. */
  196. len = ( ( flsl ( value ) + 10 ) / 8 );
  197. if ( len >= iob_tailroom ( iobuf ) ) {
  198. DBGC2 ( slam, "SLAM %p cannot add %zd-byte value\n",
  199. slam, len );
  200. return -ENOBUFS;
  201. }
  202. /* There is no valid way within the protocol that we can end
  203. * up trying to push a full-sized long (i.e. without space for
  204. * the length encoding).
  205. */
  206. assert ( len <= sizeof ( value ) );
  207. /* Add value */
  208. data = iob_put ( iobuf, len );
  209. for ( i = len ; i-- ; ) {
  210. data[i] = value;
  211. value >>= 8;
  212. }
  213. *data |= ( len << 5 );
  214. assert ( value == 0 );
  215. return 0;
  216. }
  217. /**
  218. * Send SLAM NACK packet
  219. *
  220. * @v slam SLAM request
  221. * @ret rc Return status code
  222. */
  223. static int slam_tx_nack ( struct slam_request *slam ) {
  224. struct io_buffer *iobuf;
  225. unsigned long first_block;
  226. unsigned long num_blocks;
  227. uint8_t *nul;
  228. int rc;
  229. /* Mark NACK as sent, so that we know we have to disconnect later */
  230. slam->nack_sent = 1;
  231. /* Allocate I/O buffer */
  232. iobuf = xfer_alloc_iob ( &slam->socket, SLAM_MAX_NACK_LEN );
  233. if ( ! iobuf ) {
  234. DBGC ( slam, "SLAM %p could not allocate I/O buffer\n",
  235. slam );
  236. return -ENOMEM;
  237. }
  238. /* Construct NACK. We always request only a single packet;
  239. * this allows us to force multicast-TFTP-style flow control
  240. * on the SLAM server, which will otherwise just blast the
  241. * data out as fast as it can. On a gigabit network, without
  242. * RX checksumming, this would inevitably cause packet drops.
  243. */
  244. first_block = bitmap_first_gap ( &slam->bitmap );
  245. for ( num_blocks = 1 ; ; num_blocks++ ) {
  246. if ( num_blocks >= SLAM_MAX_BLOCKS_PER_NACK )
  247. break;
  248. if ( ( first_block + num_blocks ) >= slam->num_blocks )
  249. break;
  250. if ( bitmap_test ( &slam->bitmap,
  251. ( first_block + num_blocks ) ) )
  252. break;
  253. }
  254. if ( first_block ) {
  255. DBGCP ( slam, "SLAM %p transmitting NACK for blocks "
  256. "%ld-%ld\n", slam, first_block,
  257. ( first_block + num_blocks - 1 ) );
  258. } else {
  259. DBGC ( slam, "SLAM %p transmitting initial NACK for blocks "
  260. "0-%ld\n", slam, ( num_blocks - 1 ) );
  261. }
  262. if ( ( rc = slam_put_value ( slam, iobuf, first_block ) ) != 0 )
  263. return rc;
  264. if ( ( rc = slam_put_value ( slam, iobuf, num_blocks ) ) != 0 )
  265. return rc;
  266. nul = iob_put ( iobuf, 1 );
  267. *nul = 0;
  268. /* Transmit packet */
  269. return xfer_deliver_iob ( &slam->socket, iobuf );
  270. }
  271. /**
  272. * Handle SLAM master client retry timer expiry
  273. *
  274. * @v timer Master retry timer
  275. * @v fail Failure indicator
  276. */
  277. static void slam_master_timer_expired ( struct retry_timer *timer,
  278. int fail ) {
  279. struct slam_request *slam =
  280. container_of ( timer, struct slam_request, master_timer );
  281. if ( fail ) {
  282. /* Allow timer to stop running. We will terminate the
  283. * connection only if the slave timer times out.
  284. */
  285. DBGC ( slam, "SLAM %p giving up acting as master client\n",
  286. slam );
  287. } else {
  288. /* Retransmit NACK */
  289. start_timer ( timer );
  290. slam_tx_nack ( slam );
  291. }
  292. }
  293. /**
  294. * Handle SLAM slave client retry timer expiry
  295. *
  296. * @v timer Master retry timer
  297. * @v fail Failure indicator
  298. */
  299. static void slam_slave_timer_expired ( struct retry_timer *timer,
  300. int fail ) {
  301. struct slam_request *slam =
  302. container_of ( timer, struct slam_request, slave_timer );
  303. if ( fail ) {
  304. /* Terminate connection */
  305. slam_finished ( slam, -ETIMEDOUT );
  306. } else {
  307. /* Try sending a NACK */
  308. DBGC ( slam, "SLAM %p trying to become master client\n",
  309. slam );
  310. start_timer ( timer );
  311. slam_tx_nack ( slam );
  312. }
  313. }
  314. /****************************************************************************
  315. *
  316. * RX datapath
  317. *
  318. */
  319. /**
  320. * Read and strip a variable-length value from a SLAM packet
  321. *
  322. * @v slam SLAM request
  323. * @v iobuf I/O buffer
  324. * @v value Value to fill in, or NULL to ignore value
  325. * @ret rc Return status code
  326. *
  327. * Reads a variable-length value from the start of the I/O buffer.
  328. */
  329. static int slam_pull_value ( struct slam_request *slam,
  330. struct io_buffer *iobuf,
  331. unsigned long *value ) {
  332. uint8_t *data;
  333. size_t len;
  334. /* Sanity check */
  335. if ( iob_len ( iobuf ) == 0 ) {
  336. DBGC ( slam, "SLAM %p empty value\n", slam );
  337. return -EINVAL;
  338. }
  339. /* Read and verify length of value */
  340. data = iobuf->data;
  341. len = ( *data >> 5 );
  342. if ( ( len == 0 ) ||
  343. ( value && ( len > sizeof ( *value ) ) ) ) {
  344. DBGC ( slam, "SLAM %p invalid value length %zd bytes\n",
  345. slam, len );
  346. return -EINVAL;
  347. }
  348. if ( len > iob_len ( iobuf ) ) {
  349. DBGC ( slam, "SLAM %p value extends beyond I/O buffer\n",
  350. slam );
  351. return -EINVAL;
  352. }
  353. /* Read value */
  354. iob_pull ( iobuf, len );
  355. *value = ( *data & 0x1f );
  356. while ( --len ) {
  357. *value <<= 8;
  358. *value |= *(++data);
  359. }
  360. return 0;
  361. }
  362. /**
  363. * Read and strip SLAM header
  364. *
  365. * @v slam SLAM request
  366. * @v iobuf I/O buffer
  367. * @ret rc Return status code
  368. */
  369. static int slam_pull_header ( struct slam_request *slam,
  370. struct io_buffer *iobuf ) {
  371. void *header = iobuf->data;
  372. int rc;
  373. /* If header matches cached header, just pull it and return */
  374. if ( ( slam->header_len <= iob_len ( iobuf ) ) &&
  375. ( memcmp ( slam->header, iobuf->data, slam->header_len ) == 0 )){
  376. iob_pull ( iobuf, slam->header_len );
  377. return 0;
  378. }
  379. DBGC ( slam, "SLAM %p detected changed header; resetting\n", slam );
  380. /* Read and strip transaction ID, total number of bytes, and
  381. * block size.
  382. */
  383. if ( ( rc = slam_pull_value ( slam, iobuf, NULL ) ) != 0 )
  384. return rc;
  385. if ( ( rc = slam_pull_value ( slam, iobuf,
  386. &slam->total_bytes ) ) != 0 )
  387. return rc;
  388. if ( ( rc = slam_pull_value ( slam, iobuf,
  389. &slam->block_size ) ) != 0 )
  390. return rc;
  391. /* Update the cached header */
  392. slam->header_len = ( iobuf->data - header );
  393. assert ( slam->header_len <= sizeof ( slam->header ) );
  394. memcpy ( slam->header, header, slam->header_len );
  395. /* Calculate number of blocks */
  396. slam->num_blocks = ( ( slam->total_bytes + slam->block_size - 1 ) /
  397. slam->block_size );
  398. DBGC ( slam, "SLAM %p has total bytes %ld, block size %ld, num "
  399. "blocks %ld\n", slam, slam->total_bytes, slam->block_size,
  400. slam->num_blocks );
  401. /* Discard and reset the bitmap */
  402. bitmap_free ( &slam->bitmap );
  403. memset ( &slam->bitmap, 0, sizeof ( slam->bitmap ) );
  404. /* Allocate a new bitmap */
  405. if ( ( rc = bitmap_resize ( &slam->bitmap,
  406. slam->num_blocks ) ) != 0 ) {
  407. /* Failure to allocate a bitmap is fatal */
  408. DBGC ( slam, "SLAM %p could not allocate bitmap for %ld "
  409. "blocks: %s\n", slam, slam->num_blocks,
  410. strerror ( rc ) );
  411. slam_finished ( slam, rc );
  412. return rc;
  413. }
  414. /* Notify recipient of file size */
  415. xfer_seek ( &slam->xfer, slam->total_bytes, SEEK_SET );
  416. return 0;
  417. }
  418. /**
  419. * Receive SLAM data packet
  420. *
  421. * @v mc_socket SLAM multicast socket
  422. * @v iobuf I/O buffer
  423. * @ret rc Return status code
  424. */
  425. static int slam_mc_socket_deliver ( struct xfer_interface *mc_socket,
  426. struct io_buffer *iobuf,
  427. struct xfer_metadata *rx_meta __unused ) {
  428. struct slam_request *slam =
  429. container_of ( mc_socket, struct slam_request, mc_socket );
  430. struct xfer_metadata meta;
  431. unsigned long packet;
  432. size_t len;
  433. int rc;
  434. /* Stop the master client timer. Restart the slave client timer. */
  435. stop_timer ( &slam->master_timer );
  436. stop_timer ( &slam->slave_timer );
  437. start_timer_fixed ( &slam->slave_timer, SLAM_SLAVE_TIMEOUT );
  438. /* Read and strip packet header */
  439. if ( ( rc = slam_pull_header ( slam, iobuf ) ) != 0 )
  440. goto err_discard;
  441. /* Read and strip packet number */
  442. if ( ( rc = slam_pull_value ( slam, iobuf, &packet ) ) != 0 )
  443. goto err_discard;
  444. /* Sanity check packet number */
  445. if ( packet >= slam->num_blocks ) {
  446. DBGC ( slam, "SLAM %p received out-of-range packet %ld "
  447. "(num_blocks=%ld)\n", slam, packet, slam->num_blocks );
  448. rc = -EINVAL;
  449. goto err_discard;
  450. }
  451. /* Sanity check length */
  452. len = iob_len ( iobuf );
  453. if ( len > slam->block_size ) {
  454. DBGC ( slam, "SLAM %p received oversize packet of %zd bytes "
  455. "(block_size=%ld)\n", slam, len, slam->block_size );
  456. rc = -EINVAL;
  457. goto err_discard;
  458. }
  459. if ( ( packet != ( slam->num_blocks - 1 ) ) &&
  460. ( len < slam->block_size ) ) {
  461. DBGC ( slam, "SLAM %p received short packet of %zd bytes "
  462. "(block_size=%ld)\n", slam, len, slam->block_size );
  463. rc = -EINVAL;
  464. goto err_discard;
  465. }
  466. /* If we have already seen this packet, discard it */
  467. if ( bitmap_test ( &slam->bitmap, packet ) ) {
  468. goto discard;
  469. }
  470. /* Pass to recipient */
  471. memset ( &meta, 0, sizeof ( meta ) );
  472. meta.whence = SEEK_SET;
  473. meta.offset = ( packet * slam->block_size );
  474. if ( ( rc = xfer_deliver_iob_meta ( &slam->xfer, iobuf,
  475. &meta ) ) != 0 )
  476. goto err;
  477. /* Mark block as received */
  478. bitmap_set ( &slam->bitmap, packet );
  479. /* If we have received all blocks, terminate */
  480. if ( bitmap_full ( &slam->bitmap ) )
  481. slam_finished ( slam, 0 );
  482. return 0;
  483. err_discard:
  484. discard:
  485. free_iob ( iobuf );
  486. err:
  487. return rc;
  488. }
  489. /**
  490. * Receive SLAM non-data packet
  491. *
  492. * @v socket SLAM unicast socket
  493. * @v iobuf I/O buffer
  494. * @ret rc Return status code
  495. */
  496. static int slam_socket_deliver ( struct xfer_interface *socket,
  497. struct io_buffer *iobuf,
  498. struct xfer_metadata *rx_meta __unused ) {
  499. struct slam_request *slam =
  500. container_of ( socket, struct slam_request, socket );
  501. int rc;
  502. /* Restart the master client timer */
  503. stop_timer ( &slam->master_timer );
  504. start_timer ( &slam->master_timer );
  505. /* Read and strip packet header */
  506. if ( ( rc = slam_pull_header ( slam, iobuf ) ) != 0 )
  507. goto discard;
  508. /* Sanity check */
  509. if ( iob_len ( iobuf ) != 0 ) {
  510. DBGC ( slam, "SLAM %p received trailing garbage:\n", slam );
  511. DBGC_HD ( slam, iobuf->data, iob_len ( iobuf ) );
  512. rc = -EINVAL;
  513. goto discard;
  514. }
  515. /* Discard packet */
  516. free_iob ( iobuf );
  517. /* Send NACK in reply */
  518. slam_tx_nack ( slam );
  519. return 0;
  520. discard:
  521. free_iob ( iobuf );
  522. return rc;
  523. }
  524. /**
  525. * Close SLAM unicast socket
  526. *
  527. * @v socket SLAM unicast socket
  528. * @v rc Reason for close
  529. */
  530. static void slam_socket_close ( struct xfer_interface *socket, int rc ) {
  531. struct slam_request *slam =
  532. container_of ( socket, struct slam_request, socket );
  533. DBGC ( slam, "SLAM %p unicast socket closed: %s\n",
  534. slam, strerror ( rc ) );
  535. slam_finished ( slam, rc );
  536. }
  537. /** SLAM unicast socket data transfer operations */
  538. static struct xfer_interface_operations slam_socket_operations = {
  539. .close = slam_socket_close,
  540. .vredirect = xfer_vopen,
  541. .window = unlimited_xfer_window,
  542. .alloc_iob = default_xfer_alloc_iob,
  543. .deliver_iob = slam_socket_deliver,
  544. .deliver_raw = xfer_deliver_as_iob,
  545. };
  546. /**
  547. * Close SLAM multicast socket
  548. *
  549. * @v mc_socket SLAM multicast socket
  550. * @v rc Reason for close
  551. */
  552. static void slam_mc_socket_close ( struct xfer_interface *mc_socket, int rc ){
  553. struct slam_request *slam =
  554. container_of ( mc_socket, struct slam_request, mc_socket );
  555. DBGC ( slam, "SLAM %p multicast socket closed: %s\n",
  556. slam, strerror ( rc ) );
  557. slam_finished ( slam, rc );
  558. }
  559. /** SLAM multicast socket data transfer operations */
  560. static struct xfer_interface_operations slam_mc_socket_operations = {
  561. .close = slam_mc_socket_close,
  562. .vredirect = xfer_vopen,
  563. .window = unlimited_xfer_window,
  564. .alloc_iob = default_xfer_alloc_iob,
  565. .deliver_iob = slam_mc_socket_deliver,
  566. .deliver_raw = xfer_deliver_as_iob,
  567. };
  568. /****************************************************************************
  569. *
  570. * Data transfer interface
  571. *
  572. */
  573. /**
  574. * Close SLAM data transfer interface
  575. *
  576. * @v xfer SLAM data transfer interface
  577. * @v rc Reason for close
  578. */
  579. static void slam_xfer_close ( struct xfer_interface *xfer, int rc ) {
  580. struct slam_request *slam =
  581. container_of ( xfer, struct slam_request, xfer );
  582. DBGC ( slam, "SLAM %p data transfer interface closed: %s\n",
  583. slam, strerror ( rc ) );
  584. slam_finished ( slam, rc );
  585. }
  586. /** SLAM data transfer operations */
  587. static struct xfer_interface_operations slam_xfer_operations = {
  588. .close = slam_xfer_close,
  589. .vredirect = ignore_xfer_vredirect,
  590. .window = unlimited_xfer_window,
  591. .alloc_iob = default_xfer_alloc_iob,
  592. .deliver_iob = xfer_deliver_as_raw,
  593. .deliver_raw = ignore_xfer_deliver_raw,
  594. };
  595. /**
  596. * Parse SLAM URI multicast address
  597. *
  598. * @v slam SLAM request
  599. * @v path Path portion of x-slam:// URI
  600. * @v address Socket address to fill in
  601. * @ret rc Return status code
  602. */
  603. static int slam_parse_multicast_address ( struct slam_request *slam,
  604. const char *path,
  605. struct sockaddr_in *address ) {
  606. char path_dup[ strlen ( path ) /* no +1 */ ];
  607. char *sep;
  608. char *end;
  609. /* Create temporary copy of path, minus the leading '/' */
  610. assert ( *path == '/' );
  611. memcpy ( path_dup, ( path + 1 ) , sizeof ( path_dup ) );
  612. /* Parse port, if present */
  613. sep = strchr ( path_dup, ':' );
  614. if ( sep ) {
  615. *(sep++) = '\0';
  616. address->sin_port = htons ( strtoul ( sep, &end, 0 ) );
  617. if ( *end != '\0' ) {
  618. DBGC ( slam, "SLAM %p invalid multicast port "
  619. "\"%s\"\n", slam, sep );
  620. return -EINVAL;
  621. }
  622. }
  623. /* Parse address */
  624. if ( inet_aton ( path_dup, &address->sin_addr ) == 0 ) {
  625. DBGC ( slam, "SLAM %p invalid multicast address \"%s\"\n",
  626. slam, path_dup );
  627. return -EINVAL;
  628. }
  629. return 0;
  630. }
  631. /**
  632. * Initiate a SLAM request
  633. *
  634. * @v xfer Data transfer interface
  635. * @v uri Uniform Resource Identifier
  636. * @ret rc Return status code
  637. */
  638. static int slam_open ( struct xfer_interface *xfer, struct uri *uri ) {
  639. static const struct sockaddr_in default_multicast = {
  640. .sin_family = AF_INET,
  641. .sin_port = htons ( SLAM_DEFAULT_MULTICAST_PORT ),
  642. .sin_addr = { htonl ( SLAM_DEFAULT_MULTICAST_IP ) },
  643. };
  644. struct slam_request *slam;
  645. struct sockaddr_tcpip server;
  646. struct sockaddr_in multicast;
  647. int rc;
  648. /* Sanity checks */
  649. if ( ! uri->host )
  650. return -EINVAL;
  651. /* Allocate and populate structure */
  652. slam = zalloc ( sizeof ( *slam ) );
  653. if ( ! slam )
  654. return -ENOMEM;
  655. slam->refcnt.free = slam_free;
  656. xfer_init ( &slam->xfer, &slam_xfer_operations, &slam->refcnt );
  657. xfer_init ( &slam->socket, &slam_socket_operations, &slam->refcnt );
  658. xfer_init ( &slam->mc_socket, &slam_mc_socket_operations,
  659. &slam->refcnt );
  660. slam->master_timer.expired = slam_master_timer_expired;
  661. slam->slave_timer.expired = slam_slave_timer_expired;
  662. /* Fake an invalid cached header of { 0x00, ... } */
  663. slam->header_len = 1;
  664. /* Fake parameters for initial NACK */
  665. slam->num_blocks = 1;
  666. if ( ( rc = bitmap_resize ( &slam->bitmap, 1 ) ) != 0 ) {
  667. DBGC ( slam, "SLAM %p could not allocate initial bitmap: "
  668. "%s\n", slam, strerror ( rc ) );
  669. goto err;
  670. }
  671. /* Open unicast socket */
  672. memset ( &server, 0, sizeof ( server ) );
  673. server.st_port = htons ( uri_port ( uri, SLAM_DEFAULT_PORT ) );
  674. if ( ( rc = xfer_open_named_socket ( &slam->socket, SOCK_DGRAM,
  675. ( struct sockaddr * ) &server,
  676. uri->host, NULL ) ) != 0 ) {
  677. DBGC ( slam, "SLAM %p could not open unicast socket: %s\n",
  678. slam, strerror ( rc ) );
  679. goto err;
  680. }
  681. /* Open multicast socket */
  682. memcpy ( &multicast, &default_multicast, sizeof ( multicast ) );
  683. if ( uri->path &&
  684. ( ( rc = slam_parse_multicast_address ( slam, uri->path,
  685. &multicast ) ) != 0 ) ) {
  686. goto err;
  687. }
  688. if ( ( rc = xfer_open_socket ( &slam->mc_socket, SOCK_DGRAM,
  689. ( struct sockaddr * ) &multicast,
  690. ( struct sockaddr * ) &multicast ) ) != 0 ) {
  691. DBGC ( slam, "SLAM %p could not open multicast socket: %s\n",
  692. slam, strerror ( rc ) );
  693. goto err;
  694. }
  695. /* Start slave retry timer */
  696. start_timer_fixed ( &slam->slave_timer, SLAM_SLAVE_TIMEOUT );
  697. /* Attach to parent interface, mortalise self, and return */
  698. xfer_plug_plug ( &slam->xfer, xfer );
  699. ref_put ( &slam->refcnt );
  700. return 0;
  701. err:
  702. slam_finished ( slam, rc );
  703. ref_put ( &slam->refcnt );
  704. return rc;
  705. }
  706. /** SLAM URI opener */
  707. struct uri_opener slam_uri_opener __uri_opener = {
  708. .scheme = "x-slam",
  709. .open = slam_open,
  710. };