Skip to content

Commit c691511

Browse files
committed
updates
1 parent 6e605cb commit c691511

File tree

4 files changed

+17
-8
lines changed

4 files changed

+17
-8
lines changed

engine/api/graphql/server/schema.resolvers.go

Lines changed: 2 additions & 5 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

engine/plugins/enrich/tls_cert.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import (
1414
"strings"
1515
"time"
1616

17+
"github.com/google/uuid"
1718
"github.com/owasp-amass/amass/v4/config"
1819
"github.com/owasp-amass/amass/v4/engine/plugins/support"
1920
et "github.com/owasp-amass/amass/v4/engine/types"
@@ -371,7 +372,10 @@ func (te *tlsexpand) storeContact(e *et.Event, c *tlsContact, asset *dbt.Entity,
371372
}
372373
}
373374
if len(ct.Organization) > 0 && ct.Organization[0] != "" && m.IsMatch(string(oam.Organization)) {
374-
if a, err := e.Session.Cache().CreateAsset(&org.Organization{Name: ct.Organization[0]}); err == nil && a != nil {
375+
if a, err := e.Session.Cache().CreateAsset(&org.Organization{
376+
ID: uuid.New().String(),
377+
Name: ct.Organization[0],
378+
}); err == nil && a != nil {
375379
if edge, err := e.Session.Cache().CreateEdge(&dbt.Edge{
376380
Relation: &general.SimpleRelation{Name: "organization"},
377381
FromEntity: cr,

engine/plugins/rdap/plugin.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import (
1414
"strings"
1515

1616
"github.com/caffix/stringset"
17+
"github.com/google/uuid"
1718
"github.com/openrdap/rdap"
1819
"github.com/openrdap/rdap/bootstrap"
1920
"github.com/openrdap/rdap/bootstrap/cache"
@@ -243,7 +244,10 @@ func (rd *rdapPlugin) storeEntity(e *et.Event, level int, entity *rdap.Entity, a
243244
}
244245
}
245246
} else if name != "" && m.IsMatch(string(oam.Organization)) {
246-
if a, err := e.Session.Cache().CreateAsset(&org.Organization{Name: name}); err == nil && a != nil {
247+
if a, err := e.Session.Cache().CreateAsset(&org.Organization{
248+
ID: uuid.New().String(),
249+
Name: name,
250+
}); err == nil && a != nil {
247251
_ = rd.createContactEdge(e.Session, cr, a, &general.SimpleRelation{Name: "organization"}, src)
248252
}
249253
}

engine/plugins/whois/domain_record.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ import (
1010
"strings"
1111
"time"
1212

13+
"github.com/google/uuid"
1314
whoisparser "github.com/likexian/whois-parser"
1415
"github.com/owasp-amass/amass/v4/config"
1516
"github.com/owasp-amass/amass/v4/engine/plugins/support"
@@ -224,7 +225,10 @@ func (r *domrec) storeContact(e *et.Event, c *domrecContact, dr *dbt.Entity, m *
224225
}
225226
}
226227
if wc.Organization != "" && m.IsMatch(string(oam.Organization)) {
227-
if a, err := e.Session.Cache().CreateAsset(&org.Organization{Name: wc.Organization}); err == nil && a != nil {
228+
if a, err := e.Session.Cache().CreateAsset(&org.Organization{
229+
ID: uuid.New().String(),
230+
Name: wc.Organization,
231+
}); err == nil && a != nil {
228232
r.createSimpleEdge(e.Session.Cache(), &general.SimpleRelation{Name: "organization"}, cr, a)
229233
}
230234
}

0 commit comments

Comments
 (0)