parent
f33b02689c
commit
1b7492be6e
41 changed files with 3537 additions and 0 deletions
12
Gopkg.lock
generated
12
Gopkg.lock
generated
|
@ -373,6 +373,18 @@
|
|||
revision = "1195e3a8ee1a529d53eed7c624527a68555ddf1f"
|
||||
version = "v1.5.1"
|
||||
|
||||
[[projects]]
|
||||
branch = "master"
|
||||
name = "github.com/farsightsec/golang-framestream"
|
||||
packages = ["."]
|
||||
revision = "dec85654e8b8cf6712870afb14ee53d1c98cd5e2"
|
||||
|
||||
[[projects]]
|
||||
branch = "master"
|
||||
name = "github.com/dnstap/golang-dnstap"
|
||||
packages = ["."]
|
||||
revision = "c32b266c26b040b00f522a0183527a9eb2a1cf8b"
|
||||
|
||||
[solve-meta]
|
||||
analyzer-name = "dep"
|
||||
analyzer-version = 1
|
||||
|
|
|
@ -11,6 +11,7 @@ import (
|
|||
_ "github.com/coredns/coredns/middleware/cache"
|
||||
_ "github.com/coredns/coredns/middleware/chaos"
|
||||
_ "github.com/coredns/coredns/middleware/dnssec"
|
||||
_ "github.com/coredns/coredns/middleware/dnstap"
|
||||
_ "github.com/coredns/coredns/middleware/erratic"
|
||||
_ "github.com/coredns/coredns/middleware/errors"
|
||||
_ "github.com/coredns/coredns/middleware/etcd"
|
||||
|
|
|
@ -21,6 +21,7 @@ var directives = []string{
|
|||
"prometheus",
|
||||
"errors",
|
||||
"log",
|
||||
"dnstap",
|
||||
"chaos",
|
||||
"cache",
|
||||
"rewrite",
|
||||
|
|
|
@ -29,6 +29,7 @@
|
|||
70:prometheus:metrics
|
||||
80:errors:errors
|
||||
90:log:log
|
||||
95:dnstap:dnstap
|
||||
100:chaos:chaos
|
||||
110:cache:cache
|
||||
120:rewrite:rewrite
|
||||
|
|
20
middleware/dnstap/README.md
Normal file
20
middleware/dnstap/README.md
Normal file
|
@ -0,0 +1,20 @@
|
|||
# Dnstap
|
||||
|
||||
## Syntax
|
||||
|
||||
`dnstap SOCKET [full]`
|
||||
|
||||
* **SOCKET** is the socket path supplied to the dnstap command line tool.
|
||||
* `full` to include the wire-format dns message.
|
||||
|
||||
## Dnstap command line tool
|
||||
|
||||
```sh
|
||||
go get github.com/dnstap/golang-dnstap
|
||||
cd $GOPATH/src/github.com/dnstap/golang-dnstap/dnstap
|
||||
go build
|
||||
./dnstap -u /tmp/dnstap.sock
|
||||
./dnstap -u /tmp/dnstap.sock -y
|
||||
```
|
||||
|
||||
There is a buffer, expect at least 13 requests before the server sends its dnstap messages to the socket.
|
51
middleware/dnstap/handler.go
Normal file
51
middleware/dnstap/handler.go
Normal file
|
@ -0,0 +1,51 @@
|
|||
package dnstap
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"golang.org/x/net/context"
|
||||
"io"
|
||||
|
||||
"github.com/coredns/coredns/middleware"
|
||||
"github.com/coredns/coredns/middleware/dnstap/msg"
|
||||
"github.com/coredns/coredns/middleware/dnstap/taprw"
|
||||
|
||||
tap "github.com/dnstap/golang-dnstap"
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
type Dnstap struct {
|
||||
Next middleware.Handler
|
||||
Out io.Writer
|
||||
Pack bool
|
||||
}
|
||||
|
||||
func tapMessageTo(w io.Writer, m *tap.Message) error {
|
||||
frame, err := msg.Marshal(m)
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal: %s", err)
|
||||
}
|
||||
_, err = w.Write(frame)
|
||||
return err
|
||||
}
|
||||
|
||||
func (h Dnstap) TapMessage(m *tap.Message) error {
|
||||
return tapMessageTo(h.Out, m)
|
||||
}
|
||||
|
||||
func (h Dnstap) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
|
||||
rw := taprw.ResponseWriter{ResponseWriter: w, Taper: &h, Query: r, Pack: h.Pack}
|
||||
rw.QueryEpoch()
|
||||
|
||||
code, err := middleware.NextOrFailure(h.Name(), h.Next, ctx, &rw, r)
|
||||
if err != nil {
|
||||
// ignore dnstap errors
|
||||
return code, err
|
||||
}
|
||||
|
||||
if err := rw.DnstapError(); err != nil {
|
||||
return code, middleware.Error("dnstap", err)
|
||||
}
|
||||
|
||||
return code, nil
|
||||
}
|
||||
func (h Dnstap) Name() string { return "dnstap" }
|
65
middleware/dnstap/handler_test.go
Normal file
65
middleware/dnstap/handler_test.go
Normal file
|
@ -0,0 +1,65 @@
|
|||
package dnstap
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/coredns/coredns/middleware/dnstap/test"
|
||||
mwtest "github.com/coredns/coredns/middleware/test"
|
||||
|
||||
tap "github.com/dnstap/golang-dnstap"
|
||||
"github.com/golang/protobuf/proto"
|
||||
"github.com/miekg/dns"
|
||||
"golang.org/x/net/context"
|
||||
)
|
||||
|
||||
func testCase(t *testing.T, tapq, tapr *tap.Message, q, r *dns.Msg) {
|
||||
w := writer{}
|
||||
w.queue = append(w.queue, tapq, tapr)
|
||||
h := Dnstap{
|
||||
Next: mwtest.HandlerFunc(func(_ context.Context,
|
||||
w dns.ResponseWriter, _ *dns.Msg) (int, error) {
|
||||
|
||||
return 0, w.WriteMsg(r)
|
||||
}),
|
||||
Out: &w,
|
||||
Pack: false,
|
||||
}
|
||||
_, err := h.ServeDNS(context.TODO(), &mwtest.ResponseWriter{}, q)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
type writer struct {
|
||||
queue []*tap.Message
|
||||
}
|
||||
|
||||
func (w *writer) Write(b []byte) (int, error) {
|
||||
e := tap.Dnstap{}
|
||||
if err := proto.Unmarshal(b, &e); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if len(w.queue) == 0 {
|
||||
return 0, errors.New("message not expected")
|
||||
}
|
||||
if !test.MsgEqual(w.queue[0], e.Message) {
|
||||
return 0, fmt.Errorf("want: %v, have: %v", w.queue[0], e.Message)
|
||||
}
|
||||
w.queue = w.queue[1:]
|
||||
return len(b), nil
|
||||
}
|
||||
|
||||
func TestDnstap(t *testing.T) {
|
||||
q := mwtest.Case{Qname: "example.org", Qtype: dns.TypeA}.Msg()
|
||||
r := mwtest.Case{
|
||||
Qname: "example.org.", Qtype: dns.TypeA,
|
||||
Answer: []dns.RR{
|
||||
mwtest.A("example.org. 3600 IN A 10.0.0.1"),
|
||||
},
|
||||
}.Msg()
|
||||
tapq := test.TestingData().ToClientQuery()
|
||||
tapr := test.TestingData().ToClientResponse()
|
||||
testCase(t, tapq, tapr, q, r)
|
||||
}
|
89
middleware/dnstap/msg/msg.go
Normal file
89
middleware/dnstap/msg/msg.go
Normal file
|
@ -0,0 +1,89 @@
|
|||
// Package msg helps to build a dnstap Message.
|
||||
package msg
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net"
|
||||
"time"
|
||||
|
||||
"github.com/coredns/coredns/request"
|
||||
|
||||
tap "github.com/dnstap/golang-dnstap"
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
// Data helps to build a dnstap Message.
|
||||
// It can be transformed into the actual Message using this package.
|
||||
type Data struct {
|
||||
Type tap.Message_Type
|
||||
Packed []byte
|
||||
SocketProto tap.SocketProtocol
|
||||
SocketFam tap.SocketFamily
|
||||
Address []byte
|
||||
Port uint32
|
||||
TimeSec uint64
|
||||
}
|
||||
|
||||
func (d *Data) FromRequest(r request.Request) error {
|
||||
switch addr := r.W.RemoteAddr().(type) {
|
||||
case *net.TCPAddr:
|
||||
d.Address = addr.IP
|
||||
d.Port = uint32(addr.Port)
|
||||
d.SocketProto = tap.SocketProtocol_TCP
|
||||
case *net.UDPAddr:
|
||||
d.Address = addr.IP
|
||||
d.Port = uint32(addr.Port)
|
||||
d.SocketProto = tap.SocketProtocol_UDP
|
||||
default:
|
||||
return errors.New("unknown remote address type")
|
||||
}
|
||||
|
||||
if a := net.IP(d.Address); a.To4() != nil {
|
||||
d.SocketFam = tap.SocketFamily_INET
|
||||
} else {
|
||||
d.SocketFam = tap.SocketFamily_INET6
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *Data) Pack(m *dns.Msg) error {
|
||||
packed, err := m.Pack()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
d.Packed = packed
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *Data) Epoch() {
|
||||
d.TimeSec = uint64(time.Now().Unix())
|
||||
}
|
||||
|
||||
// Transform the data into a client response message.
|
||||
func (d *Data) ToClientResponse() *tap.Message {
|
||||
d.Type = tap.Message_CLIENT_RESPONSE
|
||||
return &tap.Message{
|
||||
Type: &d.Type,
|
||||
SocketFamily: &d.SocketFam,
|
||||
SocketProtocol: &d.SocketProto,
|
||||
ResponseTimeSec: &d.TimeSec,
|
||||
ResponseMessage: d.Packed,
|
||||
QueryAddress: d.Address,
|
||||
QueryPort: &d.Port,
|
||||
}
|
||||
}
|
||||
|
||||
// Transform the data into a client query message.
|
||||
func (d *Data) ToClientQuery() *tap.Message {
|
||||
d.Type = tap.Message_CLIENT_QUERY
|
||||
return &tap.Message{
|
||||
Type: &d.Type,
|
||||
SocketFamily: &d.SocketFam,
|
||||
SocketProtocol: &d.SocketProto,
|
||||
QueryTimeSec: &d.TimeSec,
|
||||
QueryMessage: d.Packed,
|
||||
QueryAddress: d.Address,
|
||||
QueryPort: &d.Port,
|
||||
}
|
||||
}
|
42
middleware/dnstap/msg/msg_test.go
Normal file
42
middleware/dnstap/msg/msg_test.go
Normal file
|
@ -0,0 +1,42 @@
|
|||
package msg
|
||||
|
||||
import (
|
||||
"net"
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/coredns/coredns/middleware/test"
|
||||
"github.com/coredns/coredns/request"
|
||||
|
||||
tap "github.com/dnstap/golang-dnstap"
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
func testRequest(t *testing.T, expected Data, r request.Request) {
|
||||
d := Data{}
|
||||
if err := d.FromRequest(r); err != nil {
|
||||
t.Fail()
|
||||
return
|
||||
}
|
||||
if d.SocketProto != expected.SocketProto ||
|
||||
d.SocketFam != expected.SocketFam ||
|
||||
!reflect.DeepEqual(d.Address, expected.Address) ||
|
||||
d.Port != expected.Port {
|
||||
t.Fatalf("expected: %v, have: %v", expected, d)
|
||||
return
|
||||
}
|
||||
}
|
||||
func TestRequest(t *testing.T) {
|
||||
testRequest(t, Data{
|
||||
SocketProto: tap.SocketProtocol_UDP,
|
||||
SocketFam: tap.SocketFamily_INET,
|
||||
Address: net.ParseIP("10.240.0.1"),
|
||||
Port: 40212,
|
||||
}, testingRequest())
|
||||
}
|
||||
func testingRequest() request.Request {
|
||||
m := new(dns.Msg)
|
||||
m.SetQuestion("example.com.", dns.TypeA)
|
||||
m.SetEdns0(4097, true)
|
||||
return request.Request{W: &test.ResponseWriter{}, Req: m}
|
||||
}
|
25
middleware/dnstap/msg/wrapper.go
Normal file
25
middleware/dnstap/msg/wrapper.go
Normal file
|
@ -0,0 +1,25 @@
|
|||
package msg
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
lib "github.com/dnstap/golang-dnstap"
|
||||
"github.com/golang/protobuf/proto"
|
||||
)
|
||||
|
||||
func wrap(m *lib.Message) *lib.Dnstap {
|
||||
t := lib.Dnstap_MESSAGE
|
||||
return &lib.Dnstap{
|
||||
Type: &t,
|
||||
Message: m,
|
||||
}
|
||||
}
|
||||
|
||||
func Marshal(m *lib.Message) (data []byte, err error) {
|
||||
data, err = proto.Marshal(wrap(m))
|
||||
if err != nil {
|
||||
err = fmt.Errorf("proto: %s", err)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
86
middleware/dnstap/out/socket.go
Normal file
86
middleware/dnstap/out/socket.go
Normal file
|
@ -0,0 +1,86 @@
|
|||
package out
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
|
||||
fs "github.com/farsightsec/golang-framestream"
|
||||
)
|
||||
|
||||
// Socket is a Frame Streams encoder over a UNIX socket.
|
||||
type Socket struct {
|
||||
path string
|
||||
enc *fs.Encoder
|
||||
conn net.Conn
|
||||
err error
|
||||
}
|
||||
|
||||
func openSocket(s *Socket) error {
|
||||
conn, err := net.Dial("unix", s.path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
s.conn = conn
|
||||
|
||||
enc, err := fs.NewEncoder(conn, &fs.EncoderOptions{
|
||||
ContentType: []byte("protobuf:dnstap.Dnstap"),
|
||||
Bidirectional: true,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
s.enc = enc
|
||||
|
||||
s.err = nil
|
||||
return nil
|
||||
}
|
||||
|
||||
// NewSocket will always return a new Socket.
|
||||
// err if nothing is listening to it, it will attempt to reconnect on the next Write.
|
||||
func NewSocket(path string) (s *Socket, err error) {
|
||||
s = &Socket{path: path}
|
||||
if err = openSocket(s); err != nil {
|
||||
err = fmt.Errorf("open socket: %s", err)
|
||||
s.err = err
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Write a single Frame Streams frame.
|
||||
func (s *Socket) Write(frame []byte) (int, error) {
|
||||
if s.err != nil {
|
||||
// is the dnstap tool listening?
|
||||
if err := openSocket(s); err != nil {
|
||||
return 0, fmt.Errorf("open socket: %s", err)
|
||||
}
|
||||
}
|
||||
n, err := s.enc.Write(frame)
|
||||
if err != nil {
|
||||
// the dnstap command line tool is down
|
||||
s.conn.Close()
|
||||
s.err = err
|
||||
return 0, err
|
||||
}
|
||||
return n, nil
|
||||
|
||||
}
|
||||
|
||||
// Close the socket and flush the remaining frames.
|
||||
func (s *Socket) Close() error {
|
||||
if s.err != nil {
|
||||
// nothing to close
|
||||
return nil
|
||||
}
|
||||
|
||||
defer s.conn.Close()
|
||||
|
||||
if err := s.enc.Flush(); err != nil {
|
||||
return fmt.Errorf("flush: %s", err)
|
||||
}
|
||||
if err := s.enc.Close(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
94
middleware/dnstap/out/socket_test.go
Normal file
94
middleware/dnstap/out/socket_test.go
Normal file
|
@ -0,0 +1,94 @@
|
|||
package out
|
||||
|
||||
import (
|
||||
"net"
|
||||
"testing"
|
||||
|
||||
fs "github.com/farsightsec/golang-framestream"
|
||||
)
|
||||
|
||||
func acceptOne(t *testing.T, l net.Listener) {
|
||||
server, err := l.Accept()
|
||||
if err != nil {
|
||||
t.Fatalf("server accept: %s", err)
|
||||
return
|
||||
}
|
||||
|
||||
dec, err := fs.NewDecoder(server, &fs.DecoderOptions{
|
||||
ContentType: []byte("protobuf:dnstap.Dnstap"),
|
||||
Bidirectional: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("server decoder: %s", err)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err := dec.Decode(); err != nil {
|
||||
t.Errorf("server decode: %s", err)
|
||||
}
|
||||
|
||||
if err := server.Close(); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
func sendOne(socket *Socket) error {
|
||||
if _, err := socket.Write([]byte("frame")); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := socket.enc.Flush(); err != nil {
|
||||
// Would happen during Write in real life.
|
||||
socket.conn.Close()
|
||||
socket.err = err
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
func TestSocket(t *testing.T) {
|
||||
socket, err := NewSocket("dnstap.sock")
|
||||
if err == nil {
|
||||
t.Fatal("new socket: not listening but no error")
|
||||
return
|
||||
}
|
||||
|
||||
if err := sendOne(socket); err == nil {
|
||||
t.Fatal("not listening but no error")
|
||||
return
|
||||
}
|
||||
|
||||
l, err := net.Listen("unix", "dnstap.sock")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
}
|
||||
|
||||
wait := make(chan bool)
|
||||
go func() {
|
||||
acceptOne(t, l)
|
||||
wait <- true
|
||||
}()
|
||||
|
||||
if err := sendOne(socket); err != nil {
|
||||
t.Fatalf("send one: %s", err)
|
||||
return
|
||||
}
|
||||
|
||||
<-wait
|
||||
if err := sendOne(socket); err == nil {
|
||||
panic("must fail")
|
||||
}
|
||||
|
||||
go func() {
|
||||
acceptOne(t, l)
|
||||
wait <- true
|
||||
}()
|
||||
|
||||
if err := sendOne(socket); err != nil {
|
||||
t.Fatalf("send one: %s", err)
|
||||
return
|
||||
}
|
||||
|
||||
<-wait
|
||||
if err := l.Close(); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
70
middleware/dnstap/setup.go
Normal file
70
middleware/dnstap/setup.go
Normal file
|
@ -0,0 +1,70 @@
|
|||
package dnstap
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
"github.com/coredns/coredns/core/dnsserver"
|
||||
"github.com/coredns/coredns/middleware"
|
||||
"github.com/coredns/coredns/middleware/dnstap/out"
|
||||
|
||||
"github.com/mholt/caddy"
|
||||
"github.com/mholt/caddy/caddyfile"
|
||||
)
|
||||
|
||||
func init() {
|
||||
caddy.RegisterPlugin("dnstap", caddy.Plugin{
|
||||
ServerType: "dns",
|
||||
Action: wrapSetup,
|
||||
})
|
||||
}
|
||||
|
||||
func wrapSetup(c *caddy.Controller) error {
|
||||
if err := setup(c); err != nil {
|
||||
return middleware.Error("dnstap", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func parseConfig(c *caddyfile.Dispenser) (path string, full bool, err error) {
|
||||
c.Next() // directive name
|
||||
|
||||
if !c.Args(&path) {
|
||||
err = c.ArgErr()
|
||||
return
|
||||
}
|
||||
|
||||
full = c.NextArg() && c.Val() == "full"
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func setup(c *caddy.Controller) error {
|
||||
path, full, err := parseConfig(&c.Dispenser)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
dnstap := Dnstap{Pack: full}
|
||||
|
||||
o, err := out.NewSocket(path)
|
||||
if err != nil {
|
||||
log.Printf("[WARN] Can't connect to %s at the moment", path)
|
||||
}
|
||||
dnstap.Out = o
|
||||
|
||||
c.OnShutdown(func() error {
|
||||
if err := o.Close(); err != nil {
|
||||
return fmt.Errorf("output: %s", err)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
dnsserver.GetConfig(c).AddMiddleware(
|
||||
func(next middleware.Handler) middleware.Handler {
|
||||
dnstap.Next = next
|
||||
return dnstap
|
||||
})
|
||||
|
||||
return nil
|
||||
}
|
19
middleware/dnstap/setup_test.go
Normal file
19
middleware/dnstap/setup_test.go
Normal file
|
@ -0,0 +1,19 @@
|
|||
package dnstap
|
||||
|
||||
import (
|
||||
"github.com/mholt/caddy"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestConfig(t *testing.T) {
|
||||
file := "dnstap dnstap.sock full"
|
||||
c := caddy.NewTestController("dns", file)
|
||||
if path, full, err := parseConfig(&c.Dispenser); path != "dnstap.sock" || !full {
|
||||
t.Fatalf("%s: %s", file, err)
|
||||
}
|
||||
file = "dnstap dnstap.sock"
|
||||
c = caddy.NewTestController("dns", file)
|
||||
if path, full, err := parseConfig(&c.Dispenser); path != "dnstap.sock" || full {
|
||||
t.Fatalf("%s: %s", file, err)
|
||||
}
|
||||
}
|
84
middleware/dnstap/taprw/writer.go
Normal file
84
middleware/dnstap/taprw/writer.go
Normal file
|
@ -0,0 +1,84 @@
|
|||
// Package taprw takes a query and intercepts the response.
|
||||
// It will log both after the response is written.
|
||||
package taprw
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/coredns/coredns/middleware/dnstap/msg"
|
||||
"github.com/coredns/coredns/request"
|
||||
|
||||
tap "github.com/dnstap/golang-dnstap"
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
type Taper interface {
|
||||
TapMessage(m *tap.Message) error
|
||||
}
|
||||
|
||||
// Single request use.
|
||||
type ResponseWriter struct {
|
||||
queryData msg.Data
|
||||
Query *dns.Msg
|
||||
dns.ResponseWriter
|
||||
Taper
|
||||
Pack bool
|
||||
err error
|
||||
}
|
||||
|
||||
// Check if a dnstap error occured.
|
||||
// Set during ResponseWriter.Write.
|
||||
func (w ResponseWriter) DnstapError() error {
|
||||
return w.err
|
||||
}
|
||||
|
||||
// To be called as soon as possible.
|
||||
func (w *ResponseWriter) QueryEpoch() {
|
||||
w.queryData.Epoch()
|
||||
}
|
||||
|
||||
// Write back the response to the client and THEN work on logging the request
|
||||
// and response to dnstap.
|
||||
// Dnstap errors to be checked by DnstapError.
|
||||
func (w *ResponseWriter) WriteMsg(resp *dns.Msg) error {
|
||||
writeErr := w.ResponseWriter.WriteMsg(resp)
|
||||
|
||||
if err := tapQuery(w); err != nil {
|
||||
w.err = fmt.Errorf("client query: %s", err)
|
||||
// don't forget to call DnstapError later
|
||||
}
|
||||
|
||||
if writeErr == nil {
|
||||
if err := tapResponse(w, resp); err != nil {
|
||||
w.err = fmt.Errorf("client response: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
return writeErr
|
||||
}
|
||||
func tapQuery(w *ResponseWriter) error {
|
||||
req := request.Request{W: w.ResponseWriter, Req: w.Query}
|
||||
if err := w.queryData.FromRequest(req); err != nil {
|
||||
return err
|
||||
}
|
||||
if w.Pack {
|
||||
if err := w.queryData.Pack(w.Query); err != nil {
|
||||
return fmt.Errorf("pack: %s", err)
|
||||
}
|
||||
}
|
||||
return w.Taper.TapMessage(w.queryData.ToClientQuery())
|
||||
}
|
||||
func tapResponse(w *ResponseWriter, resp *dns.Msg) error {
|
||||
d := &msg.Data{}
|
||||
d.Epoch()
|
||||
req := request.Request{W: w, Req: resp}
|
||||
if err := d.FromRequest(req); err != nil {
|
||||
return err
|
||||
}
|
||||
if w.Pack {
|
||||
if err := d.Pack(resp); err != nil {
|
||||
return fmt.Errorf("pack: %s", err)
|
||||
}
|
||||
}
|
||||
return w.Taper.TapMessage(d.ToClientResponse())
|
||||
}
|
96
middleware/dnstap/taprw/writer_test.go
Normal file
96
middleware/dnstap/taprw/writer_test.go
Normal file
|
@ -0,0 +1,96 @@
|
|||
package taprw
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"testing"
|
||||
|
||||
"github.com/coredns/coredns/middleware/dnstap/test"
|
||||
mwtest "github.com/coredns/coredns/middleware/test"
|
||||
|
||||
tap "github.com/dnstap/golang-dnstap"
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
type TapFailer struct {
|
||||
}
|
||||
|
||||
func (TapFailer) TapMessage(*tap.Message) error {
|
||||
return errors.New("failed")
|
||||
}
|
||||
|
||||
func TestDnstapError(t *testing.T) {
|
||||
rw := ResponseWriter{
|
||||
Query: new(dns.Msg),
|
||||
ResponseWriter: &mwtest.ResponseWriter{},
|
||||
Taper: TapFailer{},
|
||||
}
|
||||
if err := rw.WriteMsg(new(dns.Msg)); err != nil {
|
||||
t.Errorf("dnstap error during Write: %s", err)
|
||||
}
|
||||
if rw.DnstapError() == nil {
|
||||
t.Fatal("no dnstap error")
|
||||
}
|
||||
}
|
||||
|
||||
func testingMsg() (m *dns.Msg) {
|
||||
m = new(dns.Msg)
|
||||
m.SetQuestion("example.com.", dns.TypeA)
|
||||
m.SetEdns0(4097, true)
|
||||
return
|
||||
}
|
||||
|
||||
func TestClientResponse(t *testing.T) {
|
||||
trapper := test.TrapTaper{}
|
||||
rw := ResponseWriter{
|
||||
Pack: true,
|
||||
Taper: &trapper,
|
||||
ResponseWriter: &mwtest.ResponseWriter{},
|
||||
}
|
||||
d := test.TestingData()
|
||||
m := testingMsg()
|
||||
|
||||
// will the wire-format msg be reported?
|
||||
bin, err := m.Pack()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
}
|
||||
d.Packed = bin
|
||||
|
||||
if err := tapResponse(&rw, m); err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
}
|
||||
want := d.ToClientResponse()
|
||||
if l := len(trapper.Trap); l != 1 {
|
||||
t.Fatalf("%d msg trapped", l)
|
||||
return
|
||||
}
|
||||
have := trapper.Trap[0]
|
||||
if !test.MsgEqual(want, have) {
|
||||
t.Fatalf("want: %v\nhave: %v", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func TestClientQuery(t *testing.T) {
|
||||
trapper := test.TrapTaper{}
|
||||
rw := ResponseWriter{
|
||||
Pack: false, // no binary this time
|
||||
Taper: &trapper,
|
||||
ResponseWriter: &mwtest.ResponseWriter{},
|
||||
Query: testingMsg(),
|
||||
}
|
||||
if err := tapQuery(&rw); err != nil {
|
||||
t.Fatal(err)
|
||||
return
|
||||
}
|
||||
want := test.TestingData().ToClientQuery()
|
||||
if l := len(trapper.Trap); l != 1 {
|
||||
t.Fatalf("%d msg trapped", l)
|
||||
return
|
||||
}
|
||||
have := trapper.Trap[0]
|
||||
if !test.MsgEqual(want, have) {
|
||||
t.Fatalf("want: %v\nhave: %v", want, have)
|
||||
}
|
||||
}
|
64
middleware/dnstap/test/helpers.go
Normal file
64
middleware/dnstap/test/helpers.go
Normal file
|
@ -0,0 +1,64 @@
|
|||
package test
|
||||
|
||||
import (
|
||||
"net"
|
||||
"reflect"
|
||||
|
||||
"github.com/coredns/coredns/middleware/dnstap/msg"
|
||||
|
||||
tap "github.com/dnstap/golang-dnstap"
|
||||
)
|
||||
|
||||
func TestingData() (d *msg.Data) {
|
||||
d = &msg.Data{
|
||||
Type: tap.Message_CLIENT_RESPONSE,
|
||||
SocketFam: tap.SocketFamily_INET,
|
||||
SocketProto: tap.SocketProtocol_UDP,
|
||||
Address: net.ParseIP("10.240.0.1"),
|
||||
Port: 40212,
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
type comp struct {
|
||||
Type *tap.Message_Type
|
||||
SF *tap.SocketFamily
|
||||
SP *tap.SocketProtocol
|
||||
QA []byte
|
||||
RA []byte
|
||||
QP *uint32
|
||||
RP *uint32
|
||||
QTSec bool
|
||||
RTSec bool
|
||||
RM []byte
|
||||
QM []byte
|
||||
}
|
||||
|
||||
func toComp(m *tap.Message) comp {
|
||||
return comp{
|
||||
Type: m.Type,
|
||||
SF: m.SocketFamily,
|
||||
SP: m.SocketProtocol,
|
||||
QA: m.QueryAddress,
|
||||
RA: m.ResponseAddress,
|
||||
QP: m.QueryPort,
|
||||
RP: m.ResponsePort,
|
||||
QTSec: m.QueryTimeSec != nil,
|
||||
RTSec: m.ResponseTimeSec != nil,
|
||||
RM: m.ResponseMessage,
|
||||
QM: m.QueryMessage,
|
||||
}
|
||||
}
|
||||
|
||||
func MsgEqual(a, b *tap.Message) bool {
|
||||
return reflect.DeepEqual(toComp(a), toComp(b))
|
||||
}
|
||||
|
||||
type TrapTaper struct {
|
||||
Trap []*tap.Message
|
||||
}
|
||||
|
||||
func (t *TrapTaper) TapMessage(m *tap.Message) error {
|
||||
t.Trap = append(t.Trap, m)
|
||||
return nil
|
||||
}
|
14
vendor/github.com/dnstap/golang-dnstap/COPYRIGHT
generated
vendored
Normal file
14
vendor/github.com/dnstap/golang-dnstap/COPYRIGHT
generated
vendored
Normal file
|
@ -0,0 +1,14 @@
|
|||
Copyright (c) 2013-2014 by Farsight Security, Inc.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
|
73
vendor/github.com/dnstap/golang-dnstap/FrameStreamInput.go
generated
vendored
Normal file
73
vendor/github.com/dnstap/golang-dnstap/FrameStreamInput.go
generated
vendored
Normal file
|
@ -0,0 +1,73 @@
|
|||
/*
|
||||
* Copyright (c) 2013-2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package dnstap
|
||||
|
||||
import (
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/farsightsec/golang-framestream"
|
||||
)
|
||||
|
||||
type FrameStreamInput struct {
|
||||
wait chan bool
|
||||
decoder *framestream.Decoder
|
||||
}
|
||||
|
||||
func NewFrameStreamInput(r io.ReadWriter, bi bool) (input *FrameStreamInput, err error) {
|
||||
input = new(FrameStreamInput)
|
||||
decoderOptions := framestream.DecoderOptions{
|
||||
ContentType: FSContentType,
|
||||
Bidirectional: bi,
|
||||
}
|
||||
input.decoder, err = framestream.NewDecoder(r, &decoderOptions)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
input.wait = make(chan bool)
|
||||
return
|
||||
}
|
||||
|
||||
func NewFrameStreamInputFromFilename(fname string) (input *FrameStreamInput, err error) {
|
||||
file, err := os.Open(fname)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
input, err = NewFrameStreamInput(file, false)
|
||||
return
|
||||
}
|
||||
|
||||
func (input *FrameStreamInput) ReadInto(output chan []byte) {
|
||||
for {
|
||||
buf, err := input.decoder.Decode()
|
||||
if err != nil {
|
||||
if err != io.EOF {
|
||||
log.Printf("framestream.Decoder.Decode() failed: %s\n", err)
|
||||
}
|
||||
break
|
||||
}
|
||||
newbuf := make([]byte, len(buf))
|
||||
copy(newbuf, buf)
|
||||
output <- newbuf
|
||||
}
|
||||
close(input.wait)
|
||||
}
|
||||
|
||||
func (input *FrameStreamInput) Wait() {
|
||||
<-input.wait
|
||||
}
|
74
vendor/github.com/dnstap/golang-dnstap/FrameStreamOutput.go
generated
vendored
Normal file
74
vendor/github.com/dnstap/golang-dnstap/FrameStreamOutput.go
generated
vendored
Normal file
|
@ -0,0 +1,74 @@
|
|||
/*
|
||||
* Copyright (c) 2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package dnstap
|
||||
|
||||
import (
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/farsightsec/golang-framestream"
|
||||
)
|
||||
|
||||
type FrameStreamOutput struct {
|
||||
outputChannel chan []byte
|
||||
wait chan bool
|
||||
enc *framestream.Encoder
|
||||
}
|
||||
|
||||
func NewFrameStreamOutput(w io.Writer) (o *FrameStreamOutput, err error) {
|
||||
o = new(FrameStreamOutput)
|
||||
o.outputChannel = make(chan []byte, outputChannelSize)
|
||||
o.enc, err = framestream.NewEncoder(w, &framestream.EncoderOptions{ContentType: FSContentType})
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
o.wait = make(chan bool)
|
||||
return
|
||||
}
|
||||
|
||||
func NewFrameStreamOutputFromFilename(fname string) (o *FrameStreamOutput, err error) {
|
||||
if fname == "" || fname == "-" {
|
||||
return NewFrameStreamOutput(os.Stdout)
|
||||
}
|
||||
w, err := os.Create(fname)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return NewFrameStreamOutput(w)
|
||||
}
|
||||
|
||||
func (o *FrameStreamOutput) GetOutputChannel() chan []byte {
|
||||
return o.outputChannel
|
||||
}
|
||||
|
||||
func (o *FrameStreamOutput) RunOutputLoop() {
|
||||
for frame := range o.outputChannel {
|
||||
if _, err := o.enc.Write(frame); err != nil {
|
||||
log.Fatalf("framestream.Encoder.Write() failed: %s\n", err)
|
||||
break
|
||||
}
|
||||
}
|
||||
close(o.wait)
|
||||
}
|
||||
|
||||
func (o *FrameStreamOutput) Close() {
|
||||
close(o.outputChannel)
|
||||
<-o.wait
|
||||
o.enc.Flush()
|
||||
o.enc.Close()
|
||||
}
|
64
vendor/github.com/dnstap/golang-dnstap/FrameStreamSockInput.go
generated
vendored
Normal file
64
vendor/github.com/dnstap/golang-dnstap/FrameStreamSockInput.go
generated
vendored
Normal file
|
@ -0,0 +1,64 @@
|
|||
/*
|
||||
* Copyright (c) 2013-2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package dnstap
|
||||
|
||||
import (
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
)
|
||||
|
||||
type FrameStreamSockInput struct {
|
||||
wait chan bool
|
||||
listener net.Listener
|
||||
}
|
||||
|
||||
func NewFrameStreamSockInput(listener net.Listener) (input *FrameStreamSockInput) {
|
||||
input = new(FrameStreamSockInput)
|
||||
input.listener = listener
|
||||
return
|
||||
}
|
||||
|
||||
func NewFrameStreamSockInputFromPath(socketPath string) (input *FrameStreamSockInput, err error) {
|
||||
os.Remove(socketPath)
|
||||
listener, err := net.Listen("unix", socketPath)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return NewFrameStreamSockInput(listener), nil
|
||||
}
|
||||
|
||||
func (input *FrameStreamSockInput) ReadInto(output chan []byte) {
|
||||
for {
|
||||
conn, err := input.listener.Accept()
|
||||
if err != nil {
|
||||
log.Printf("net.Listener.Accept() failed: %s\n", err)
|
||||
continue
|
||||
}
|
||||
i, err := NewFrameStreamInput(conn, true)
|
||||
if err != nil {
|
||||
log.Printf("dnstap.NewFrameStreamInput() failed: %s\n", err)
|
||||
continue
|
||||
}
|
||||
log.Printf("dnstap.FrameStreamSockInput: accepted a socket connection\n")
|
||||
go i.ReadInto(output)
|
||||
}
|
||||
}
|
||||
|
||||
func (input *FrameStreamSockInput) Wait() {
|
||||
select {}
|
||||
}
|
202
vendor/github.com/dnstap/golang-dnstap/LICENSE
generated
vendored
Normal file
202
vendor/github.com/dnstap/golang-dnstap/LICENSE
generated
vendored
Normal file
|
@ -0,0 +1,202 @@
|
|||
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright [yyyy] [name of copyright owner]
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
172
vendor/github.com/dnstap/golang-dnstap/QuietTextFormat.go
generated
vendored
Normal file
172
vendor/github.com/dnstap/golang-dnstap/QuietTextFormat.go
generated
vendored
Normal file
|
@ -0,0 +1,172 @@
|
|||
/*
|
||||
* Copyright (c) 2013-2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package dnstap
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
const quietTimeFormat = "15:04:05"
|
||||
|
||||
func textConvertTime(s *bytes.Buffer, secs *uint64, nsecs *uint32) {
|
||||
if secs != nil {
|
||||
s.WriteString(time.Unix(int64(*secs), 0).Format(quietTimeFormat))
|
||||
} else {
|
||||
s.WriteString("??:??:??")
|
||||
}
|
||||
if nsecs != nil {
|
||||
s.WriteString(fmt.Sprintf(".%06d", *nsecs/1000))
|
||||
} else {
|
||||
s.WriteString(".??????")
|
||||
}
|
||||
}
|
||||
|
||||
func textConvertIP(s *bytes.Buffer, ip []byte) {
|
||||
if ip != nil {
|
||||
s.WriteString(net.IP(ip).String())
|
||||
} else {
|
||||
s.WriteString("MISSING_ADDRESS")
|
||||
}
|
||||
}
|
||||
|
||||
func textConvertMessage(m *Message, s *bytes.Buffer) {
|
||||
isQuery := false
|
||||
printQueryAddress := false
|
||||
|
||||
switch *m.Type {
|
||||
case Message_CLIENT_QUERY,
|
||||
Message_RESOLVER_QUERY,
|
||||
Message_AUTH_QUERY,
|
||||
Message_FORWARDER_QUERY,
|
||||
Message_TOOL_QUERY:
|
||||
isQuery = true
|
||||
case Message_CLIENT_RESPONSE,
|
||||
Message_RESOLVER_RESPONSE,
|
||||
Message_AUTH_RESPONSE,
|
||||
Message_FORWARDER_RESPONSE,
|
||||
Message_TOOL_RESPONSE:
|
||||
isQuery = false
|
||||
default:
|
||||
s.WriteString("[unhandled Message.Type]\n")
|
||||
return
|
||||
}
|
||||
|
||||
if isQuery {
|
||||
textConvertTime(s, m.QueryTimeSec, m.QueryTimeNsec)
|
||||
} else {
|
||||
textConvertTime(s, m.ResponseTimeSec, m.ResponseTimeNsec)
|
||||
}
|
||||
s.WriteString(" ")
|
||||
|
||||
switch *m.Type {
|
||||
case Message_CLIENT_QUERY,
|
||||
Message_CLIENT_RESPONSE:
|
||||
{
|
||||
s.WriteString("C")
|
||||
}
|
||||
case Message_RESOLVER_QUERY,
|
||||
Message_RESOLVER_RESPONSE:
|
||||
{
|
||||
s.WriteString("R")
|
||||
}
|
||||
case Message_AUTH_QUERY,
|
||||
Message_AUTH_RESPONSE:
|
||||
{
|
||||
s.WriteString("A")
|
||||
}
|
||||
case Message_FORWARDER_QUERY,
|
||||
Message_FORWARDER_RESPONSE:
|
||||
{
|
||||
s.WriteString("F")
|
||||
}
|
||||
case Message_STUB_QUERY,
|
||||
Message_STUB_RESPONSE:
|
||||
{
|
||||
s.WriteString("S")
|
||||
}
|
||||
case Message_TOOL_QUERY,
|
||||
Message_TOOL_RESPONSE:
|
||||
{
|
||||
s.WriteString("T")
|
||||
}
|
||||
}
|
||||
|
||||
if isQuery {
|
||||
s.WriteString("Q ")
|
||||
} else {
|
||||
s.WriteString("R ")
|
||||
}
|
||||
|
||||
switch *m.Type {
|
||||
case Message_CLIENT_QUERY,
|
||||
Message_CLIENT_RESPONSE,
|
||||
Message_AUTH_QUERY,
|
||||
Message_AUTH_RESPONSE:
|
||||
printQueryAddress = true
|
||||
}
|
||||
|
||||
if printQueryAddress {
|
||||
textConvertIP(s, m.QueryAddress)
|
||||
} else {
|
||||
textConvertIP(s, m.ResponseAddress)
|
||||
}
|
||||
s.WriteString(" ")
|
||||
|
||||
if m.SocketProtocol != nil {
|
||||
s.WriteString(m.SocketProtocol.String())
|
||||
}
|
||||
s.WriteString(" ")
|
||||
|
||||
var err error
|
||||
msg := new(dns.Msg)
|
||||
if isQuery {
|
||||
s.WriteString(strconv.Itoa(len(m.QueryMessage)))
|
||||
s.WriteString("b ")
|
||||
err = msg.Unpack(m.QueryMessage)
|
||||
} else {
|
||||
s.WriteString(strconv.Itoa(len(m.ResponseMessage)))
|
||||
s.WriteString("b ")
|
||||
err = msg.Unpack(m.ResponseMessage)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
s.WriteString("X ")
|
||||
} else {
|
||||
s.WriteString("\"" + msg.Question[0].Name + "\" ")
|
||||
s.WriteString(dns.Class(msg.Question[0].Qclass).String() + " ")
|
||||
s.WriteString(dns.Type(msg.Question[0].Qtype).String())
|
||||
}
|
||||
|
||||
s.WriteString("\n")
|
||||
}
|
||||
|
||||
func TextFormat(dt *Dnstap) (out []byte, ok bool) {
|
||||
var s bytes.Buffer
|
||||
|
||||
if *dt.Type == Dnstap_MESSAGE {
|
||||
textConvertMessage(dt.Message, &s)
|
||||
return s.Bytes(), true
|
||||
}
|
||||
|
||||
return nil, false
|
||||
}
|
15
vendor/github.com/dnstap/golang-dnstap/README
generated
vendored
Normal file
15
vendor/github.com/dnstap/golang-dnstap/README
generated
vendored
Normal file
|
@ -0,0 +1,15 @@
|
|||
dnstap: flexible, structured event replication format for DNS servers
|
||||
---------------------------------------------------------------------
|
||||
|
||||
dnstap implements an encoding format for DNS server events. It uses a
|
||||
lightweight framing on top of event payloads encoded using Protocol Buffers and
|
||||
is transport neutral.
|
||||
|
||||
dnstap can represent internal state inside a DNS server that is difficult to
|
||||
obtain using techniques based on traditional packet capture or unstructured
|
||||
textual format logging.
|
||||
|
||||
This repository contains a command-line tool named "dnstap" developed in the
|
||||
Go programming language. It can be installed with the following command:
|
||||
|
||||
go get -u github.com/dnstap/golang-dnstap/dnstap
|
86
vendor/github.com/dnstap/golang-dnstap/TextOutput.go
generated
vendored
Normal file
86
vendor/github.com/dnstap/golang-dnstap/TextOutput.go
generated
vendored
Normal file
|
@ -0,0 +1,86 @@
|
|||
/*
|
||||
* Copyright (c) 2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package dnstap
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/golang/protobuf/proto"
|
||||
)
|
||||
|
||||
type TextFormatFunc func(*Dnstap) ([]byte, bool)
|
||||
|
||||
type TextOutput struct {
|
||||
format TextFormatFunc
|
||||
outputChannel chan []byte
|
||||
wait chan bool
|
||||
writer *bufio.Writer
|
||||
}
|
||||
|
||||
func NewTextOutput(writer io.Writer, format TextFormatFunc) (o *TextOutput) {
|
||||
o = new(TextOutput)
|
||||
o.format = format
|
||||
o.outputChannel = make(chan []byte, outputChannelSize)
|
||||
o.writer = bufio.NewWriter(writer)
|
||||
o.wait = make(chan bool)
|
||||
return
|
||||
}
|
||||
|
||||
func NewTextOutputFromFilename(fname string, format TextFormatFunc) (o *TextOutput, err error) {
|
||||
if fname == "" || fname == "-" {
|
||||
return NewTextOutput(os.Stdout, format), nil
|
||||
}
|
||||
writer, err := os.Create(fname)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return NewTextOutput(writer, format), nil
|
||||
}
|
||||
|
||||
func (o *TextOutput) GetOutputChannel() chan []byte {
|
||||
return o.outputChannel
|
||||
}
|
||||
|
||||
func (o *TextOutput) RunOutputLoop() {
|
||||
dt := &Dnstap{}
|
||||
for frame := range o.outputChannel {
|
||||
if err := proto.Unmarshal(frame, dt); err != nil {
|
||||
log.Fatalf("dnstap.TextOutput: proto.Unmarshal() failed: %s\n", err)
|
||||
break
|
||||
}
|
||||
buf, ok := o.format(dt)
|
||||
if !ok {
|
||||
log.Fatalf("dnstap.TextOutput: text format function failed\n")
|
||||
break
|
||||
}
|
||||
if _, err := o.writer.Write(buf); err != nil {
|
||||
log.Fatalf("dnstap.TextOutput: write failed: %s\n", err)
|
||||
break
|
||||
}
|
||||
o.writer.Flush()
|
||||
}
|
||||
close(o.wait)
|
||||
}
|
||||
|
||||
func (o *TextOutput) Close() {
|
||||
close(o.outputChannel)
|
||||
<-o.wait
|
||||
o.writer.Flush()
|
||||
}
|
116
vendor/github.com/dnstap/golang-dnstap/YamlFormat.go
generated
vendored
Normal file
116
vendor/github.com/dnstap/golang-dnstap/YamlFormat.go
generated
vendored
Normal file
|
@ -0,0 +1,116 @@
|
|||
/*
|
||||
* Copyright (c) 2013-2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package dnstap
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
const yamlTimeFormat = "2006-01-02 15:04:05.999999999"
|
||||
|
||||
func yamlConvertMessage(m *Message, s *bytes.Buffer) {
|
||||
s.WriteString(fmt.Sprint(" type: ", m.Type, "\n"))
|
||||
|
||||
if m.QueryTimeSec != nil && m.QueryTimeNsec != nil {
|
||||
t := time.Unix(int64(*m.QueryTimeSec), int64(*m.QueryTimeNsec)).UTC()
|
||||
s.WriteString(fmt.Sprint(" query_time: !!timestamp ", t.Format(yamlTimeFormat), "\n"))
|
||||
}
|
||||
|
||||
if m.ResponseTimeSec != nil && m.ResponseTimeNsec != nil {
|
||||
t := time.Unix(int64(*m.ResponseTimeSec), int64(*m.ResponseTimeNsec)).UTC()
|
||||
s.WriteString(fmt.Sprint(" response_time: !!timestamp ", t.Format(yamlTimeFormat), "\n"))
|
||||
}
|
||||
|
||||
if m.SocketFamily != nil {
|
||||
s.WriteString(fmt.Sprint(" socket_family: ", m.SocketFamily, "\n"))
|
||||
}
|
||||
|
||||
if m.SocketProtocol != nil {
|
||||
s.WriteString(fmt.Sprint(" socket_protocol: ", m.SocketProtocol, "\n"))
|
||||
}
|
||||
|
||||
if m.QueryAddress != nil {
|
||||
s.WriteString(fmt.Sprint(" query_address: ", net.IP(m.QueryAddress), "\n"))
|
||||
}
|
||||
|
||||
if m.ResponseAddress != nil {
|
||||
s.WriteString(fmt.Sprint(" response_address: ", net.IP(m.ResponseAddress), "\n"))
|
||||
}
|
||||
|
||||
if m.QueryPort != nil {
|
||||
s.WriteString(fmt.Sprint(" query_port: ", *m.QueryPort, "\n"))
|
||||
}
|
||||
|
||||
if m.ResponsePort != nil {
|
||||
s.WriteString(fmt.Sprint(" response_port: ", *m.ResponsePort, "\n"))
|
||||
}
|
||||
|
||||
if m.QueryZone != nil {
|
||||
name, _, err := dns.UnpackDomainName(m.QueryZone, 0)
|
||||
if err != nil {
|
||||
s.WriteString(" # query_zone: parse failed\n")
|
||||
} else {
|
||||
s.WriteString(fmt.Sprint(" query_zone: ", strconv.Quote(name), "\n"))
|
||||
}
|
||||
}
|
||||
|
||||
if m.QueryMessage != nil {
|
||||
msg := new(dns.Msg)
|
||||
err := msg.Unpack(m.QueryMessage)
|
||||
if err != nil {
|
||||
s.WriteString(" # query_message: parse failed\n")
|
||||
} else {
|
||||
s.WriteString(" query_message: |\n")
|
||||
s.WriteString(" " + strings.Replace(strings.TrimSpace(msg.String()), "\n", "\n ", -1) + "\n")
|
||||
}
|
||||
}
|
||||
if m.ResponseMessage != nil {
|
||||
msg := new(dns.Msg)
|
||||
err := msg.Unpack(m.ResponseMessage)
|
||||
if err != nil {
|
||||
s.WriteString(fmt.Sprint(" # response_message: parse failed: ", err, "\n"))
|
||||
} else {
|
||||
s.WriteString(" response_message: |\n")
|
||||
s.WriteString(" " + strings.Replace(strings.TrimSpace(msg.String()), "\n", "\n ", -1) + "\n")
|
||||
}
|
||||
}
|
||||
s.WriteString("---\n")
|
||||
}
|
||||
|
||||
func YamlFormat(dt *Dnstap) (out []byte, ok bool) {
|
||||
var s bytes.Buffer
|
||||
|
||||
s.WriteString(fmt.Sprint("type: ", dt.Type, "\n"))
|
||||
if dt.Identity != nil {
|
||||
s.WriteString(fmt.Sprint("identity: ", strconv.Quote(string(dt.Identity)), "\n"))
|
||||
}
|
||||
if dt.Version != nil {
|
||||
s.WriteString(fmt.Sprint("version: ", strconv.Quote(string(dt.Version)), "\n"))
|
||||
}
|
||||
if *dt.Type == Dnstap_MESSAGE {
|
||||
s.WriteString("message:\n")
|
||||
yamlConvertMessage(dt.Message, &s)
|
||||
}
|
||||
return s.Bytes(), true
|
||||
}
|
32
vendor/github.com/dnstap/golang-dnstap/dnstap.go
generated
vendored
Normal file
32
vendor/github.com/dnstap/golang-dnstap/dnstap.go
generated
vendored
Normal file
|
@ -0,0 +1,32 @@
|
|||
/*
|
||||
* Copyright (c) 2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package dnstap
|
||||
|
||||
const outputChannelSize = 32
|
||||
|
||||
var FSContentType = []byte("protobuf:dnstap.Dnstap")
|
||||
|
||||
type Input interface {
|
||||
ReadInto(chan []byte)
|
||||
Wait()
|
||||
}
|
||||
|
||||
type Output interface {
|
||||
GetOutputChannel() chan []byte
|
||||
RunOutputLoop()
|
||||
Close()
|
||||
}
|
448
vendor/github.com/dnstap/golang-dnstap/dnstap.pb.go
generated
vendored
Normal file
448
vendor/github.com/dnstap/golang-dnstap/dnstap.pb.go
generated
vendored
Normal file
|
@ -0,0 +1,448 @@
|
|||
// Code generated by protoc-gen-go.
|
||||
// source: dnstap.proto
|
||||
// DO NOT EDIT!
|
||||
|
||||
/*
|
||||
Package dnstap is a generated protocol buffer package.
|
||||
|
||||
It is generated from these files:
|
||||
dnstap.proto
|
||||
|
||||
It has these top-level messages:
|
||||
Dnstap
|
||||
Message
|
||||
*/
|
||||
package dnstap
|
||||
|
||||
import proto "github.com/golang/protobuf/proto"
|
||||
import json "encoding/json"
|
||||
import math "math"
|
||||
|
||||
// Reference proto, json, and math imports to suppress error if they are not otherwise used.
|
||||
var _ = proto.Marshal
|
||||
var _ = &json.SyntaxError{}
|
||||
var _ = math.Inf
|
||||
|
||||
// SocketFamily: the network protocol family of a socket. This specifies how
|
||||
// to interpret "network address" fields.
|
||||
type SocketFamily int32
|
||||
|
||||
const (
|
||||
SocketFamily_INET SocketFamily = 1
|
||||
SocketFamily_INET6 SocketFamily = 2
|
||||
)
|
||||
|
||||
var SocketFamily_name = map[int32]string{
|
||||
1: "INET",
|
||||
2: "INET6",
|
||||
}
|
||||
var SocketFamily_value = map[string]int32{
|
||||
"INET": 1,
|
||||
"INET6": 2,
|
||||
}
|
||||
|
||||
func (x SocketFamily) Enum() *SocketFamily {
|
||||
p := new(SocketFamily)
|
||||
*p = x
|
||||
return p
|
||||
}
|
||||
func (x SocketFamily) String() string {
|
||||
return proto.EnumName(SocketFamily_name, int32(x))
|
||||
}
|
||||
func (x *SocketFamily) UnmarshalJSON(data []byte) error {
|
||||
value, err := proto.UnmarshalJSONEnum(SocketFamily_value, data, "SocketFamily")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*x = SocketFamily(value)
|
||||
return nil
|
||||
}
|
||||
|
||||
// SocketProtocol: the transport protocol of a socket. This specifies how to
|
||||
// interpret "transport port" fields.
|
||||
type SocketProtocol int32
|
||||
|
||||
const (
|
||||
SocketProtocol_UDP SocketProtocol = 1
|
||||
SocketProtocol_TCP SocketProtocol = 2
|
||||
)
|
||||
|
||||
var SocketProtocol_name = map[int32]string{
|
||||
1: "UDP",
|
||||
2: "TCP",
|
||||
}
|
||||
var SocketProtocol_value = map[string]int32{
|
||||
"UDP": 1,
|
||||
"TCP": 2,
|
||||
}
|
||||
|
||||
func (x SocketProtocol) Enum() *SocketProtocol {
|
||||
p := new(SocketProtocol)
|
||||
*p = x
|
||||
return p
|
||||
}
|
||||
func (x SocketProtocol) String() string {
|
||||
return proto.EnumName(SocketProtocol_name, int32(x))
|
||||
}
|
||||
func (x *SocketProtocol) UnmarshalJSON(data []byte) error {
|
||||
value, err := proto.UnmarshalJSONEnum(SocketProtocol_value, data, "SocketProtocol")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*x = SocketProtocol(value)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Identifies which field below is filled in.
|
||||
type Dnstap_Type int32
|
||||
|
||||
const (
|
||||
Dnstap_MESSAGE Dnstap_Type = 1
|
||||
)
|
||||
|
||||
var Dnstap_Type_name = map[int32]string{
|
||||
1: "MESSAGE",
|
||||
}
|
||||
var Dnstap_Type_value = map[string]int32{
|
||||
"MESSAGE": 1,
|
||||
}
|
||||
|
||||
func (x Dnstap_Type) Enum() *Dnstap_Type {
|
||||
p := new(Dnstap_Type)
|
||||
*p = x
|
||||
return p
|
||||
}
|
||||
func (x Dnstap_Type) String() string {
|
||||
return proto.EnumName(Dnstap_Type_name, int32(x))
|
||||
}
|
||||
func (x *Dnstap_Type) UnmarshalJSON(data []byte) error {
|
||||
value, err := proto.UnmarshalJSONEnum(Dnstap_Type_value, data, "Dnstap_Type")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*x = Dnstap_Type(value)
|
||||
return nil
|
||||
}
|
||||
|
||||
type Message_Type int32
|
||||
|
||||
const (
|
||||
// AUTH_QUERY is a DNS query message received from a resolver by an
|
||||
// authoritative name server, from the perspective of the authorative
|
||||
// name server.
|
||||
Message_AUTH_QUERY Message_Type = 1
|
||||
// AUTH_RESPONSE is a DNS response message sent from an authoritative
|
||||
// name server to a resolver, from the perspective of the authoritative
|
||||
// name server.
|
||||
Message_AUTH_RESPONSE Message_Type = 2
|
||||
// RESOLVER_QUERY is a DNS query message sent from a resolver to an
|
||||
// authoritative name server, from the perspective of the resolver.
|
||||
// Resolvers typically clear the RD (recursion desired) bit when
|
||||
// sending queries.
|
||||
Message_RESOLVER_QUERY Message_Type = 3
|
||||
// RESOLVER_RESPONSE is a DNS response message received from an
|
||||
// authoritative name server by a resolver, from the perspective of
|
||||
// the resolver.
|
||||
Message_RESOLVER_RESPONSE Message_Type = 4
|
||||
// CLIENT_QUERY is a DNS query message sent from a client to a DNS
|
||||
// server which is expected to perform further recursion, from the
|
||||
// perspective of the DNS server. The client may be a stub resolver or
|
||||
// forwarder or some other type of software which typically sets the RD
|
||||
// (recursion desired) bit when querying the DNS server. The DNS server
|
||||
// may be a simple forwarding proxy or it may be a full recursive
|
||||
// resolver.
|
||||
Message_CLIENT_QUERY Message_Type = 5
|
||||
// CLIENT_RESPONSE is a DNS response message sent from a DNS server to
|
||||
// a client, from the perspective of the DNS server. The DNS server
|
||||
// typically sets the RA (recursion available) bit when responding.
|
||||
Message_CLIENT_RESPONSE Message_Type = 6
|
||||
// FORWARDER_QUERY is a DNS query message sent from a downstream DNS
|
||||
// server to an upstream DNS server which is expected to perform
|
||||
// further recursion, from the perspective of the downstream DNS
|
||||
// server.
|
||||
Message_FORWARDER_QUERY Message_Type = 7
|
||||
// FORWARDER_RESPONSE is a DNS response message sent from an upstream
|
||||
// DNS server performing recursion to a downstream DNS server, from the
|
||||
// perspective of the downstream DNS server.
|
||||
Message_FORWARDER_RESPONSE Message_Type = 8
|
||||
// STUB_QUERY is a DNS query message sent from a stub resolver to a DNS
|
||||
// server, from the perspective of the stub resolver.
|
||||
Message_STUB_QUERY Message_Type = 9
|
||||
// STUB_RESPONSE is a DNS response message sent from a DNS server to a
|
||||
// stub resolver, from the perspective of the stub resolver.
|
||||
Message_STUB_RESPONSE Message_Type = 10
|
||||
// TOOL_QUERY is a DNS query message sent from a DNS software tool to a
|
||||
// DNS server, from the perspective of the tool.
|
||||
Message_TOOL_QUERY Message_Type = 11
|
||||
// TOOL_RESPONSE is a DNS response message received by a DNS software
|
||||
// tool from a DNS server, from the perspective of the tool.
|
||||
Message_TOOL_RESPONSE Message_Type = 12
|
||||
)
|
||||
|
||||
var Message_Type_name = map[int32]string{
|
||||
1: "AUTH_QUERY",
|
||||
2: "AUTH_RESPONSE",
|
||||
3: "RESOLVER_QUERY",
|
||||
4: "RESOLVER_RESPONSE",
|
||||
5: "CLIENT_QUERY",
|
||||
6: "CLIENT_RESPONSE",
|
||||
7: "FORWARDER_QUERY",
|
||||
8: "FORWARDER_RESPONSE",
|
||||
9: "STUB_QUERY",
|
||||
10: "STUB_RESPONSE",
|
||||
11: "TOOL_QUERY",
|
||||
12: "TOOL_RESPONSE",
|
||||
}
|
||||
var Message_Type_value = map[string]int32{
|
||||
"AUTH_QUERY": 1,
|
||||
"AUTH_RESPONSE": 2,
|
||||
"RESOLVER_QUERY": 3,
|
||||
"RESOLVER_RESPONSE": 4,
|
||||
"CLIENT_QUERY": 5,
|
||||
"CLIENT_RESPONSE": 6,
|
||||
"FORWARDER_QUERY": 7,
|
||||
"FORWARDER_RESPONSE": 8,
|
||||
"STUB_QUERY": 9,
|
||||
"STUB_RESPONSE": 10,
|
||||
"TOOL_QUERY": 11,
|
||||
"TOOL_RESPONSE": 12,
|
||||
}
|
||||
|
||||
func (x Message_Type) Enum() *Message_Type {
|
||||
p := new(Message_Type)
|
||||
*p = x
|
||||
return p
|
||||
}
|
||||
func (x Message_Type) String() string {
|
||||
return proto.EnumName(Message_Type_name, int32(x))
|
||||
}
|
||||
func (x *Message_Type) UnmarshalJSON(data []byte) error {
|
||||
value, err := proto.UnmarshalJSONEnum(Message_Type_value, data, "Message_Type")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*x = Message_Type(value)
|
||||
return nil
|
||||
}
|
||||
|
||||
// "Dnstap": this is the top-level dnstap type, which is a "union" type that
|
||||
// contains other kinds of dnstap payloads, although currently only one type
|
||||
// of dnstap payload is defined.
|
||||
// See: https://developers.google.com/protocol-buffers/docs/techniques#union
|
||||
type Dnstap struct {
|
||||
// DNS server identity.
|
||||
// If enabled, this is the identity string of the DNS server which generated
|
||||
// this message. Typically this would be the same string as returned by an
|
||||
// "NSID" (RFC 5001) query.
|
||||
Identity []byte `protobuf:"bytes,1,opt,name=identity" json:"identity,omitempty"`
|
||||
// DNS server version.
|
||||
// If enabled, this is the version string of the DNS server which generated
|
||||
// this message. Typically this would be the same string as returned by a
|
||||
// "version.bind" query.
|
||||
Version []byte `protobuf:"bytes,2,opt,name=version" json:"version,omitempty"`
|
||||
// Extra data for this payload.
|
||||
// This field can be used for adding an arbitrary byte-string annotation to
|
||||
// the payload. No encoding or interpretation is applied or enforced.
|
||||
Extra []byte `protobuf:"bytes,3,opt,name=extra" json:"extra,omitempty"`
|
||||
Type *Dnstap_Type `protobuf:"varint,15,req,name=type,enum=dnstap.Dnstap_Type" json:"type,omitempty"`
|
||||
// One of the following will be filled in.
|
||||
Message *Message `protobuf:"bytes,14,opt,name=message" json:"message,omitempty"`
|
||||
XXX_unrecognized []byte `json:"-"`
|
||||
}
|
||||
|
||||
func (m *Dnstap) Reset() { *m = Dnstap{} }
|
||||
func (m *Dnstap) String() string { return proto.CompactTextString(m) }
|
||||
func (*Dnstap) ProtoMessage() {}
|
||||
|
||||
func (m *Dnstap) GetIdentity() []byte {
|
||||
if m != nil {
|
||||
return m.Identity
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Dnstap) GetVersion() []byte {
|
||||
if m != nil {
|
||||
return m.Version
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Dnstap) GetExtra() []byte {
|
||||
if m != nil {
|
||||
return m.Extra
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Dnstap) GetType() Dnstap_Type {
|
||||
if m != nil && m.Type != nil {
|
||||
return *m.Type
|
||||
}
|
||||
return Dnstap_MESSAGE
|
||||
}
|
||||
|
||||
func (m *Dnstap) GetMessage() *Message {
|
||||
if m != nil {
|
||||
return m.Message
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Message: a wire-format (RFC 1035 section 4) DNS message and associated
|
||||
// metadata. Applications generating "Message" payloads should follow
|
||||
// certain requirements based on the MessageType, see below.
|
||||
type Message struct {
|
||||
// One of the Type values described above.
|
||||
Type *Message_Type `protobuf:"varint,1,req,name=type,enum=dnstap.Message_Type" json:"type,omitempty"`
|
||||
// One of the SocketFamily values described above.
|
||||
SocketFamily *SocketFamily `protobuf:"varint,2,opt,name=socket_family,enum=dnstap.SocketFamily" json:"socket_family,omitempty"`
|
||||
// One of the SocketProtocol values described above.
|
||||
SocketProtocol *SocketProtocol `protobuf:"varint,3,opt,name=socket_protocol,enum=dnstap.SocketProtocol" json:"socket_protocol,omitempty"`
|
||||
// The network address of the message initiator.
|
||||
// For SocketFamily INET, this field is 4 octets (IPv4 address).
|
||||
// For SocketFamily INET6, this field is 16 octets (IPv6 address).
|
||||
QueryAddress []byte `protobuf:"bytes,4,opt,name=query_address" json:"query_address,omitempty"`
|
||||
// The network address of the message responder.
|
||||
// For SocketFamily INET, this field is 4 octets (IPv4 address).
|
||||
// For SocketFamily INET6, this field is 16 octets (IPv6 address).
|
||||
ResponseAddress []byte `protobuf:"bytes,5,opt,name=response_address" json:"response_address,omitempty"`
|
||||
// The transport port of the message initiator.
|
||||
// This is a 16-bit UDP or TCP port number, depending on SocketProtocol.
|
||||
QueryPort *uint32 `protobuf:"varint,6,opt,name=query_port" json:"query_port,omitempty"`
|
||||
// The transport port of the message responder.
|
||||
// This is a 16-bit UDP or TCP port number, depending on SocketProtocol.
|
||||
ResponsePort *uint32 `protobuf:"varint,7,opt,name=response_port" json:"response_port,omitempty"`
|
||||
// The time at which the DNS query message was sent or received, depending
|
||||
// on whether this is an AUTH_QUERY, RESOLVER_QUERY, or CLIENT_QUERY.
|
||||
// This is the number of seconds since the UNIX epoch.
|
||||
QueryTimeSec *uint64 `protobuf:"varint,8,opt,name=query_time_sec" json:"query_time_sec,omitempty"`
|
||||
// The time at which the DNS query message was sent or received.
|
||||
// This is the seconds fraction, expressed as a count of nanoseconds.
|
||||
QueryTimeNsec *uint32 `protobuf:"fixed32,9,opt,name=query_time_nsec" json:"query_time_nsec,omitempty"`
|
||||
// The initiator's original wire-format DNS query message, verbatim.
|
||||
QueryMessage []byte `protobuf:"bytes,10,opt,name=query_message" json:"query_message,omitempty"`
|
||||
// The "zone" or "bailiwick" pertaining to the DNS query message.
|
||||
// This is a wire-format DNS domain name.
|
||||
QueryZone []byte `protobuf:"bytes,11,opt,name=query_zone" json:"query_zone,omitempty"`
|
||||
// The time at which the DNS response message was sent or received,
|
||||
// depending on whether this is an AUTH_RESPONSE, RESOLVER_RESPONSE, or
|
||||
// CLIENT_RESPONSE.
|
||||
// This is the number of seconds since the UNIX epoch.
|
||||
ResponseTimeSec *uint64 `protobuf:"varint,12,opt,name=response_time_sec" json:"response_time_sec,omitempty"`
|
||||
// The time at which the DNS response message was sent or received.
|
||||
// This is the seconds fraction, expressed as a count of nanoseconds.
|
||||
ResponseTimeNsec *uint32 `protobuf:"fixed32,13,opt,name=response_time_nsec" json:"response_time_nsec,omitempty"`
|
||||
// The responder's original wire-format DNS response message, verbatim.
|
||||
ResponseMessage []byte `protobuf:"bytes,14,opt,name=response_message" json:"response_message,omitempty"`
|
||||
XXX_unrecognized []byte `json:"-"`
|
||||
}
|
||||
|
||||
func (m *Message) Reset() { *m = Message{} }
|
||||
func (m *Message) String() string { return proto.CompactTextString(m) }
|
||||
func (*Message) ProtoMessage() {}
|
||||
|
||||
func (m *Message) GetType() Message_Type {
|
||||
if m != nil && m.Type != nil {
|
||||
return *m.Type
|
||||
}
|
||||
return Message_AUTH_QUERY
|
||||
}
|
||||
|
||||
func (m *Message) GetSocketFamily() SocketFamily {
|
||||
if m != nil && m.SocketFamily != nil {
|
||||
return *m.SocketFamily
|
||||
}
|
||||
return SocketFamily_INET
|
||||
}
|
||||
|
||||
func (m *Message) GetSocketProtocol() SocketProtocol {
|
||||
if m != nil && m.SocketProtocol != nil {
|
||||
return *m.SocketProtocol
|
||||
}
|
||||
return SocketProtocol_UDP
|
||||
}
|
||||
|
||||
func (m *Message) GetQueryAddress() []byte {
|
||||
if m != nil {
|
||||
return m.QueryAddress
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Message) GetResponseAddress() []byte {
|
||||
if m != nil {
|
||||
return m.ResponseAddress
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Message) GetQueryPort() uint32 {
|
||||
if m != nil && m.QueryPort != nil {
|
||||
return *m.QueryPort
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (m *Message) GetResponsePort() uint32 {
|
||||
if m != nil && m.ResponsePort != nil {
|
||||
return *m.ResponsePort
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (m *Message) GetQueryTimeSec() uint64 {
|
||||
if m != nil && m.QueryTimeSec != nil {
|
||||
return *m.QueryTimeSec
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (m *Message) GetQueryTimeNsec() uint32 {
|
||||
if m != nil && m.QueryTimeNsec != nil {
|
||||
return *m.QueryTimeNsec
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (m *Message) GetQueryMessage() []byte {
|
||||
if m != nil {
|
||||
return m.QueryMessage
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Message) GetQueryZone() []byte {
|
||||
if m != nil {
|
||||
return m.QueryZone
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Message) GetResponseTimeSec() uint64 {
|
||||
if m != nil && m.ResponseTimeSec != nil {
|
||||
return *m.ResponseTimeSec
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (m *Message) GetResponseTimeNsec() uint32 {
|
||||
if m != nil && m.ResponseTimeNsec != nil {
|
||||
return *m.ResponseTimeNsec
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (m *Message) GetResponseMessage() []byte {
|
||||
if m != nil {
|
||||
return m.ResponseMessage
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
proto.RegisterEnum("dnstap.SocketFamily", SocketFamily_name, SocketFamily_value)
|
||||
proto.RegisterEnum("dnstap.SocketProtocol", SocketProtocol_name, SocketProtocol_value)
|
||||
proto.RegisterEnum("dnstap.Dnstap_Type", Dnstap_Type_name, Dnstap_Type_value)
|
||||
proto.RegisterEnum("dnstap.Message_Type", Message_Type_name, Message_Type_value)
|
||||
}
|
10
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/.gitignore
generated
vendored
Normal file
10
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/.gitignore
generated
vendored
Normal file
|
@ -0,0 +1,10 @@
|
|||
.deps/
|
||||
.dirstamp
|
||||
.libs/
|
||||
*.pb-c.c
|
||||
*.pb-c.h
|
||||
*.pb.cc
|
||||
*.pb.h
|
||||
*.pb.go
|
||||
*_pb2.py
|
||||
*_pb2.pyc
|
121
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/LICENSE
generated
vendored
Normal file
121
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/LICENSE
generated
vendored
Normal file
|
@ -0,0 +1,121 @@
|
|||
Creative Commons Legal Code
|
||||
|
||||
CC0 1.0 Universal
|
||||
|
||||
CREATIVE COMMONS CORPORATION IS NOT A LAW FIRM AND DOES NOT PROVIDE
|
||||
LEGAL SERVICES. DISTRIBUTION OF THIS DOCUMENT DOES NOT CREATE AN
|
||||
ATTORNEY-CLIENT RELATIONSHIP. CREATIVE COMMONS PROVIDES THIS
|
||||
INFORMATION ON AN "AS-IS" BASIS. CREATIVE COMMONS MAKES NO WARRANTIES
|
||||
REGARDING THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS
|
||||
PROVIDED HEREUNDER, AND DISCLAIMS LIABILITY FOR DAMAGES RESULTING FROM
|
||||
THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS PROVIDED
|
||||
HEREUNDER.
|
||||
|
||||
Statement of Purpose
|
||||
|
||||
The laws of most jurisdictions throughout the world automatically confer
|
||||
exclusive Copyright and Related Rights (defined below) upon the creator
|
||||
and subsequent owner(s) (each and all, an "owner") of an original work of
|
||||
authorship and/or a database (each, a "Work").
|
||||
|
||||
Certain owners wish to permanently relinquish those rights to a Work for
|
||||
the purpose of contributing to a commons of creative, cultural and
|
||||
scientific works ("Commons") that the public can reliably and without fear
|
||||
of later claims of infringement build upon, modify, incorporate in other
|
||||
works, reuse and redistribute as freely as possible in any form whatsoever
|
||||
and for any purposes, including without limitation commercial purposes.
|
||||
These owners may contribute to the Commons to promote the ideal of a free
|
||||
culture and the further production of creative, cultural and scientific
|
||||
works, or to gain reputation or greater distribution for their Work in
|
||||
part through the use and efforts of others.
|
||||
|
||||
For these and/or other purposes and motivations, and without any
|
||||
expectation of additional consideration or compensation, the person
|
||||
associating CC0 with a Work (the "Affirmer"), to the extent that he or she
|
||||
is an owner of Copyright and Related Rights in the Work, voluntarily
|
||||
elects to apply CC0 to the Work and publicly distribute the Work under its
|
||||
terms, with knowledge of his or her Copyright and Related Rights in the
|
||||
Work and the meaning and intended legal effect of CC0 on those rights.
|
||||
|
||||
1. Copyright and Related Rights. A Work made available under CC0 may be
|
||||
protected by copyright and related or neighboring rights ("Copyright and
|
||||
Related Rights"). Copyright and Related Rights include, but are not
|
||||
limited to, the following:
|
||||
|
||||
i. the right to reproduce, adapt, distribute, perform, display,
|
||||
communicate, and translate a Work;
|
||||
ii. moral rights retained by the original author(s) and/or performer(s);
|
||||
iii. publicity and privacy rights pertaining to a person's image or
|
||||
likeness depicted in a Work;
|
||||
iv. rights protecting against unfair competition in regards to a Work,
|
||||
subject to the limitations in paragraph 4(a), below;
|
||||
v. rights protecting the extraction, dissemination, use and reuse of data
|
||||
in a Work;
|
||||
vi. database rights (such as those arising under Directive 96/9/EC of the
|
||||
European Parliament and of the Council of 11 March 1996 on the legal
|
||||
protection of databases, and under any national implementation
|
||||
thereof, including any amended or successor version of such
|
||||
directive); and
|
||||
vii. other similar, equivalent or corresponding rights throughout the
|
||||
world based on applicable law or treaty, and any national
|
||||
implementations thereof.
|
||||
|
||||
2. Waiver. To the greatest extent permitted by, but not in contravention
|
||||
of, applicable law, Affirmer hereby overtly, fully, permanently,
|
||||
irrevocably and unconditionally waives, abandons, and surrenders all of
|
||||
Affirmer's Copyright and Related Rights and associated claims and causes
|
||||
of action, whether now known or unknown (including existing as well as
|
||||
future claims and causes of action), in the Work (i) in all territories
|
||||
worldwide, (ii) for the maximum duration provided by applicable law or
|
||||
treaty (including future time extensions), (iii) in any current or future
|
||||
medium and for any number of copies, and (iv) for any purpose whatsoever,
|
||||
including without limitation commercial, advertising or promotional
|
||||
purposes (the "Waiver"). Affirmer makes the Waiver for the benefit of each
|
||||
member of the public at large and to the detriment of Affirmer's heirs and
|
||||
successors, fully intending that such Waiver shall not be subject to
|
||||
revocation, rescission, cancellation, termination, or any other legal or
|
||||
equitable action to disrupt the quiet enjoyment of the Work by the public
|
||||
as contemplated by Affirmer's express Statement of Purpose.
|
||||
|
||||
3. Public License Fallback. Should any part of the Waiver for any reason
|
||||
be judged legally invalid or ineffective under applicable law, then the
|
||||
Waiver shall be preserved to the maximum extent permitted taking into
|
||||
account Affirmer's express Statement of Purpose. In addition, to the
|
||||
extent the Waiver is so judged Affirmer hereby grants to each affected
|
||||
person a royalty-free, non transferable, non sublicensable, non exclusive,
|
||||
irrevocable and unconditional license to exercise Affirmer's Copyright and
|
||||
Related Rights in the Work (i) in all territories worldwide, (ii) for the
|
||||
maximum duration provided by applicable law or treaty (including future
|
||||
time extensions), (iii) in any current or future medium and for any number
|
||||
of copies, and (iv) for any purpose whatsoever, including without
|
||||
limitation commercial, advertising or promotional purposes (the
|
||||
"License"). The License shall be deemed effective as of the date CC0 was
|
||||
applied by Affirmer to the Work. Should any part of the License for any
|
||||
reason be judged legally invalid or ineffective under applicable law, such
|
||||
partial invalidity or ineffectiveness shall not invalidate the remainder
|
||||
of the License, and in such case Affirmer hereby affirms that he or she
|
||||
will not (i) exercise any of his or her remaining Copyright and Related
|
||||
Rights in the Work or (ii) assert any associated claims and causes of
|
||||
action with respect to the Work, in either case contrary to Affirmer's
|
||||
express Statement of Purpose.
|
||||
|
||||
4. Limitations and Disclaimers.
|
||||
|
||||
a. No trademark or patent rights held by Affirmer are waived, abandoned,
|
||||
surrendered, licensed or otherwise affected by this document.
|
||||
b. Affirmer offers the Work as-is and makes no representations or
|
||||
warranties of any kind concerning the Work, express, implied,
|
||||
statutory or otherwise, including without limitation warranties of
|
||||
title, merchantability, fitness for a particular purpose, non
|
||||
infringement, or the absence of latent or other defects, accuracy, or
|
||||
the present or absence of errors, whether or not discoverable, all to
|
||||
the greatest extent permissible under applicable law.
|
||||
c. Affirmer disclaims responsibility for clearing rights of other persons
|
||||
that may apply to the Work or any use thereof, including without
|
||||
limitation any person's Copyright and Related Rights in the Work.
|
||||
Further, Affirmer disclaims responsibility for obtaining any necessary
|
||||
consents, permissions or other rights required for any use of the
|
||||
Work.
|
||||
d. Affirmer understands and acknowledges that Creative Commons is not a
|
||||
party to this document and has no duty or obligation with respect to
|
||||
this CC0 or use of the Work.
|
5
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/README
generated
vendored
Normal file
5
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/README
generated
vendored
Normal file
|
@ -0,0 +1,5 @@
|
|||
dnstap: flexible, structured event replication format for DNS software
|
||||
----------------------------------------------------------------------
|
||||
|
||||
This directory contains only the protobuf schemas for dnstap, and is the root of
|
||||
a repository named "dnstap.pb".
|
268
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/dnstap.proto
generated
vendored
Normal file
268
vendor/github.com/dnstap/golang-dnstap/dnstap.pb/dnstap.proto
generated
vendored
Normal file
|
@ -0,0 +1,268 @@
|
|||
// dnstap: flexible, structured event replication format for DNS software
|
||||
//
|
||||
// This file contains the protobuf schemas for the "dnstap" structured event
|
||||
// replication format for DNS software.
|
||||
|
||||
// Written in 2013-2014 by Farsight Security, Inc.
|
||||
//
|
||||
// To the extent possible under law, the author(s) have dedicated all
|
||||
// copyright and related and neighboring rights to this file to the public
|
||||
// domain worldwide. This file is distributed without any warranty.
|
||||
//
|
||||
// You should have received a copy of the CC0 Public Domain Dedication along
|
||||
// with this file. If not, see:
|
||||
//
|
||||
// <http://creativecommons.org/publicdomain/zero/1.0/>.
|
||||
|
||||
package dnstap;
|
||||
|
||||
// "Dnstap": this is the top-level dnstap type, which is a "union" type that
|
||||
// contains other kinds of dnstap payloads, although currently only one type
|
||||
// of dnstap payload is defined.
|
||||
// See: https://developers.google.com/protocol-buffers/docs/techniques#union
|
||||
message Dnstap {
|
||||
// DNS server identity.
|
||||
// If enabled, this is the identity string of the DNS server which generated
|
||||
// this message. Typically this would be the same string as returned by an
|
||||
// "NSID" (RFC 5001) query.
|
||||
optional bytes identity = 1;
|
||||
|
||||
// DNS server version.
|
||||
// If enabled, this is the version string of the DNS server which generated
|
||||
// this message. Typically this would be the same string as returned by a
|
||||
// "version.bind" query.
|
||||
optional bytes version = 2;
|
||||
|
||||
// Extra data for this payload.
|
||||
// This field can be used for adding an arbitrary byte-string annotation to
|
||||
// the payload. No encoding or interpretation is applied or enforced.
|
||||
optional bytes extra = 3;
|
||||
|
||||
// Identifies which field below is filled in.
|
||||
enum Type {
|
||||
MESSAGE = 1;
|
||||
}
|
||||
required Type type = 15;
|
||||
|
||||
// One of the following will be filled in.
|
||||
optional Message message = 14;
|
||||
}
|
||||
|
||||
// SocketFamily: the network protocol family of a socket. This specifies how
|
||||
// to interpret "network address" fields.
|
||||
enum SocketFamily {
|
||||
INET = 1; // IPv4 (RFC 791)
|
||||
INET6 = 2; // IPv6 (RFC 2460)
|
||||
}
|
||||
|
||||
// SocketProtocol: the transport protocol of a socket. This specifies how to
|
||||
// interpret "transport port" fields.
|
||||
enum SocketProtocol {
|
||||
UDP = 1; // User Datagram Protocol (RFC 768)
|
||||
TCP = 2; // Transmission Control Protocol (RFC 793)
|
||||
}
|
||||
|
||||
// Message: a wire-format (RFC 1035 section 4) DNS message and associated
|
||||
// metadata. Applications generating "Message" payloads should follow
|
||||
// certain requirements based on the MessageType, see below.
|
||||
message Message {
|
||||
|
||||
// There are eight types of "Message" defined that correspond to the
|
||||
// four arrows in the following diagram, slightly modified from RFC 1035
|
||||
// section 2:
|
||||
|
||||
// +---------+ +----------+ +--------+
|
||||
// | | query | | query | |
|
||||
// | Stub |-SQ--------CQ->| Recursive|-RQ----AQ->| Auth. |
|
||||
// | Resolver| | Server | | Name |
|
||||
// | |<-SR--------CR-| |<-RR----AR-| Server |
|
||||
// +---------+ response | | response | |
|
||||
// +----------+ +--------+
|
||||
|
||||
// Each arrow has two Type values each, one for each "end" of each arrow,
|
||||
// because these are considered to be distinct events. Each end of each
|
||||
// arrow on the diagram above has been marked with a two-letter Type
|
||||
// mnemonic. Clockwise from upper left, these mnemonic values are:
|
||||
//
|
||||
// SQ: STUB_QUERY
|
||||
// CQ: CLIENT_QUERY
|
||||
// RQ: RESOLVER_QUERY
|
||||
// AQ: AUTH_QUERY
|
||||
// AR: AUTH_RESPONSE
|
||||
// RR: RESOLVER_RESPONSE
|
||||
// CR: CLIENT_RESPONSE
|
||||
// SR: STUB_RESPONSE
|
||||
|
||||
// Two additional types of "Message" have been defined for the
|
||||
// "forwarding" case where an upstream DNS server is responsible for
|
||||
// further recursion. These are not shown on the diagram above, but have
|
||||
// the following mnemonic values:
|
||||
|
||||
// FQ: FORWARDER_QUERY
|
||||
// FR: FORWARDER_RESPONSE
|
||||
|
||||
// The "Message" Type values are defined below.
|
||||
|
||||
enum Type {
|
||||
// AUTH_QUERY is a DNS query message received from a resolver by an
|
||||
// authoritative name server, from the perspective of the authorative
|
||||
// name server.
|
||||
AUTH_QUERY = 1;
|
||||
|
||||
// AUTH_RESPONSE is a DNS response message sent from an authoritative
|
||||
// name server to a resolver, from the perspective of the authoritative
|
||||
// name server.
|
||||
AUTH_RESPONSE = 2;
|
||||
|
||||
// RESOLVER_QUERY is a DNS query message sent from a resolver to an
|
||||
// authoritative name server, from the perspective of the resolver.
|
||||
// Resolvers typically clear the RD (recursion desired) bit when
|
||||
// sending queries.
|
||||
RESOLVER_QUERY = 3;
|
||||
|
||||
// RESOLVER_RESPONSE is a DNS response message received from an
|
||||
// authoritative name server by a resolver, from the perspective of
|
||||
// the resolver.
|
||||
RESOLVER_RESPONSE = 4;
|
||||
|
||||
// CLIENT_QUERY is a DNS query message sent from a client to a DNS
|
||||
// server which is expected to perform further recursion, from the
|
||||
// perspective of the DNS server. The client may be a stub resolver or
|
||||
// forwarder or some other type of software which typically sets the RD
|
||||
// (recursion desired) bit when querying the DNS server. The DNS server
|
||||
// may be a simple forwarding proxy or it may be a full recursive
|
||||
// resolver.
|
||||
CLIENT_QUERY = 5;
|
||||
|
||||
// CLIENT_RESPONSE is a DNS response message sent from a DNS server to
|
||||
// a client, from the perspective of the DNS server. The DNS server
|
||||
// typically sets the RA (recursion available) bit when responding.
|
||||
CLIENT_RESPONSE = 6;
|
||||
|
||||
// FORWARDER_QUERY is a DNS query message sent from a downstream DNS
|
||||
// server to an upstream DNS server which is expected to perform
|
||||
// further recursion, from the perspective of the downstream DNS
|
||||
// server.
|
||||
FORWARDER_QUERY = 7;
|
||||
|
||||
// FORWARDER_RESPONSE is a DNS response message sent from an upstream
|
||||
// DNS server performing recursion to a downstream DNS server, from the
|
||||
// perspective of the downstream DNS server.
|
||||
FORWARDER_RESPONSE = 8;
|
||||
|
||||
// STUB_QUERY is a DNS query message sent from a stub resolver to a DNS
|
||||
// server, from the perspective of the stub resolver.
|
||||
STUB_QUERY = 9;
|
||||
|
||||
// STUB_RESPONSE is a DNS response message sent from a DNS server to a
|
||||
// stub resolver, from the perspective of the stub resolver.
|
||||
STUB_RESPONSE = 10;
|
||||
|
||||
// TOOL_QUERY is a DNS query message sent from a DNS software tool to a
|
||||
// DNS server, from the perspective of the tool.
|
||||
TOOL_QUERY = 11;
|
||||
|
||||
// TOOL_RESPONSE is a DNS response message received by a DNS software
|
||||
// tool from a DNS server, from the perspective of the tool.
|
||||
TOOL_RESPONSE = 12;
|
||||
}
|
||||
|
||||
// One of the Type values described above.
|
||||
required Type type = 1;
|
||||
|
||||
// One of the SocketFamily values described above.
|
||||
optional SocketFamily socket_family = 2;
|
||||
|
||||
// One of the SocketProtocol values described above.
|
||||
optional SocketProtocol socket_protocol = 3;
|
||||
|
||||
// The network address of the message initiator.
|
||||
// For SocketFamily INET, this field is 4 octets (IPv4 address).
|
||||
// For SocketFamily INET6, this field is 16 octets (IPv6 address).
|
||||
optional bytes query_address = 4;
|
||||
|
||||
// The network address of the message responder.
|
||||
// For SocketFamily INET, this field is 4 octets (IPv4 address).
|
||||
// For SocketFamily INET6, this field is 16 octets (IPv6 address).
|
||||
optional bytes response_address = 5;
|
||||
|
||||
// The transport port of the message initiator.
|
||||
// This is a 16-bit UDP or TCP port number, depending on SocketProtocol.
|
||||
optional uint32 query_port = 6;
|
||||
|
||||
// The transport port of the message responder.
|
||||
// This is a 16-bit UDP or TCP port number, depending on SocketProtocol.
|
||||
optional uint32 response_port = 7;
|
||||
|
||||
// The time at which the DNS query message was sent or received, depending
|
||||
// on whether this is an AUTH_QUERY, RESOLVER_QUERY, or CLIENT_QUERY.
|
||||
// This is the number of seconds since the UNIX epoch.
|
||||
optional uint64 query_time_sec = 8;
|
||||
|
||||
// The time at which the DNS query message was sent or received.
|
||||
// This is the seconds fraction, expressed as a count of nanoseconds.
|
||||
optional fixed32 query_time_nsec = 9;
|
||||
|
||||
// The initiator's original wire-format DNS query message, verbatim.
|
||||
optional bytes query_message = 10;
|
||||
|
||||
// The "zone" or "bailiwick" pertaining to the DNS query message.
|
||||
// This is a wire-format DNS domain name.
|
||||
optional bytes query_zone = 11;
|
||||
|
||||
// The time at which the DNS response message was sent or received,
|
||||
// depending on whether this is an AUTH_RESPONSE, RESOLVER_RESPONSE, or
|
||||
// CLIENT_RESPONSE.
|
||||
// This is the number of seconds since the UNIX epoch.
|
||||
optional uint64 response_time_sec = 12;
|
||||
|
||||
// The time at which the DNS response message was sent or received.
|
||||
// This is the seconds fraction, expressed as a count of nanoseconds.
|
||||
optional fixed32 response_time_nsec = 13;
|
||||
|
||||
// The responder's original wire-format DNS response message, verbatim.
|
||||
optional bytes response_message = 14;
|
||||
}
|
||||
|
||||
// All fields except for 'type' in the Message schema are optional.
|
||||
// It is recommended that at least the following fields be filled in for
|
||||
// particular types of Messages.
|
||||
|
||||
// AUTH_QUERY:
|
||||
// socket_family, socket_protocol
|
||||
// query_address, query_port
|
||||
// query_message
|
||||
// query_time_sec, query_time_nsec
|
||||
|
||||
// AUTH_RESPONSE:
|
||||
// socket_family, socket_protocol
|
||||
// query_address, query_port
|
||||
// query_time_sec, query_time_nsec
|
||||
// response_message
|
||||
// response_time_sec, response_time_nsec
|
||||
|
||||
// RESOLVER_QUERY:
|
||||
// socket_family, socket_protocol
|
||||
// query_message
|
||||
// query_time_sec, query_time_nsec
|
||||
// query_zone
|
||||
// response_address, response_port
|
||||
|
||||
// RESOLVER_RESPONSE:
|
||||
// socket_family, socket_protocol
|
||||
// query_time_sec, query_time_nsec
|
||||
// query_zone
|
||||
// response_address, response_port
|
||||
// response_message
|
||||
// response_time_sec, response_time_nsec
|
||||
|
||||
// CLIENT_QUERY:
|
||||
// socket_family, socket_protocol
|
||||
// query_message
|
||||
// query_time_sec, query_time_nsec
|
||||
|
||||
// CLIENT_RESPONSE:
|
||||
// socket_family, socket_protocol
|
||||
// query_time_sec, query_time_nsec
|
||||
// response_message
|
||||
// response_time_sec, response_time_nsec
|
164
vendor/github.com/dnstap/golang-dnstap/dnstap/main.go
generated
vendored
Normal file
164
vendor/github.com/dnstap/golang-dnstap/dnstap/main.go
generated
vendored
Normal file
|
@ -0,0 +1,164 @@
|
|||
/*
|
||||
* Copyright (c) 2013-2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"runtime"
|
||||
"syscall"
|
||||
|
||||
"github.com/dnstap/golang-dnstap"
|
||||
)
|
||||
|
||||
var (
|
||||
flagReadFile = flag.String("r", "", "read dnstap payloads from file")
|
||||
flagReadSock = flag.String("u", "", "read dnstap payloads from unix socket")
|
||||
flagWriteFile = flag.String("w", "-", "write output to file")
|
||||
flagQuietText = flag.Bool("q", false, "use quiet text output")
|
||||
flagYamlText = flag.Bool("y", false, "use verbose YAML output")
|
||||
)
|
||||
|
||||
func usage() {
|
||||
fmt.Fprintf(os.Stderr, "Usage: %s [OPTION]...\n", os.Args[0])
|
||||
flag.PrintDefaults()
|
||||
fmt.Fprintf(os.Stderr, `
|
||||
Quiet text output format mnemonics:
|
||||
AQ: AUTH_QUERY
|
||||
AR: AUTH_RESPONSE
|
||||
RQ: RESOLVER_QUERY
|
||||
RR: RESOLVER_RESPONSE
|
||||
CQ: CLIENT_QUERY
|
||||
CR: CLIENT_RESPONSE
|
||||
FQ: FORWARDER_QUERY
|
||||
FR: FORWARDER_RESPONSE
|
||||
SQ: STUB_QUERY
|
||||
SR: STUB_RESPONSE
|
||||
TQ: TOOL_QUERY
|
||||
TR: TOOL_RESPONSE
|
||||
`)
|
||||
}
|
||||
|
||||
func outputOpener(fname string, text, yaml bool) func() dnstap.Output {
|
||||
return func() dnstap.Output {
|
||||
var o dnstap.Output
|
||||
var err error
|
||||
if text {
|
||||
o, err = dnstap.NewTextOutputFromFilename(fname, dnstap.TextFormat)
|
||||
} else if yaml {
|
||||
o, err = dnstap.NewTextOutputFromFilename(fname, dnstap.YamlFormat)
|
||||
} else {
|
||||
o, err = dnstap.NewFrameStreamOutputFromFilename(fname)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "dnstap: Failed to open output file: %s\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
go o.RunOutputLoop()
|
||||
return o
|
||||
}
|
||||
}
|
||||
|
||||
func outputLoop(opener func() dnstap.Output, data <-chan []byte, done chan<- struct{}) {
|
||||
sigch := make(chan os.Signal, 1)
|
||||
signal.Notify(sigch, os.Interrupt, syscall.SIGHUP)
|
||||
o := opener()
|
||||
defer func() {
|
||||
o.Close()
|
||||
close(done)
|
||||
os.Exit(0)
|
||||
}()
|
||||
for {
|
||||
select {
|
||||
case b, ok := <-data:
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
o.GetOutputChannel() <- b
|
||||
case sig := <-sigch:
|
||||
if sig == syscall.SIGHUP {
|
||||
o.Close()
|
||||
o = opener()
|
||||
continue
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
var i dnstap.Input
|
||||
|
||||
runtime.GOMAXPROCS(runtime.NumCPU())
|
||||
log.SetFlags(0)
|
||||
flag.Usage = usage
|
||||
|
||||
// Handle command-line arguments.
|
||||
flag.Parse()
|
||||
|
||||
if *flagReadFile == "" && *flagReadSock == "" {
|
||||
fmt.Fprintf(os.Stderr, "dnstap: Error: no inputs specified.\n")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if *flagWriteFile == "-" {
|
||||
if *flagQuietText == false && *flagYamlText == false {
|
||||
*flagQuietText = true
|
||||
}
|
||||
}
|
||||
|
||||
if *flagReadFile != "" && *flagReadSock != "" {
|
||||
fmt.Fprintf(os.Stderr, "dnstap: Error: specify exactly one of -r or -u.\n")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Start the output loop.
|
||||
output := make(chan []byte, 1)
|
||||
opener := outputOpener(*flagWriteFile, *flagQuietText, *flagYamlText)
|
||||
outDone := make(chan struct{})
|
||||
go outputLoop(opener, output, outDone)
|
||||
|
||||
// Open the input and start the input loop.
|
||||
if *flagReadFile != "" {
|
||||
i, err = dnstap.NewFrameStreamInputFromFilename(*flagReadFile)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "dnstap: Failed to open input file: %s\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
fmt.Fprintf(os.Stderr, "dnstap: opened input file %s\n", *flagReadFile)
|
||||
} else if *flagReadSock != "" {
|
||||
i, err = dnstap.NewFrameStreamSockInputFromPath(*flagReadSock)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "dnstap: Failed to open input socket: %s\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
fmt.Fprintf(os.Stderr, "dnstap: opened input socket %s\n", *flagReadSock)
|
||||
}
|
||||
i.ReadInto(output)
|
||||
|
||||
// Wait for input loop to finish.
|
||||
i.Wait()
|
||||
close(output)
|
||||
|
||||
<-outDone
|
||||
}
|
13
vendor/github.com/farsightsec/golang-framestream/COPYRIGHT
generated
vendored
Normal file
13
vendor/github.com/farsightsec/golang-framestream/COPYRIGHT
generated
vendored
Normal file
|
@ -0,0 +1,13 @@
|
|||
Copyright (c) 2014 by Farsight Security, Inc.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
156
vendor/github.com/farsightsec/golang-framestream/Control.go
generated
vendored
Normal file
156
vendor/github.com/farsightsec/golang-framestream/Control.go
generated
vendored
Normal file
|
@ -0,0 +1,156 @@
|
|||
package framestream
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"encoding/binary"
|
||||
"io"
|
||||
)
|
||||
|
||||
const CONTROL_ACCEPT = 0x01
|
||||
const CONTROL_START = 0x02
|
||||
const CONTROL_STOP = 0x03
|
||||
const CONTROL_READY = 0x04
|
||||
const CONTROL_FINISH = 0x05
|
||||
|
||||
const CONTROL_FIELD_CONTENT_TYPE = 0x01
|
||||
|
||||
type ControlFrame struct {
|
||||
ControlType uint32
|
||||
ContentTypes [][]byte
|
||||
}
|
||||
|
||||
var ControlStart = ControlFrame{ControlType: CONTROL_START}
|
||||
var ControlStop = ControlFrame{ControlType: CONTROL_STOP}
|
||||
var ControlReady = ControlFrame{ControlType: CONTROL_READY}
|
||||
var ControlAccept = ControlFrame{ControlType: CONTROL_ACCEPT}
|
||||
var ControlFinish = ControlFrame{ControlType: CONTROL_FINISH}
|
||||
|
||||
func (c *ControlFrame) Encode(w io.Writer) (err error) {
|
||||
var buf bytes.Buffer
|
||||
err = binary.Write(&buf, binary.BigEndian, c.ControlType)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
for _, ctype := range c.ContentTypes {
|
||||
err = binary.Write(&buf, binary.BigEndian, uint32(CONTROL_FIELD_CONTENT_TYPE))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
err = binary.Write(&buf, binary.BigEndian, uint32(len(ctype)))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
_, err = buf.Write(ctype)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
err = binary.Write(w, binary.BigEndian, uint32(0))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = binary.Write(w, binary.BigEndian, uint32(buf.Len()))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
_, err = buf.WriteTo(w)
|
||||
return
|
||||
}
|
||||
|
||||
func (c *ControlFrame) EncodeFlush(w *bufio.Writer) error {
|
||||
if err := c.Encode(w); err != nil {
|
||||
return err
|
||||
}
|
||||
return w.Flush()
|
||||
}
|
||||
|
||||
func (c *ControlFrame) Decode(r io.Reader) (err error) {
|
||||
var cflen uint32
|
||||
err = binary.Read(r, binary.BigEndian, &cflen)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
err = binary.Read(r, binary.BigEndian, &c.ControlType)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
cflen -= 4
|
||||
if cflen > 0 {
|
||||
cfields := make([]byte, int(cflen))
|
||||
_, err = io.ReadFull(r, cfields)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
for len(cfields) > 8 {
|
||||
cftype := binary.BigEndian.Uint32(cfields[:4])
|
||||
cfields = cfields[4:]
|
||||
if cftype != CONTROL_FIELD_CONTENT_TYPE {
|
||||
return ErrDecode
|
||||
}
|
||||
|
||||
cflen := int(binary.BigEndian.Uint32(cfields[:4]))
|
||||
cfields = cfields[4:]
|
||||
if cflen > len(cfields) {
|
||||
return ErrDecode
|
||||
}
|
||||
|
||||
c.ContentTypes = append(c.ContentTypes, cfields[:cflen])
|
||||
cfields = cfields[cflen:]
|
||||
}
|
||||
|
||||
if len(cfields) > 0 {
|
||||
return ErrDecode
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (c *ControlFrame) DecodeEscape(r io.Reader) error {
|
||||
var zero uint32
|
||||
err := binary.Read(r, binary.BigEndian, &zero)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if zero != 0 {
|
||||
return ErrDecode
|
||||
}
|
||||
return c.Decode(r)
|
||||
}
|
||||
|
||||
func (c *ControlFrame) DecodeTypeEscape(r io.Reader, ctype uint32) error {
|
||||
err := c.DecodeEscape(r)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if ctype != c.ControlType {
|
||||
return ErrDecode
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *ControlFrame) MatchContentType(ctype []byte) bool {
|
||||
if ctype == nil {
|
||||
return true
|
||||
}
|
||||
for _, cfctype := range c.ContentTypes {
|
||||
if bytes.Compare(ctype, cfctype) == 0 {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (c *ControlFrame) SetContentType(ctype []byte) {
|
||||
if ctype != nil {
|
||||
c.ContentTypes = [][]byte{ctype}
|
||||
}
|
||||
}
|
154
vendor/github.com/farsightsec/golang-framestream/Decoder.go
generated
vendored
Normal file
154
vendor/github.com/farsightsec/golang-framestream/Decoder.go
generated
vendored
Normal file
|
@ -0,0 +1,154 @@
|
|||
/*
|
||||
* Copyright (c) 2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package framestream
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/binary"
|
||||
"io"
|
||||
)
|
||||
|
||||
type DecoderOptions struct {
|
||||
MaxPayloadSize uint32
|
||||
ContentType []byte
|
||||
Bidirectional bool
|
||||
}
|
||||
|
||||
type Decoder struct {
|
||||
buf []byte
|
||||
opt DecoderOptions
|
||||
reader *bufio.Reader
|
||||
writer *bufio.Writer
|
||||
stopped bool
|
||||
}
|
||||
|
||||
func NewDecoder(r io.Reader, opt *DecoderOptions) (dec *Decoder, err error) {
|
||||
if opt == nil {
|
||||
opt = &DecoderOptions{}
|
||||
}
|
||||
if opt.MaxPayloadSize == 0 {
|
||||
opt.MaxPayloadSize = DEFAULT_MAX_PAYLOAD_SIZE
|
||||
}
|
||||
dec = &Decoder{
|
||||
buf: make([]byte, opt.MaxPayloadSize),
|
||||
opt: *opt,
|
||||
reader: bufio.NewReader(r),
|
||||
writer: nil,
|
||||
}
|
||||
|
||||
var cf ControlFrame
|
||||
if opt.Bidirectional {
|
||||
w, ok := r.(io.Writer)
|
||||
if !ok {
|
||||
return dec, ErrType
|
||||
}
|
||||
dec.writer = bufio.NewWriter(w)
|
||||
|
||||
// Read the ready control frame.
|
||||
err = cf.DecodeTypeEscape(dec.reader, CONTROL_READY)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// Check content type.
|
||||
if !cf.MatchContentType(dec.opt.ContentType) {
|
||||
return dec, ErrContentTypeMismatch
|
||||
}
|
||||
|
||||
// Send the accept control frame.
|
||||
accept := ControlAccept
|
||||
accept.SetContentType(dec.opt.ContentType)
|
||||
err = accept.EncodeFlush(dec.writer)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Read the start control frame.
|
||||
err = cf.DecodeTypeEscape(dec.reader, CONTROL_START)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// Check content type.
|
||||
if !cf.MatchContentType(dec.opt.ContentType) {
|
||||
return dec, ErrContentTypeMismatch
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (dec *Decoder) readFrame(frameLen uint32) (err error) {
|
||||
// Enforce limits on frame size.
|
||||
if frameLen > dec.opt.MaxPayloadSize {
|
||||
err = ErrDataFrameTooLarge
|
||||
return
|
||||
}
|
||||
|
||||
// Read the frame.
|
||||
n, err := io.ReadFull(dec.reader, dec.buf[0:frameLen])
|
||||
if err != nil || uint32(n) != frameLen {
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (dec *Decoder) Decode() (frameData []byte, err error) {
|
||||
if dec.stopped {
|
||||
err = EOF
|
||||
return
|
||||
}
|
||||
|
||||
// Read the frame length.
|
||||
var frameLen uint32
|
||||
err = binary.Read(dec.reader, binary.BigEndian, &frameLen)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if frameLen == 0 {
|
||||
// This is a control frame.
|
||||
var cf ControlFrame
|
||||
err = cf.Decode(dec.reader)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if cf.ControlType == CONTROL_STOP {
|
||||
dec.stopped = true
|
||||
if dec.opt.Bidirectional {
|
||||
ff := &ControlFrame{ControlType: CONTROL_FINISH}
|
||||
err = ff.EncodeFlush(dec.writer)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
return nil, EOF
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
} else {
|
||||
// This is a data frame.
|
||||
err = dec.readFrame(frameLen)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
frameData = dec.buf[0:frameLen]
|
||||
}
|
||||
|
||||
return
|
||||
}
|
99
vendor/github.com/farsightsec/golang-framestream/Encoder.go
generated
vendored
Normal file
99
vendor/github.com/farsightsec/golang-framestream/Encoder.go
generated
vendored
Normal file
|
@ -0,0 +1,99 @@
|
|||
/*
|
||||
* Copyright (c) 2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package framestream
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/binary"
|
||||
"io"
|
||||
)
|
||||
|
||||
type EncoderOptions struct {
|
||||
ContentType []byte
|
||||
Bidirectional bool
|
||||
}
|
||||
|
||||
type Encoder struct {
|
||||
writer *bufio.Writer
|
||||
reader *bufio.Reader
|
||||
opt EncoderOptions
|
||||
buf []byte
|
||||
}
|
||||
|
||||
func NewEncoder(w io.Writer, opt *EncoderOptions) (enc *Encoder, err error) {
|
||||
if opt == nil {
|
||||
opt = &EncoderOptions{}
|
||||
}
|
||||
enc = &Encoder{
|
||||
writer: bufio.NewWriter(w),
|
||||
opt: *opt,
|
||||
}
|
||||
|
||||
if opt.Bidirectional {
|
||||
r, ok := w.(io.Reader)
|
||||
if !ok {
|
||||
return nil, ErrType
|
||||
}
|
||||
enc.reader = bufio.NewReader(r)
|
||||
ready := ControlReady
|
||||
ready.SetContentType(opt.ContentType)
|
||||
if err = ready.EncodeFlush(enc.writer); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
var accept ControlFrame
|
||||
if err = accept.DecodeTypeEscape(enc.reader, CONTROL_ACCEPT); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if !accept.MatchContentType(opt.ContentType) {
|
||||
return nil, ErrContentTypeMismatch
|
||||
}
|
||||
}
|
||||
|
||||
// Write the start control frame.
|
||||
start := ControlStart
|
||||
start.SetContentType(opt.ContentType)
|
||||
err = start.Encode(enc.writer)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (enc *Encoder) Close() (err error) {
|
||||
err = ControlStop.EncodeFlush(enc.writer)
|
||||
if err != nil || !enc.opt.Bidirectional {
|
||||
return
|
||||
}
|
||||
|
||||
var finish ControlFrame
|
||||
return finish.DecodeTypeEscape(enc.reader, CONTROL_FINISH)
|
||||
}
|
||||
|
||||
func (enc *Encoder) Write(frame []byte) (n int, err error) {
|
||||
err = binary.Write(enc.writer, binary.BigEndian, uint32(len(frame)))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return enc.writer.Write(frame)
|
||||
}
|
||||
|
||||
func (enc *Encoder) Flush() error {
|
||||
return enc.writer.Flush()
|
||||
}
|
202
vendor/github.com/farsightsec/golang-framestream/LICENSE
generated
vendored
Normal file
202
vendor/github.com/farsightsec/golang-framestream/LICENSE
generated
vendored
Normal file
|
@ -0,0 +1,202 @@
|
|||
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright [yyyy] [name of copyright owner]
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
32
vendor/github.com/farsightsec/golang-framestream/framestream.go
generated
vendored
Normal file
32
vendor/github.com/farsightsec/golang-framestream/framestream.go
generated
vendored
Normal file
|
@ -0,0 +1,32 @@
|
|||
/*
|
||||
* Copyright (c) 2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package framestream
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io"
|
||||
)
|
||||
|
||||
const DEFAULT_MAX_PAYLOAD_SIZE = 1048576
|
||||
const MAX_CONTROL_FRAME_SIZE = 512
|
||||
|
||||
var EOF = io.EOF
|
||||
var ErrContentTypeMismatch = errors.New("content type mismatch")
|
||||
var ErrDataFrameTooLarge = errors.New("data frame too large")
|
||||
var ErrShortRead = errors.New("short read")
|
||||
var ErrDecode = errors.New("decoding error")
|
||||
var ErrType = errors.New("invalid type")
|
63
vendor/github.com/farsightsec/golang-framestream/framestream_dump/main.go
generated
vendored
Normal file
63
vendor/github.com/farsightsec/golang-framestream/framestream_dump/main.go
generated
vendored
Normal file
|
@ -0,0 +1,63 @@
|
|||
/*
|
||||
* Copyright (c) 2014 by Farsight Security, Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/farsightsec/golang-framestream"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Arguments.
|
||||
if len(os.Args) != 2 {
|
||||
fmt.Fprintf(os.Stderr, "Usage: %s <INPUT FILE>\n", os.Args[0])
|
||||
fmt.Fprintf(os.Stderr, "Dumps a FrameStreams formatted input file.\n\n")
|
||||
os.Exit(1)
|
||||
}
|
||||
fname := os.Args[1]
|
||||
|
||||
// Open the input file.
|
||||
file, err := os.Open(fname)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
// Create the decoder.
|
||||
fs, err := framestream.NewDecoder(file, nil)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
// Print the frames.
|
||||
fmt.Printf("Control frame [START] (%v bytes): %x\n", len(fs.ControlStart), fs.ControlStart)
|
||||
for {
|
||||
frame, err := fs.Decode()
|
||||
if err == framestream.EOF {
|
||||
break
|
||||
}
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
fmt.Printf("Data frame (%v bytes): %x\n", len(frame), frame)
|
||||
}
|
||||
if fs.ControlStop != nil {
|
||||
fmt.Printf("Control frame [STOP] (%v bytes): %x\n", len(fs.ControlStop), fs.ControlStop)
|
||||
}
|
||||
}
|
134
vendor/github.com/farsightsec/golang-framestream/framestream_test.go
generated
vendored
Normal file
134
vendor/github.com/farsightsec/golang-framestream/framestream_test.go
generated
vendored
Normal file
|
@ -0,0 +1,134 @@
|
|||
package framestream_test
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"net"
|
||||
"testing"
|
||||
|
||||
framestream "github.com/farsightsec/golang-framestream"
|
||||
)
|
||||
|
||||
func testDecoder(t *testing.T, dec *framestream.Decoder, nframes int) {
|
||||
i := 1
|
||||
for {
|
||||
tf, err := dec.Decode()
|
||||
if err != nil {
|
||||
if i < nframes+1 {
|
||||
t.Fatalf("testDecoder(%d): %v", i, err)
|
||||
}
|
||||
if err != framestream.EOF {
|
||||
t.Fatalf("unexpected error: %v != EOF", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
if i > nframes {
|
||||
t.Errorf("extra frame received: %d", i)
|
||||
}
|
||||
f := make([]byte, i)
|
||||
if bytes.Compare(tf, f) != 0 {
|
||||
t.Errorf("testDecoder: received %v != %v", tf, f)
|
||||
}
|
||||
i++
|
||||
}
|
||||
}
|
||||
|
||||
func TestUnidirectional(t *testing.T) {
|
||||
buf := new(bytes.Buffer)
|
||||
enc, err := framestream.NewEncoder(buf, nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
for i := 1; i < 10; i++ {
|
||||
b := make([]byte, i)
|
||||
if _, err = enc.Write(b); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
enc.Close()
|
||||
|
||||
dec, err := framestream.NewDecoder(buf, nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
testDecoder(t, dec, 9)
|
||||
}
|
||||
|
||||
func TestBidirectional(t *testing.T) {
|
||||
client, server := net.Pipe()
|
||||
|
||||
go func() {
|
||||
dec, err := framestream.NewDecoder(server,
|
||||
&framestream.DecoderOptions{
|
||||
Bidirectional: true,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
testDecoder(t, dec, 9)
|
||||
}()
|
||||
|
||||
enc, err := framestream.NewEncoder(client,
|
||||
&framestream.EncoderOptions{
|
||||
Bidirectional: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
for i := 1; i < 10; i++ {
|
||||
b := make([]byte, i)
|
||||
if _, err := enc.Write(b); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
enc.Close()
|
||||
}
|
||||
|
||||
func TestContentTypeMismatch(t *testing.T) {
|
||||
buf := new(bytes.Buffer)
|
||||
|
||||
enc, err := framestream.NewEncoder(buf,
|
||||
&framestream.EncoderOptions{
|
||||
ContentType: []byte("test"),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
enc.Write([]byte("hello, world"))
|
||||
enc.Close()
|
||||
|
||||
_, err = framestream.NewDecoder(buf,
|
||||
&framestream.DecoderOptions{
|
||||
ContentType: []byte("wrong"),
|
||||
})
|
||||
if err != framestream.ErrContentTypeMismatch {
|
||||
t.Error("expected %v, received %v",
|
||||
framestream.ErrContentTypeMismatch,
|
||||
err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestOversizeFrame(t *testing.T) {
|
||||
buf := new(bytes.Buffer)
|
||||
enc, err := framestream.NewEncoder(buf, nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
enc.Write(make([]byte, 15))
|
||||
enc.Close()
|
||||
|
||||
dec, err := framestream.NewDecoder(buf,
|
||||
&framestream.DecoderOptions{
|
||||
MaxPayloadSize: 10,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
_, err = dec.Decode()
|
||||
if err != framestream.ErrDataFrameTooLarge {
|
||||
t.Error("data frame too large, received %v", err)
|
||||
}
|
||||
}
|
Loading…
Add table
Reference in a new issue