From f17cf0ecd0363523dde781338fb1f3b5f5a7c6f7 Mon Sep 17 00:00:00 2001 From: Michael Brown Date: Tue, 21 Mar 2017 15:20:45 +0200 Subject: [PATCH] [http] Add missing check for memory allocation failure Signed-off-by: Michael Brown --- src/net/tcp/httpconn.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/net/tcp/httpconn.c b/src/net/tcp/httpconn.c index a2c01a41..2cfca9c9 100644 --- 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; }