Procházet zdrojové kódy

[tftp] Prevent potential division by zero

Signed-off-by: Michael Brown <mcb30@ipxe.org>
tags/v1.20.1
Michael Brown před 6 roky
rodič
revize
b11ae1d91b
1 změnil soubory, kde provedl 2 přidání a 0 odebrání
  1. 2
    0
      src/net/udp/tftp.c

+ 2
- 0
src/net/udp/tftp.c Zobrazit soubor

@@ -279,6 +279,8 @@ static int tftp_presize ( struct tftp_request *tftp, size_t filesize ) {
279 279
 	 * length is an exact multiple of the blocksize will have a
280 280
 	 * trailing zero-length block, which must be included.
281 281
 	 */
282
+	if ( tftp->blksize == 0 )
283
+		return -EINVAL;
282 284
 	num_blocks = ( ( filesize / tftp->blksize ) + 1 );
283 285
 	if ( ( rc = bitmap_resize ( &tftp->bitmap, num_blocks ) ) != 0 ) {
284 286
 		DBGC ( tftp, "TFTP %p could not resize bitmap to %d blocks: "

Načítá se…
Zrušit
Uložit