Consistently handle nil in Copy implementations (#3534)

Check for `nil`-receiver in `Copy`.

The trigger for this change is a recently introduced small bug in `snet.Dial` where copying a `nil`-Path causes a panic.

go/lib/addr/appaddr.go

@@ -36,6 +36,9 @@ func AppAddrFromUDP(a *net.UDPAddr) *AppAddr {
 }
 
 func (a *AppAddr) Copy() *AppAddr {
+	if a == nil {
+		return nil
+	}
 	var l3 HostAddr
 	if a.L3 != nil {
 		l3 = a.L3.Copy()

go/lib/ctrl/ctrl.go

@@ -103,6 +103,9 @@ func (p *Pld) Len() int {
 }
 
 func (p *Pld) Copy() (*Pld, error) {
+	if p == nil {
+		return nil, nil
+	}
 	raw, err := proto.PackRoot(p)
 	if err != nil {
 		return nil, err

go/lib/ctrl/seg/seg.go

@@ -382,6 +382,9 @@ func (ps *PathSegment) validateIdx(idx int) error {
 
 // ShallowCopy creates a shallow copy of the path segment.
 func (ps *PathSegment) ShallowCopy() *PathSegment {
+	if ps == nil {
+		return nil
+	}
 	rawEntries := make([]*proto.SignedBlobS, len(ps.RawASEntries))
 	copy(rawEntries, ps.RawASEntries)
 	entries := make([]*ASEntry, len(ps.ASEntries))

go/lib/l4/udp.go

@@ -103,6 +103,9 @@ func (u *UDP) SetPldLen(pldLen int) {
 }
 
 func (u *UDP) Copy() L4Header {
+	if u == nil {
+		return nil
+	}
 	return &UDP{u.SrcPort, u.DstPort, u.TotalLen, append(common.RawBytes(nil), u.Checksum...)}
 }
 

go/lib/layers/extensions.go

@@ -149,6 +149,9 @@ func ExtnSCMPFromRaw(b common.RawBytes) (*ExtnSCMP, error) {
 }
 
 func (e *ExtnSCMP) Copy() common.Extension {
+	if e == nil {
+		return nil
+	}
 	return &ExtnSCMP{Error: e.Error, HopByHop: e.HopByHop}
 }
 

go/lib/sciond/types.go

@@ -130,6 +130,9 @@ type PathReq struct {
 }
 
 func (pathReq *PathReq) Copy() *PathReq {
+	if pathReq == nil {
+		return nil
+	}
 	return &PathReq{
 		Dst:      pathReq.Dst,
 		Src:      pathReq.Src,

go/lib/scmp/hdr.go

@@ -121,6 +121,9 @@ func (h *Hdr) L4Len() int {
 func (h *Hdr) Reverse() {}
 
 func (h *Hdr) Copy() l4.L4Header {
+	if h == nil {
+		return nil
+	}
 	return &Hdr{
 		h.Class, h.Type, h.TotalLen, append(common.RawBytes(nil), h.Checksum...), h.Timestamp,
 	}

go/lib/scmp/info.go

@@ -76,6 +76,9 @@ func InfoEchoFromRaw(b common.RawBytes) (*InfoEcho, error) {
 }
 
 func (e *InfoEcho) Copy() Info {
+	if e == nil {
+		return nil
+	}
 	return &InfoEcho{Id: e.Id, Seq: e.Seq}
 }
 
@@ -110,6 +113,9 @@ func InfoPktSizeFromRaw(b common.RawBytes) (*InfoPktSize, error) {
 }
 
 func (p *InfoPktSize) Copy() Info {
+	if p == nil {
+		return nil
+	}
 	return &InfoPktSize{Size: p.Size, MTU: p.MTU}
 }
 
@@ -160,6 +166,9 @@ func InfoPathOffsetsFromRaw(b common.RawBytes) (*InfoPathOffsets, error) {
 }
 
 func (p *InfoPathOffsets) Copy() Info {
+	if p == nil {
+		return nil
+	}
 	return &InfoPathOffsets{InfoF: p.InfoF, HopF: p.HopF, IfID: p.IfID, Ingress: p.Ingress}
 }
 
@@ -213,6 +222,9 @@ func InfoRevocationFromRaw(b common.RawBytes) (*InfoRevocation, error) {
 	return p, nil
 }
 func (r *InfoRevocation) Copy() Info {
+	if r == nil {
+		return nil
+	}
 	return &InfoRevocation{
 		r.InfoPathOffsets.Copy().(*InfoPathOffsets),
 		append(common.RawBytes(nil), r.RawSRev...),
@@ -257,6 +269,9 @@ func InfoExtIdxFromRaw(b common.RawBytes) (*InfoExtIdx, error) {
 }
 
 func (e *InfoExtIdx) Copy() Info {
+	if e == nil {
+		return nil
+	}
 	return &InfoExtIdx{Idx: e.Idx}
 }
 

go/lib/scmp/info_recordpath.go

@@ -125,6 +125,9 @@ func InfoRecordPathFromRaw(b common.RawBytes) (*InfoRecordPath, error) {
 }
 
 func (rec *InfoRecordPath) Copy() Info {
+	if rec == nil {
+		return nil
+	}
 	r := &InfoRecordPath{Id: rec.Id}
 	r.Entries = make([]*RecordPathEntry, rec.NumHops(), rec.TotalHops())
 	for i, e := range rec.Entries {

go/lib/scmp/info_traceroute.go

@@ -70,6 +70,9 @@ func InfoTraceRouteFromRaw(b common.RawBytes) (*InfoTraceRoute, error) {
 }
 
 func (e *InfoTraceRoute) Copy() Info {
+	if e == nil {
+		return nil
+	}
 	return &InfoTraceRoute{Id: e.Id, IfID: e.IfID, IA: e.IA, HopOff: e.HopOff, In: e.In}
 }
 

go/lib/scmp/meta.go

@@ -50,6 +50,9 @@ func MetaFromRaw(b []byte) (*Meta, error) {
 }
 
 func (m *Meta) Copy() *Meta {
+	if m == nil {
+		return nil
+	}
 	return &Meta{
 		InfoLen: m.InfoLen, CmnHdrLen: m.CmnHdrLen, AddrHdrLen: m.AddrHdrLen,
 		PathHdrLen: m.PathHdrLen, ExtHdrsLen: m.ExtHdrsLen,

go/lib/scmp/pld.go

@@ -85,6 +85,9 @@ func PldFromQuotes(ct ClassType, info Info, l4 common.L4ProtocolType, f QuoteFun
 }
 
 func (p *Payload) Copy() (common.Payload, error) {
+	if p == nil {
+		return nil, nil
+	}
 	c := &Payload{}
 	c.Meta = p.Meta.Copy()
 	c.Info = p.Info

go/lib/scrypto/cert/cert.go

@@ -177,6 +177,9 @@ func (c *Certificate) sigPack() (common.RawBytes, error) {
 }
 
 func (c *Certificate) Copy() *Certificate {
+	if c == nil {
+		return nil
+	}
 	n := &Certificate{
 		CanIssue:       c.CanIssue,
 		Comment:        c.Comment,

go/lib/scrypto/cert/chain.go

@@ -214,6 +214,9 @@ func (c *Chain) Compress() (common.RawBytes, error) {
 }
 
 func (c *Chain) Copy() *Chain {
+	if c == nil {
+		return nil
+	}
 	return &Chain{Issuer: c.Issuer.Copy(), Leaf: c.Leaf.Copy()}
 }
 

go/lib/snet/path.go

@@ -113,6 +113,9 @@ func (p *partialPath) Expiry() time.Time {
 }
 
 func (p *partialPath) Copy() Path {
+	if p == nil {
+		return nil
+	}
 	return &partialPath{
 		spath:       p.spath.Copy(),
 		overlay:     copyUDP(p.overlay),

go/lib/spath/path.go

@@ -69,6 +69,9 @@ func NewOneHop(isd addr.ISD, ifid common.IFIDType, ts time.Time, exp ExpTimeType
 }
 
 func (p *Path) Copy() *Path {
+	if p == nil {
+		return nil
+	}
 	return &Path{append(common.RawBytes(nil), p.Raw...), p.InfOff, p.HopOff}
 }
 

go/lib/spkt/spkt.go

@@ -36,31 +36,24 @@ type ScnPkt struct {
 }
 
 func (s *ScnPkt) Copy() (*ScnPkt, error) {
+	if s == nil {
+		return nil, nil
+	}
 	var err error
 	c := &ScnPkt{}
 	c.DstIA = s.DstIA
 	c.SrcIA = s.SrcIA
-	if s.DstHost != nil {
-		c.DstHost = s.DstHost.Copy()
-	}
-	if s.SrcHost != nil {
-		c.SrcHost = s.SrcHost.Copy()
-	}
-	if s.Path != nil {
-		c.Path = s.Path.Copy()
-	}
+	c.DstHost = s.DstHost.Copy()
+	c.SrcHost = s.SrcHost.Copy()
+	c.Path = s.Path.Copy()
 	for _, e := range s.HBHExt {
 		c.HBHExt = append(c.HBHExt, e.Copy())
 	}
 	for _, e := range s.E2EExt {
 		c.E2EExt = append(c.E2EExt, e.Copy())
 	}
-	if s.L4 != nil {
-		c.L4 = s.L4.Copy()
-	}
-	if s.Pld != nil {
-		c.Pld, err = s.Pld.Copy()
-	}
+	c.L4 = s.L4.Copy()
+	c.Pld, err = s.Pld.Copy()
 	return c, err
 }
 

go/lib/spse/scmp_auth/drkey.go

@@ -132,6 +132,9 @@ func (s *DRKeyExtn) Pack() (common.RawBytes, error) {
 }
 
 func (s *DRKeyExtn) Copy() common.Extension {
+	if s == nil {
+		return nil
+	}
 	c := NewDRKeyExtn()
 	c.Direction = s.Direction
 	copy(c.MAC, s.MAC)

go/lib/spse/scmp_auth/hashtree.go

@@ -134,6 +134,9 @@ func (s *HashTreeExtn) Pack() (common.RawBytes, error) {
 }
 
 func (s *HashTreeExtn) Copy() common.Extension {
+	if s == nil {
+		return nil
+	}
 	c, _ := NewHashTreeExtn(s.Height)
 	copy(c.Order, s.Order)
 	copy(c.Signature, s.Signature)

go/lib/spse/spse.go

@@ -188,6 +188,9 @@ func (s *Extn) Pack() (common.RawBytes, error) {
 }
 
 func (s *Extn) Copy() common.Extension {
+	if s == nil {
+		return nil
+	}
 	c, _ := NewExtn(s.SecMode)
 	copy(c.Metadata, s.Metadata)
 	copy(c.Authenticator, s.Authenticator)

go/lib/svc/resolver.go

@@ -202,6 +202,9 @@ func (p *path) Expiry() time.Time {
 }
 
 func (p *path) Copy() snet.Path {
+	if p == nil {
+		return nil
+	}
 	return &path{
 		spath:       p.spath.Copy(),
 		overlay:     snet.CopyUDPAddr(p.overlay),

go/lib/xtest/graph/graph.go

@@ -349,6 +349,9 @@ func newSolution(start addr.IA) *solution {
 }
 
 func (s *solution) Copy() *solution {
+	if s == nil {
+		return nil
+	}
 	newS := &solution{}
 	newS.CurrentIA = s.CurrentIA
 	newS.visited = make(map[addr.IA]struct{})

go/sig/internal/snetmigrate/path.go

@@ -57,6 +57,9 @@ func (p *emptyPath) Expiry() time.Time {
 }
 
 func (p *emptyPath) Copy() snet.Path {
+	if p == nil {
+		return nil
+	}
 	return &emptyPath{
 		source: p.source,
 	}