[tftp] Make TFTP size requests abort transfer with an error
[people/mcb30/gpxe.git] / src / arch / i386 / interface / pxe / pxe_tftp.c
index c1d831b..0e3ca3c 100644 (file)
@@ -165,7 +165,8 @@ static struct xfer_interface_operations pxe_tftp_xfer_ops = {
  * @ret rc             Return status code
  */
 static int pxe_tftp_open ( uint32_t ipaddress, unsigned int port,
-                          const unsigned char *filename, size_t blksize ) {
+                          const unsigned char *filename, size_t blksize,
+                          int sizeonly ) {
        char uri_string[PXE_TFTP_URI_LEN];
        struct in_addr address;
        int rc;
@@ -185,7 +186,8 @@ static int pxe_tftp_open ( uint32_t ipaddress, unsigned int port,
        if ( blksize < TFTP_DEFAULT_BLKSIZE )
                blksize = TFTP_DEFAULT_BLKSIZE;
        snprintf ( uri_string, sizeof ( uri_string ),
-                  "tftp://%s:%d%s%s?blksize=%zd",
+                  "tftp%s://%s:%d%s%s?blksize=%zd",
+                  sizeonly ? "size" : "",
                   inet_ntoa ( address ), ntohs ( port ),
                   ( ( filename[0] == '/' ) ? "" : "/" ), filename, blksize );
        DBG ( " %s", uri_string );
@@ -254,7 +256,8 @@ PXENV_EXIT_t pxenv_tftp_open ( struct s_PXENV_TFTP_OPEN *tftp_open ) {
        if ( ( rc = pxe_tftp_open ( tftp_open->ServerIPAddress,
                                    tftp_open->TFTPPort,
                                    tftp_open->FileName,
-                                   tftp_open->PacketSize ) ) != 0 ) {
+                                   tftp_open->PacketSize,
+                                   0) ) != 0 ) {
                tftp_open->Status = PXENV_STATUS ( rc );
                return PXENV_EXIT_FAILURE;
        }
@@ -488,7 +491,7 @@ PXENV_EXIT_t pxenv_tftp_read_file ( struct s_PXENV_TFTP_READ_FILE
 
        /* Open TFTP file */
        if ( ( rc = pxe_tftp_open ( tftp_read_file->ServerIPAddress, 0,
-                                   tftp_read_file->FileName, 0 ) ) != 0 ) {
+                                   tftp_read_file->FileName, 0, 0 ) ) != 0 ) {
                tftp_read_file->Status = PXENV_STATUS ( rc );
                return PXENV_EXIT_FAILURE;
        }
@@ -558,7 +561,7 @@ PXENV_EXIT_t pxenv_tftp_get_fsize ( struct s_PXENV_TFTP_GET_FSIZE
 
        /* Open TFTP file */
        if ( ( rc = pxe_tftp_open ( tftp_get_fsize->ServerIPAddress, 0,
-                                   tftp_get_fsize->FileName, 0 ) ) != 0 ) {
+                                   tftp_get_fsize->FileName, 0, 1 ) ) != 0 ) {
                tftp_get_fsize->Status = PXENV_STATUS ( rc );
                return PXENV_EXIT_FAILURE;
        }