fixing connect bug of reqmod

egirna · Nov 1, 2022 · 17e7101 · 17e7101
1 parent b95ade0
commit 17e7101
Show file tree

Hide file tree

Showing 6 changed files with 38 additions and 1 deletion.
diff --git a/service/services/clamav/clamav.go b/service/services/clamav/clamav.go
@@ -37,6 +37,12 @@ func (c *Clamav) Processing(partial bool) (int, interface{}, map[string]string,
 			msgHeadersAfterProcessing, vendorMsgs
 	}
 
+	//if the http method is Connect, return the request as it is because it has no body
+	if c.httpMsg.Request.Method == http.MethodConnect {
+		return utils.OkStatusCodeStr, c.generalFunc.ReturningHttpMessageWithFile(c.methodName, file.Bytes()),
+			serviceHeaders, msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
+	}
+
 	//getting the extension of the file
 	var contentType []string
 	if len(contentType) == 0 {

diff --git a/service/services/cloudmersive/cloudmersive.go b/service/services/cloudmersive/cloudmersive.go
@@ -39,6 +39,12 @@ func (c *CloudMersive) Processing(partial bool) (int, interface{}, map[string]st
 			msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
 	}
 
+	//if the http method is Connect, return the request as it is because it has no body
+	if c.httpMsg.Request.Method == http.MethodConnect {
+		return utils.OkStatusCodeStr, c.generalFunc.ReturningHttpMessageWithFile(c.methodName, file.Bytes()),
+			serviceHeaders, msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
+	}
+
 	//getting the extension of the file
 	var contentType []string
 	if len(contentType) == 0 {

diff --git a/service/services/echo/echo.go b/service/services/echo/echo.go
@@ -24,7 +24,6 @@ func (e *Echo) Processing(partial bool) (int, interface{}, map[string]string, ma
 		logging.Logger.Info(e.serviceName + " service has stopped processing partially")
 		return utils.Continue, nil, nil, msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
 	}
-
 	isGzip := false
 
 	//extracting the file from http message
@@ -35,6 +34,13 @@ func (e *Echo) Processing(partial bool) (int, interface{}, map[string]string, ma
 		return utils.InternalServerErrStatusCodeStr, nil, serviceHeaders,
 			msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
 	}
+
+	//if the http method is Connect, return the request as it is because it has no body
+	if e.httpMsg.Request.Method == http.MethodConnect {
+		return utils.OkStatusCodeStr, e.generalFunc.ReturningHttpMessageWithFile(e.methodName, file.Bytes()),
+			serviceHeaders, msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
+	}
+
 	//getting the extension of the file
 	var contentType []string
 	if len(contentType) == 0 {

diff --git a/service/services/grayimages/grayimages.go b/service/services/grayimages/grayimages.go
@@ -36,6 +36,12 @@ func (g *Grayimages) Processing(partial bool) (int, interface{}, map[string]stri
 			msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
 	}
 
+	//if the http method is Connect, return the request as it is because it has no body
+	if g.httpMsg.Request.Method == http.MethodConnect {
+		return utils.OkStatusCodeStr, g.generalFunc.ReturningHttpMessageWithFile(g.methodName, file.Bytes()),
+			serviceHeaders, msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
+	}
+
 	//getting the extension of the file
 	var contentType []string
 	if len(contentType) == 0 {

diff --git a/service/services/hashlookup/hashlookup.go b/service/services/hashlookup/hashlookup.go
@@ -41,6 +41,13 @@ func (h *Hashlookup) Processing(partial bool) (int, interface{}, map[string]stri
 		return utils.InternalServerErrStatusCodeStr, nil, serviceHeaders,
 			msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
 	}
+
+	//if the http method is Connect, return the request as it is because it has no body
+	if h.httpMsg.Request.Method == http.MethodConnect {
+		return utils.OkStatusCodeStr, h.generalFunc.ReturningHttpMessageWithFile(h.methodName, file.Bytes()),
+			serviceHeaders, msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
+	}
+
 	//getting the extension of the file
 	var contentType []string
 	if len(contentType) == 0 {

diff --git a/service/services/virustotal/virustotal.go b/service/services/virustotal/virustotal.go
@@ -43,6 +43,12 @@ func (v *Virustotal) Processing(partial bool) (int, interface{}, map[string]stri
 			msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
 	}
 
+	//if the http method is Connect, return the request as it is because it has no body
+	if v.httpMsg.Request.Method == http.MethodConnect {
+		return utils.OkStatusCodeStr, v.generalFunc.ReturningHttpMessageWithFile(v.methodName, file.Bytes()),
+			serviceHeaders, msgHeadersBeforeProcessing, msgHeadersAfterProcessing, vendorMsgs
+	}
+
 	//getting the extension of the file
 	var contentType []string
 	if len(contentType) == 0 {