commit 3d18b67edf1c2e9ff908287485e1d517a3706058 · seiso.moe/magna

+34

message.go

···

       1
        
       package magna

     

       2
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       3
        
       // Decode decodes a DNS packet.

     

       4
        
       func (m *Message) Decode(buf []byte) (err error) {

     

       5
        
       	offset, err := m.Header.Decode(buf, 0)

     
···

       75
        
       	return bytes

     

       76
        
       }

     

       77
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       78
        
       func (m *Message) CreateReply(req *Message) *Message {

     

       79
        
       	m.Header.ID = req.Header.ID

     

       80
        
       	m.Header.QR = true

     
···

       89
        
       

     

       90
        
       	return m

     

       91
        
       }

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0

···

       1
        
       package magna

     

       2
        
       

     

       3
       +
       import (

     

       4
       +
       	"math/rand"

     

       5
       +
       )

     

       6
       +
       

     

       7
        
       // Decode decodes a DNS packet.

     

       8
        
       func (m *Message) Decode(buf []byte) (err error) {

     

       9
        
       	offset, err := m.Header.Decode(buf, 0)

     
···

       79
        
       	return bytes

     

       80
        
       }

     

       81
        
       

     

       82
       +
       func CreateRequest(op OPCode, rd bool) *Message {

     

       83
       +
       	return &Message{

     

       84
       +
       		Header: Header{

     

       85
       +
       			ID:      uint16(rand.Intn(65534) + 1),

     

       86
       +
       			QR:      false,

     

       87
       +
       			OPCode:  op,

     

       88
       +
       			AA:      false,

     

       89
       +
       			TC:      false,

     

       90
       +
       			RD:      rd,

     

       91
       +
       			RA:      false,

     

       92
       +
       			Z:       0,

     

       93
       +
       			RCode:   NOERROR,

     

       94
       +
       			QDCount: 0,

     

       95
       +
       			ARCount: 0,

     

       96
       +
       			NSCount: 0,

     

       97
       +
       			ANCount: 0,

     

       98
       +
       		},

     

       99
       +
       		Question:   make([]Question, 0),

     

       100
       +
       		Answer:     make([]ResourceRecord, 0),

     

       101
       +
       		Additional: make([]ResourceRecord, 0),

     

       102
       +
       		Authority:  make([]ResourceRecord, 0),

     

       103
       +
       	}

     

       104
       +
       }

     

       105
       +
       

     

       106
        
       func (m *Message) CreateReply(req *Message) *Message {

     

       107
        
       	m.Header.ID = req.Header.ID

     

       108
        
       	m.Header.QR = true

     
···

       117
        
       

     

       118
        
       	return m

     

       119
        
       }

     

       120
       +
       

     

       121
       +
       func (m *Message) SetRCode(rc RCode) *Message {

     

       122
       +
       	m.Header.RCode = rc

     

       123
       +
       

     

       124
       +
       	return m

     

       125
       +
       }