feat: Enforce EDV document ID format in EDVFormatProvider

Signed-off-by: Derek Trider <Derek.Trider@securekey.com>
hyperledger-archives · Oct 24, 2020 · e19bd2e · e19bd2e
1 parent 5e3788f
commit e19bd2e
Show file tree

Hide file tree

Showing 26 changed files with 395 additions and 201 deletions.
diff --git a/component/storage/jsindexeddb/jsindexeddb.go b/component/storage/jsindexeddb/jsindexeddb.go
@@ -160,7 +160,7 @@ func (s *store) Get(k string) ([]byte, error) {
 	return []byte(data.Get("value").String()), nil
 }
 
-// Iterator returns iterator for the latest snapshot of the underlying db.
+// Range returns iterator for the latest snapshot of the underlying db.
 func (s *store) Iterator(start, limit string) storage.StoreIterator {
 	if limit == "" {
 		return newIterator(nil, false, fmt.Errorf("limit key is mandatory"))
@@ -197,7 +197,7 @@ func (s *store) Delete(k string) error {
 }
 
 // TODO #2229 - implement query method.
-func (s *store) Query(query string) (storage.StoreIterator, error) {
+func (s *store) Query(name, value string) (storage.StoreIterator, error) {
 	return nil, storage.ErrQueryingNotSupported
 }
 

diff --git a/component/storage/leveldb/leveldb_store.go b/component/storage/leveldb/leveldb_store.go
@@ -112,7 +112,7 @@ type leveldbStore struct {
 }
 
 // TODO #2227 - implement query method.
-func (s *leveldbStore) Query(_ string) (storage.StoreIterator, error) {
+func (s *leveldbStore) Query(name, value string) (storage.StoreIterator, error) {
 	return nil, storage.ErrQueryingNotSupported
 }
 
@@ -144,7 +144,7 @@ func (s *leveldbStore) Get(k string) ([]byte, error) {
 }
 
 // Iterator returns iterator for the latest snapshot of the underlying db.
-func (s *leveldbStore) Iterator(start, limit string) storage.StoreIterator {
+func (s *leveldbStore) Range(start, limit string) storage.StoreIterator {
 	return s.db.NewIterator(&util.Range{
 		Start: []byte(start),
 		Limit: []byte(strings.ReplaceAll(limit, storage.EndKeySuffix, "~")),

diff --git a/pkg/didcomm/protocol/didexchange/service_test.go b/pkg/didcomm/protocol/didexchange/service_test.go
@@ -716,7 +716,7 @@ type mockStore struct {
 	put    func(string, []byte) error
 	get    func(string) ([]byte, error)
 	delete func(string) error
-	query  func(string) (storage.StoreIterator, error)
+	query  func(string, string) (storage.StoreIterator, error)
 }
 
 // Put stores the key and the record.
@@ -734,12 +734,12 @@ func (m *mockStore) Delete(k string) error {
 	return m.delete(k)
 }
 
-func (m *mockStore) Query(query string) (storage.StoreIterator, error) {
-	return m.query(query)
+func (m *mockStore) Query(name, value string) (storage.StoreIterator, error) {
+	return m.query(name, value)
 }
 
 // Search returns storage iterator.
-func (m *mockStore) Iterator(start, limit string) storage.StoreIterator {
+func (m *mockStore) Range(start, limit string) storage.StoreIterator {
 	return nil
 }
 

diff --git a/pkg/didcomm/protocol/introduce/service.go b/pkg/didcomm/protocol/introduce/service.go
@@ -479,7 +479,7 @@ func (s *Service) currentStateName(piID string) (string, error) {
 
 // Actions returns actions for the async usage.
 func (s *Service) Actions() ([]Action, error) {
-	records := s.store.Iterator(
+	records := s.store.Range(
 		fmt.Sprintf(transitionalPayloadKey, ""),
 		fmt.Sprintf(transitionalPayloadKey, storage.EndKeySuffix),
 	)
@@ -708,7 +708,7 @@ func (s *Service) saveParticipant(piID string, p *participant) error {
 }
 
 func (s *Service) getParticipants(piID string) ([]*participant, error) {
-	records := s.store.Iterator(fmt.Sprintf(participantsKey, piID, ""),
+	records := s.store.Range(fmt.Sprintf(participantsKey, piID, ""),
 		fmt.Sprintf(participantsKey, piID, storage.EndKeySuffix))
 	defer records.Release()
 

diff --git a/pkg/didcomm/protocol/issuecredential/service.go b/pkg/didcomm/protocol/issuecredential/service.go
@@ -565,7 +565,7 @@ func (s *Service) ActionStop(piID string, cErr error) error {
 
 // Actions returns actions for the async usage.
 func (s *Service) Actions() ([]Action, error) {
-	records := s.store.Iterator(
+	records := s.store.Range(
 		fmt.Sprintf(transitionalPayloadKey, ""),
 		fmt.Sprintf(transitionalPayloadKey, storage.EndKeySuffix),
 	)

diff --git a/pkg/didcomm/protocol/mediator/service.go b/pkg/didcomm/protocol/mediator/service.go
@@ -607,7 +607,7 @@ func (s *Service) Unregister(connID string) error {
 
 // GetConnections returns the connections of the router.
 func (s *Service) GetConnections() ([]string, error) {
-	records := s.routeStore.Iterator(
+	records := s.routeStore.Range(
 		fmt.Sprintf(routeConnIDDataKey, ""),
 		fmt.Sprintf(routeConnIDDataKey, storage.EndKeySuffix),
 	)

diff --git a/pkg/didcomm/protocol/outofband/service.go b/pkg/didcomm/protocol/outofband/service.go
@@ -250,7 +250,7 @@ func (s *Service) HandleInbound(msg service.DIDCommMsg, myDID, theirDID string)
 
 // Actions returns actions for the async usage.
 func (s *Service) Actions() ([]Action, error) {
-	records := s.store.Iterator(
+	records := s.store.Range(
 		fmt.Sprintf(transitionalPayloadKey, ""),
 		fmt.Sprintf(transitionalPayloadKey, storage.EndKeySuffix),
 	)

diff --git a/pkg/didcomm/protocol/outofband/service_test.go b/pkg/didcomm/protocol/outofband/service_test.go
@@ -1265,15 +1265,15 @@ func (s *stubStore) Get(k string) ([]byte, error) {
 	panic("implement me")
 }
 
-func (s *stubStore) Iterator(start, limit string) storage.StoreIterator {
+func (s *stubStore) Range(start, limit string) storage.StoreIterator {
 	panic("implement me")
 }
 
 func (s *stubStore) Delete(k string) error {
 	panic("implement me")
 }
 
-func (s *stubStore) Query(query string) (storage.StoreIterator, error) {
+func (s *stubStore) Query(name, value string) (storage.StoreIterator, error) {
 	panic("implement me")
 }
 

diff --git a/pkg/didcomm/protocol/presentproof/service.go b/pkg/didcomm/protocol/presentproof/service.go
@@ -484,7 +484,7 @@ func (s *Service) deleteTransitionalPayload(id string) error {
 
 // Actions returns actions for the async usage.
 func (s *Service) Actions() ([]Action, error) {
-	records := s.store.Iterator(
+	records := s.store.Range(
 		fmt.Sprintf(transitionalPayloadKey, ""),
 		fmt.Sprintf(transitionalPayloadKey, storage.EndKeySuffix),
 	)

diff --git a/pkg/internal/gomocks/storage/mocks.go b/pkg/internal/gomocks/storage/mocks.go
diff --git a/pkg/mock/storage/mock_store.go b/pkg/mock/storage/mock_store.go
@@ -106,8 +106,8 @@ func (s *MockStore) Get(k string) ([]byte, error) {
 	return val, s.ErrGet
 }
 
-// Iterator returns an iterator for the underlying mockstore.
-func (s *MockStore) Iterator(start, limit string) storage.StoreIterator {
+// Range returns an iterator for the underlying mockstore.
+func (s *MockStore) Range(start, limit string) storage.StoreIterator {
 	if s.ErrItr != nil {
 		return NewMockIteratorWithError(s.ErrItr)
 	}
@@ -136,7 +136,7 @@ func (s *MockStore) Delete(k string) error {
 }
 
 // Query returns a mocked store iterator and error value.
-func (s *MockStore) Query(_ string) (storage.StoreIterator, error) {
+func (s *MockStore) Query(indexKey, indexValue string) (storage.StoreIterator, error) {
 	return s.QueryReturnValue, s.ErrQuery
 }
 

diff --git a/pkg/storage/edv/documentprocessor/documentprocessor.go b/pkg/storage/edv/documentprocessor/documentprocessor.go
@@ -43,13 +43,14 @@ func New(jweEncrypter jose.Encrypter, jweDecrypter jose.Decrypter) *DocumentProc
 }
 
 // Encrypt creates a new encrypted document based off of the given structured document.
-func (a *DocumentProcessor) Encrypt(structuredDocument *edv.StructuredDocument) (*edv.EncryptedDocument, error) {
-	structuredDocumentBytes, err := a.marshal(structuredDocument)
+func (d *DocumentProcessor) Encrypt(structuredDocument *edv.StructuredDocument,
+	indexedAttributes []edv.IndexedAttributeCollection) (*edv.EncryptedDocument, error) {
+	structuredDocumentBytes, err := d.marshal(structuredDocument)
 	if err != nil {
 		return nil, fmt.Errorf(failMarshalStructuredDocument, err)
 	}
 
-	jwe, err := a.jweEncrypter.Encrypt(structuredDocumentBytes)
+	jwe, err := d.jweEncrypter.Encrypt(structuredDocumentBytes)
 	if err != nil {
 		return nil, fmt.Errorf(failEncryptStructuredDocument, err)
 	}
@@ -60,21 +61,22 @@ func (a *DocumentProcessor) Encrypt(structuredDocument *edv.StructuredDocument)
 	}
 
 	encryptedDoc := edv.EncryptedDocument{
-		ID:  structuredDocument.ID,
-		JWE: []byte(serializedJWE),
+		ID:                          structuredDocument.ID,
+		IndexedAttributeCollections: indexedAttributes,
+		JWE:                         []byte(serializedJWE),
 	}
 
 	return &encryptedDoc, nil
 }
 
 // Decrypt decrypts the encrypted document into a structured document.
-func (a *DocumentProcessor) Decrypt(encryptedDocument *edv.EncryptedDocument) (*edv.StructuredDocument, error) {
+func (d *DocumentProcessor) Decrypt(encryptedDocument *edv.EncryptedDocument) (*edv.StructuredDocument, error) {
 	encryptedJWE, err := jose.Deserialize(string(encryptedDocument.JWE))
 	if err != nil {
 		return nil, fmt.Errorf(failDeserializeJWE, err)
 	}
 
-	structuredDocumentBytes, err := a.jweDecrypter.Decrypt(encryptedJWE)
+	structuredDocumentBytes, err := d.jweDecrypter.Decrypt(encryptedJWE)
 	if err != nil {
 		return nil, fmt.Errorf(failDecryptJWE, err)
 	}

diff --git a/pkg/storage/edv/documentprocessor/documentprocessor_test.go b/pkg/storage/edv/documentprocessor/documentprocessor_test.go
@@ -43,15 +43,15 @@ func TestAriesDocumentProcessor_Encrypt(t *testing.T) {
 		documentProcessor := DocumentProcessor{marshal: failingMarshal}
 		require.NotNil(t, documentProcessor)
 
-		encryptedDocument, err := documentProcessor.Encrypt(nil)
+		encryptedDocument, err := documentProcessor.Encrypt(nil, nil)
 		require.EqualError(t, err, fmt.Errorf(failMarshalStructuredDocument, errFailingMarshal).Error())
 		require.Nil(t, encryptedDocument)
 	})
 	t.Run("Fail to encrypt structured document", func(t *testing.T) {
 		documentProcessor := New(&failingEncrypter{}, nil)
 		require.NotNil(t, documentProcessor)
 
-		encryptedDocument, err := documentProcessor.Encrypt(createStructuredDocument())
+		encryptedDocument, err := documentProcessor.Encrypt(createStructuredDocument(), nil)
 		require.EqualError(t, err, fmt.Errorf(failEncryptStructuredDocument, errFailingEncrypter).Error())
 		require.Nil(t, encryptedDocument)
 	})
@@ -145,7 +145,7 @@ func createStructuredDocument() *edv.StructuredDocument {
 func createEncryptedDocument(t *testing.T, documentProcessor *DocumentProcessor) *edv.EncryptedDocument {
 	structuredDocument := createStructuredDocument()
 
-	encryptedDocument, err := documentProcessor.Encrypt(structuredDocument)
+	encryptedDocument, err := documentProcessor.Encrypt(structuredDocument, nil)
 	require.NoError(t, err)
 	require.NotNil(t, encryptedDocument)