[http] Add missing check for memory allocation failure

Signed-off-by: Michael Brown <mcb30@ipxe.org>

diff --git a/src/net/tcp/httpconn.c b/src/net/tcp/httpconn.c
index a2c01a418a72181e3811fcb4ff9aac59c10c6caa..2cfca9c94b7de4a00414a1e7416c143284056636 100644 (file)
--- a/src/net/tcp/httpconn.c
+++ b/src/net/tcp/httpconn.c
@@ -277,6 +277,10 @@ int http_connect ( struct interface *xfer, struct uri *uri ) {
 
        /* Allocate and initialise structure */
        conn = zalloc ( sizeof ( *conn ) );
+       if ( ! conn ) {
+               rc = -ENOMEM;
+               goto err_alloc;
+       }
        ref_init ( &conn->refcnt, http_conn_free );
        conn->uri = uri_get ( uri );
        conn->scheme = scheme;
@@ -310,5 +314,6 @@ int http_connect ( struct interface *xfer, struct uri *uri ) {
                conn->scheme->name, conn->uri->host, port, strerror ( rc ) );
        http_conn_close ( conn, rc );
        ref_put ( &conn->refcnt );
+ err_alloc:
        return rc;
 }