42 lines
611 B
Go
42 lines
611 B
Go
|
package context
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gofrs/uuid"
|
||
|
|
"github.com/miekg/dns"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
DNSTypeHost = "host"
|
||
|
|
DNSTypeFakeIP = "fakeip"
|
||
|
|
DNSTypeRaw = "raw"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DNSContext struct {
|
||
|
|
id uuid.UUID
|
||
|
|
msg *dns.Msg
|
||
|
|
tp string
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewDNSContext(msg *dns.Msg) *DNSContext {
|
||
|
|
id, _ := uuid.NewV4()
|
||
|
|
return &DNSContext{
|
||
|
|
id: id,
|
||
|
|
msg: msg,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// ID implement C.PlainContext ID
|
||
|
|
func (c *DNSContext) ID() uuid.UUID {
|
||
|
|
return c.id
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetType set type of response
|
||
|
|
func (c *DNSContext) SetType(tp string) {
|
||
|
|
c.tp = tp
|
||
|
|
}
|
||
|
|
|
||
|
|
// Type return type of response
|
||
|
|
func (c *DNSContext) Type() string {
|
||
|
|
return c.tp
|
||
|
|
}
|