Better message handling when buffer runs out of memory.

1 files changed, 4 insertions, 4 deletions

diff --git a/daemon/ntlm.c b/daemon/ntlm.c
index ab6e5e4..8e1aa20 100644
--- a/daemon/ntlm.c
+++ b/daemon/ntlm.c
@@ -406,8 +406,8 @@ int ntlm_auth_ntlm(ha_request_t* rq, ntlm_context_t* ctx, void* key,
                 ha_bufenc64(rq->buf, (unsigned char*)&msg, sizeof(msg));
             }
 
-            if(ha_buferr(rq->buf))
-                RETURN(HA_FALSE);
+            if(CHECK_RBUF(rq))
+                RETURN(HA_CRITERROR);
 
             /*
              * TODO: Our callers need to be able to keep alive
@@ -513,7 +513,7 @@ int ntlm_auth_ntlm(ha_request_t* rq, ntlm_context_t* ctx, void* key,
 
 
 finally:
-    if(ha_buferr(rq->buf))
+    if(CHECK_RBUF(rq))
         ret = HA_CRITERROR;
 
     if(conn)
@@ -759,7 +759,7 @@ int ntlm_process(ha_request_t* rq)
         {
             ha_bufmcat(rq->buf, HA_PREFIX_BASIC, "realm=\"", rq->context->realm, "\"", NULL);
 
-            if(ha_buferr(rq->buf))
+            if(CHECK_RBUF(rq))
                 return HA_CRITERROR;
 
             ha_addheader(rq, "WWW-Authenticate", ha_bufdata(rq->buf));