Adding WrapKey to p11/session

p11/session.go

@@ -40,6 +40,8 @@ type Session interface {
 	GenerateKeyPair(request GenerateKeyPairRequest) (*KeyPair, error)
 	// GenerateRandom returns random bytes generated by the token.
 	GenerateRandom(length int) ([]byte, error)
+	// Wrap key returns ciphertext bytes from wrapping a key with another key
+	WrapKey(request WrapKeyRequest) ([]byte, error)
 
 	// InitPIN initialize's the normal user's PIN.
 	InitPIN(pin string) error
@@ -191,3 +193,20 @@ func (s *sessionImpl) GenerateKeyPair(request GenerateKeyPairRequest) (*KeyPair,
 		}),
 	}, nil
 }
+
+// WrapKeyRequest contains the fields used to wrap a key.
+type WrapKeyRequest struct {
+	Mechanism      pkcs11.Mechanism
+	WrappingObject Object
+	Object         Object
+}
+
+func (s *sessionImpl) WrapKey(request WrapKeyRequest) ([]byte, error) {
+	s.Lock()
+	defer s.Unlock()
+	return s.ctx.WrapKey(s.handle,
+		[]*pkcs11.Mechanism{&request.Mechanism},
+		request.WrappingObject.objectHandle,
+		request.Object.objectHandle)
+
+}