fixes

open-amt-cloud-toolkit · Jun 14, 2024 · e02d81d · e02d81d
1 parent 475254a
commit e02d81d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 10 deletions.
diff --git a/pkg/apf/processor.go b/pkg/apf/processor.go
@@ -155,7 +155,7 @@ func ProcessGlobalRequest(data []byte) interface{} {
 			if tcpForwardRequest.Port == 16992 || tcpForwardRequest.Port == 16993 {
 				reply = TcpForwardReplySuccess(tcpForwardRequest.Port)
 			} else {
-				reply = int16(APF_REQUEST_FAILURE)
+				reply = APF_REQUEST_FAILURE
 			}
 		} else if genericHeader.String == APF_GLOBAL_REQUEST_STR_TCP_FORWARD_CANCEL_REQUEST {
 			reply = APF_REQUEST_SUCCESS

diff --git a/pkg/wsman/client/hash.go b/pkg/wsman/client/hash.go
@@ -84,7 +84,7 @@ func (c *AuthChallenge) response(method, uri, cnonce string) (string, error) {
 					return "", fmt.Errorf("%w: %w", errRandRead, err)
 				}
 
-				c.CNonce = fmt.Sprintf("%x", b)[:16]
+				c.CNonce = fmt.Sprintf("%x", b)[:6]
 			}
 
 			c.Qop = "auth"
@@ -125,11 +125,11 @@ func (c *AuthChallenge) authorize(method, uri string) (string, error) {
 	sb.WriteString(response)
 	sb.WriteString(`"`)
 
-	if c.Algorithm != "" {
-		sb.WriteString(`, algorithm="`)
-		sb.WriteString(c.Algorithm)
-		sb.WriteString(`"`)
-	}
+	// if c.Algorithm != "" {
+	// 	sb.WriteString(`, algorithm="`)
+	// 	sb.WriteString(c.Algorithm)
+	// 	sb.WriteString(`"`)
+	// }
 
 	if c.Opaque != "" {
 		sb.WriteString(`, opaque="`)

diff --git a/pkg/wsman/client/wsman.go b/pkg/wsman/client/wsman.go
@@ -105,7 +105,7 @@ func (t *Target) Post(msg string) (response []byte, err error) {
 	if err != nil {
 		return nil, err
 	}
-
+	req.Header.Set("Host", "4c4c4544-004d-4d10-8050-b2c04f325133:16992")
 	if t.username != "" && t.password != "" {
 		if t.useDigest {
 			auth, err = t.challenge.authorize("POST", "/wsman")
@@ -121,7 +121,7 @@ func (t *Target) Post(msg string) (response []byte, err error) {
 		}
 	}
 
-	req.Header.Add("content-type", ContentType)
+	// req.Header.Add("content-type", ContentType)
 
 	if t.logAMTMessages {
 		logrus.Trace(msg)
@@ -150,7 +150,8 @@ func (t *Target) Post(msg string) (response []byte, err error) {
 		}
 
 		req.Header.Set("Authorization", auth)
-		req.Header.Add("content-type", ContentType)
+		// req.Header.Add("content-type", ContentType)
+		req.Header.Set("Host", "4c4c4544-004d-4d10-8050-b2c04f325133:16992")
 
 		res, err = t.Do(req)
 		if err != nil && err.Error() != io.EOF.Error() {