From 617b10e25507b9a164b9af5436231d6f71e75133 Mon Sep 17 00:00:00 2001
From: cilame <opaquism@hotmail.com>
Date: Wed, 18 May 2022 19:05:55 +0800
Subject: [PATCH] add

---
 tools/model_funcs.js | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/tools/model_funcs.js b/tools/model_funcs.js
index 1f24a2e..78e9d45 100644
--- a/tools/model_funcs.js
+++ b/tools/model_funcs.js
@@ -160,13 +160,19 @@ function handle_proxy(req, res) {
       var message = v.message
       if(message){
         var purl = new URL(message.href).origin
-        req.headers.cookie = message.cookie || req.headers.cookie
-        req.headers['user-agent'] = message.userAgent || req.headers['user-agent']
+        if (message.headers){
+          if (message.headers['Cookie'] || req.headers.cookie){
+            req.headers.cookie = message.headers['Cookie'] || req.headers.cookie
+          }
+          if (message.headers['User-Agent'] || req.headers['user-agent']){
+            req.headers['user-agent'] = message.headers['User-Agent'] || req.headers['user-agent']
+          }
+        }
         proxy.web(req, res, { target: purl }, function(e){
           console.log(e)
         });
       }else{
-      	res.writeHead(233, {'Content-type': 'text/html'})
+        res.writeHead(233, {'Content-type': 'text/html'})
         res.write('<h1>no message obj.<h1>')
         res.end()
       }