gpxe/src/net/udp/slam.c

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