From 496e0af86ef9f95bb55ca45f79ee8d04cde1aad0 Mon Sep 17 00:00:00 2001 From: Andy Green Date: Wed, 20 Jan 2016 17:01:47 +0800 Subject: [PATCH] more return code checking Signed-off-by: Andy Green --- lib/libwebsockets.c | 3 ++- lib/parsers.c | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/lib/libwebsockets.c b/lib/libwebsockets.c index fce84139..cd2c06ff 100644 --- a/lib/libwebsockets.c +++ b/lib/libwebsockets.c @@ -388,7 +388,8 @@ libwebsockets_get_addresses(struct libwebsocket_context *context, if (addr4.sin_family == AF_UNSPEC) return -1; - lws_plat_inet_ntop(AF_INET, &addr4.sin_addr, rip, rip_len); + if (lws_plat_inet_ntop(AF_INET, &addr4.sin_addr, rip, rip_len) == NULL) + return -1; return 0; } diff --git a/lib/parsers.c b/lib/parsers.c index fca08514..c50adfad 100644 --- a/lib/parsers.c +++ b/lib/parsers.c @@ -281,7 +281,8 @@ int libwebsocket_parse( case URIES_SEEN_PERCENT_H1: if (char_to_hex(c) < 0) { /* regurgitate */ - issue_char(wsi, '%'); + if (issue_char(wsi, '%') < 0) + return -1; wsi->u.hdr.ues = URIES_IDLE; /* regurgitate + assess */ if (libwebsocket_parse(context, wsi, wsi->u.hdr.esc_stash) < 0)