From 6f933fb41a20cad7ecbb3c1d327c454380e544ca Mon Sep 17 00:00:00 2001
From: Matthew Flatt <mflatt@racket-lang.org>
Date: Thu, 15 Apr 2004 16:29:05 +0000
Subject: [PATCH] .

original commit: 632c67cb6f80d468839ad49600a7f5fbcd30d6e3
---
 collects/net/head-sig.ss  |  1 +
 collects/net/head-unit.ss | 25 +++++++++++++++++++------
 2 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/collects/net/head-sig.ss b/collects/net/head-sig.ss
index 58d90cd..d32cae1 100644
--- a/collects/net/head-sig.ss
+++ b/collects/net/head-sig.ss
@@ -9,6 +9,7 @@
      extract-field
      remove-field
      insert-field
+     replace-field
      extract-all-fields
      append-headers
      standard-message-header
diff --git a/collects/net/head-unit.ss b/collects/net/head-unit.ss
index efecd46..48654bd 100644
--- a/collects/net/head-unit.ss
+++ b/collects/net/head-unit.ss
@@ -17,7 +17,7 @@
       (define re:continue (regexp "^[ \t\v]"))
       
       (define (validate-header s)
-	(let ([m (regexp-match #rx"[^\0-\277]" s)])
+	(let ([m (regexp-match #rx"[^\000-\377]" s)])
 	  (when m
 	    (error 'validate-header "non-Latin-1 character in string: ~a" (car m))))
 	(let ([len (string-length s)])
@@ -55,7 +55,7 @@
 				       s
 				       "")))))))
 
-      (define (remove-field field header)
+      (define (replace-field field data header)
 	(let ([m (regexp-match-positions 
 		  (make-field-start-regexp field)
 		  header)])
@@ -68,10 +68,23 @@
 				  (string-length header))])
 		(let ([m (regexp-match-positions #rx"[\r][\n][^: \r\n\"]*:" s)])
 		  (if m
-		      (string-append pre (substring s (+ 2 (caar m)) 
-						    (string-length s)))
-		      pre)))
-	      header)))
+		      (string-append pre 
+				     (let ([rest 
+					    (substring s (+ 2 (caar m)) 
+						       (string-length s))])
+				       (if data
+					   (insert-field field data rest)
+					   rest)))
+		      (if data
+			  (insert-field field data pre)
+			  pre))))
+	      (if data
+		  (insert-field field data header)
+		  header))))
+
+      (define (remove-field field header)
+	(replace-field field #f header))
+	
 
       (define (insert-field field data header)
 	(let ([field (format "~a: ~a\r\n"