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"
|
revision = "1195e3a8ee1a529d53eed7c624527a68555ddf1f"
|
||||||
version = "v1.5.1"
|
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]
|
[solve-meta]
|
||||||
analyzer-name = "dep"
|
analyzer-name = "dep"
|
||||||
analyzer-version = 1
|
analyzer-version = 1
|
||||||
|
|
|
@ -11,6 +11,7 @@ import (
|
||||||
_ "github.com/coredns/coredns/middleware/cache"
|
_ "github.com/coredns/coredns/middleware/cache"
|
||||||
_ "github.com/coredns/coredns/middleware/chaos"
|
_ "github.com/coredns/coredns/middleware/chaos"
|
||||||
_ "github.com/coredns/coredns/middleware/dnssec"
|
_ "github.com/coredns/coredns/middleware/dnssec"
|
||||||
|
_ "github.com/coredns/coredns/middleware/dnstap"
|
||||||
_ "github.com/coredns/coredns/middleware/erratic"
|
_ "github.com/coredns/coredns/middleware/erratic"
|
||||||
_ "github.com/coredns/coredns/middleware/errors"
|
_ "github.com/coredns/coredns/middleware/errors"
|
||||||
_ "github.com/coredns/coredns/middleware/etcd"
|
_ "github.com/coredns/coredns/middleware/etcd"
|
||||||
|
|
|
@ -21,6 +21,7 @@ var directives = []string{
|
||||||
"prometheus",
|
"prometheus",
|
||||||
"errors",
|
"errors",
|
||||||
"log",
|
"log",
|
||||||
|
"dnstap",
|
||||||
"chaos",
|
"chaos",
|
||||||
"cache",
|
"cache",
|
||||||
"rewrite",
|
"rewrite",
|
||||||
|
|
|
@ -29,6 +29,7 @@
|
||||||
70:prometheus:metrics
|
70:prometheus:metrics
|
||||||
80:errors:errors
|
80:errors:errors
|
||||||
90:log:log
|
90:log:log
|
||||||
|
95:dnstap:dnstap
|
||||||
100:chaos:chaos
|
100:chaos:chaos
|
||||||
110:cache:cache
|
110:cache:cache
|
||||||
120:rewrite:rewrite
|
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