From ae401f7460664256de588d9a0bfee4c3b1560738 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Wed, 6 Feb 2013 23:32:31 +0100 Subject: Fix crashes when creating the request We now obtain the peer address before creating the Req object. If an error occurs, then something went wrong, we close the connection nicely directly. --- src/cowboy_req.erl | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'src/cowboy_req.erl') diff --git a/src/cowboy_req.erl b/src/cowboy_req.erl index eb3a939..c807a75 100644 --- a/src/cowboy_req.erl +++ b/src/cowboy_req.erl @@ -42,7 +42,7 @@ -module(cowboy_req). %% Request API. --export([new/14]). +-export([new/15]). -export([method/1]). -export([version/1]). -export([peer/1]). @@ -179,15 +179,16 @@ %% %% Since we always need to parse the Connection header, we do it %% in an optimized way and add the parsed value to p_headers' cache. --spec new(inet:socket(), module(), binary(), binary(), binary(), binary(), +-spec new(inet:socket(), module(), + undefined | {inet:ip_address(), inet:port_number()}, + binary(), binary(), binary(), binary(), cowboy_http:version(), cowboy_http:headers(), binary(), inet:port_number() | undefined, binary(), boolean(), boolean(), undefined | cowboy_protocol:onresponse_fun()) -> req(). -new(Socket, Transport, Method, Path, Query, Fragment, +new(Socket, Transport, Peer, Method, Path, Query, Fragment, Version, Headers, Host, Port, Buffer, CanKeepalive, Compress, OnResponse) -> - {ok, Peer} = Transport:peername(Socket), Req = #http_req{socket=Socket, transport=Transport, pid=self(), peer=Peer, method=Method, path=Path, qs=Query, fragment=Fragment, version=Version, headers=Headers, host=Host, port=Port, buffer=Buffer, @@ -219,7 +220,8 @@ version(Req) -> %% @doc Return the peer address and port number of the remote host. -spec peer(Req) - -> {{inet:ip_address(), inet:port_number()}, Req} when Req::req(). + -> {undefined | {inet:ip_address(), inet:port_number()}, Req} + when Req::req(). peer(Req) -> {Req#http_req.peer, Req}. -- cgit v1.2.3