smb2: share: export application key

Author: oiweiwei

client.go

@@ -227,6 +227,12 @@ type Share struct {
 	ctx context.Context
 }
 
+func (fs *Share) ApplicationKey() []byte {
+	key := make([]byte, len(fs.applicationKey))
+	copy(key, fs.applicationKey)
+	return key
+}
+
 func (fs *Share) WithContext(ctx context.Context) *Share {
 	if ctx == nil {
 		panic("nil context")
@@ -667,13 +673,12 @@ func (fs *Share) Stat(name string) (os.FileInfo, error) {
 		return nil, &os.PathError{Op: "stat", Path: name, Err: err}
 	}
 
-	fi, err := f.fileStat, nil
-	if e := f.close(); err == nil {
-		err = e
-	}
-	if err != nil {
+	fi := f.fileStat
+
+	if err := f.close(); err != nil {
 		return nil, &os.PathError{Op: "stat", Path: name, Err: err}
 	}
+
 	return fi, nil
 }
 
@@ -1881,6 +1886,35 @@ func (f *File) WriteString(s string) (n int, err error) {
 	return f.Write([]byte(s))
 }
 
+type encoderBytes []byte
+
+func (e encoderBytes) Size() int {
+	return len(([]byte)(e))
+}
+
+func (e encoderBytes) Encode(buf []byte) {
+	copy(buf, ([]byte)(e))
+}
+
+var _ Encoder = encoderBytes(nil)
+
+// Transcieve function is used to send a request and receive a response in a single call.
+func (f *File) Transcieve(p []byte, out []byte) (int, error) {
+	output, err := f.ioctl(&IoctlRequest{
+		CtlCode:           FSCTL_PIPE_TRANSCEIVE,
+		OutputOffset:      0,
+		OutputCount:       0,
+		MaxInputResponse:  0,
+		MaxOutputResponse: uint32(len(out)),
+		Flags:             SMB2_0_IOCTL_IS_FSCTL,
+		Input:             (encoderBytes)(p),
+	})
+	if err != nil {
+		return 0, err
+	}
+	return copy(out, output), nil
+}
+
 func (f *File) encodeSize(e Encoder) int {
 	if e == nil {
 		return 0

session.go

@@ -145,7 +145,7 @@ func sessionSetup(conn *conn, i Initiator, ctx context.Context) (*session, error
 			s.signer = cmac.New(ciph)
 			s.verifier = cmac.New(ciph)
 
-			// s.applicationKey = kdf(sessionKey, []byte("SMB2APP\x00"), []byte("SmbRpc\x00"))
+			s.applicationKey = kdf(sessionKey, []byte("SMB2APP\x00"), []byte("SmbRpc\x00"))
 
 			encryptionKey := kdf(sessionKey, []byte("SMB2AESCCM\x00"), []byte("ServerIn \x00"))
 			decryptionKey := kdf(sessionKey, []byte("SMB2AESCCM\x00"), []byte("ServerOut\x00"))
@@ -184,7 +184,7 @@ func sessionSetup(conn *conn, i Initiator, ctx context.Context) (*session, error
 			s.signer = cmac.New(ciph)
 			s.verifier = cmac.New(ciph)
 
-			// s.applicationKey = kdf(sessionKey, []byte("SMBAppKey\x00"), preauthIntegrityHashValue)
+			s.applicationKey = kdf(sessionKey, []byte("SMBAppKey\x00"), s.preauthIntegrityHashValue[:])
 
 			encryptionKey := kdf(sessionKey, []byte("SMBC2SCipherKey\x00"), s.preauthIntegrityHashValue[:])
 			decryptionKey := kdf(sessionKey, []byte("SMBS2CCipherKey\x00"), s.preauthIntegrityHashValue[:])
@@ -275,7 +275,7 @@ type session struct {
 	encrypter cipher.AEAD
 	decrypter cipher.AEAD
 
-	// applicationKey []byte
+	applicationKey []byte
 }
 
 func (s *session) logoff(ctx context.Context) error {