fix: fix when user-agent at bottom of header

Signed-off-by: Zxilly <[email protected]>
Zxilly · May 6, 2021 · 466d5b1 · 466d5b1
1 parent e8a87c2
commit 466d5b1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/Makefile b/Makefile
@@ -1,9 +1,9 @@
 include $(TOPDIR)/rules.mk
 
 PKG_NAME:=UA2F
-PKG_VERSION:=3.5
+PKG_VERSION:=3.6
 
-PKG_RELEASE:=8
+PKG_RELEASE:=9
 
 
 PKG_BUILD_DIR:=$(BUILD_DIR)/$(PKG_NAME)-$(PKG_VERSION)

diff --git a/src/ua2f.c b/src/ua2f.c
@@ -262,12 +262,12 @@ static int queue_cb(const struct nlmsghdr *nlh, void *data) {
     if (tcppkpayload) {
         char *uapointer = memmem(tcppkpayload, tcppklen, "\r\nUser", 6);
         if (uapointer) {
-            uapointer = memmem(tcppkpayload, tcppklen, "\r\nUser-Agent: ", 14);
+            uapointer = memmem(tcppkpayload, tcppklen, "\r\nUser-Agent:", 13);
             if (!uapointer){
-                uapointer = memmem(tcppkpayload, tcppklen, "\r\nUser-agent: ", 14);
+                uapointer = memmem(tcppkpayload, tcppklen, "\r\nUser-agent:", 13);
             }
         } else {
-            uapointer = memmem(tcppkpayload, tcppklen, "\r\nuser-agent: ", 14);
+            uapointer = memmem(tcppkpayload, tcppklen, "\r\nuser-agent:", 13);
         }
 
         if (uapointer) {
@@ -276,7 +276,7 @@ static int queue_cb(const struct nlmsghdr *nlh, void *data) {
             debugflag++; //flag6
 
             uaoffset = uapointer - tcppkpayload + 14;
-            for (int i = 0; i < tcppklen - uaoffset - 4; ++i) {
+            for (int i = 0; i < tcppklen - uaoffset - 2; ++i) {
                 if (*(uapointer + 14 + i) == '\r') {
                     ualength = i;
                     break;