func TestMessageProcessingV1(t *testing.T) {
snmp, _ := snmpgo.NewSNMP(snmpgo.SNMPArguments{
Version: snmpgo.V2c,
Community: "public",
})
mp := snmpgo.NewMessageProcessing(snmpgo.V2c)
pdu := snmpgo.NewPdu(snmpgo.V2c, snmpgo.GetRequest)
msg, err := mp.PrepareOutgoingMessage(snmp, pdu)
if err != nil {
t.Errorf("PrepareOutgoingMessage() - has error %v", err)
}
if len(msg.PduBytes()) == 0 {
t.Error("PrepareOutgoingMessage() - pdu bytes")
}
if pdu.RequestId() == 0 {
t.Error("PrepareOutgoingMessage() - request id")
}
requestId := pdu.RequestId()
_, err = mp.PrepareDataElements(snmp, msg, []byte{0x00, 0x00})
if err == nil {
t.Error("PrepareDataElements() - message unmarshal error")
}
b, _ := msg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err == nil {
t.Error("PrepareDataElements() - pdu type check")
}
pdu = snmpgo.NewPdu(snmpgo.V2c, snmpgo.GetResponse)
rmsg := snmpgo.ToMessageV1(snmpgo.NewMessage(snmpgo.V1, pdu))
b, _ = rmsg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err == nil {
t.Error("PrepareDataElements() - version check")
}
pdu.SetRequestId(requestId)
pduBytes, _ := pdu.Marshal()
rmsg = snmpgo.ToMessageV1(snmpgo.NewMessage(snmpgo.V2c, pdu))
rmsg.Community = []byte("public")
rmsg.SetPduBytes(pduBytes)
b, _ = rmsg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err != nil {
t.Errorf("PrepareDataElements() - has error %v", err)
}
}
func TestNewPdu(t *testing.T) {
pdu := snmpgo.NewPdu(snmpgo.V1, snmpgo.GetRequest)
if _, ok := pdu.(*snmpgo.PduV1); !ok {
t.Errorf("NewPdu() Invalid Pdu")
}
pdu = snmpgo.NewPdu(snmpgo.V2c, snmpgo.GetRequest)
if _, ok := pdu.(*snmpgo.PduV1); !ok {
t.Errorf("NewPdu() Invalid Pdu")
}
pdu = snmpgo.NewPdu(snmpgo.V3, snmpgo.GetRequest)
if _, ok := pdu.(*snmpgo.ScopedPdu); !ok {
t.Errorf("NewPdu() Invalid Pdu")
}
}
func TestMessageV1(t *testing.T) {
pdu := snmpgo.NewPdu(snmpgo.V2c, snmpgo.GetRequest)
msg := snmpgo.ToMessageV1(snmpgo.NewMessage(snmpgo.V2c, pdu))
b, _ := pdu.Marshal()
msg.SetPduBytes(b)
msg.Community = []byte("MyCommunity")
expBuf := []byte{
0x30, 0x1d, 0x02, 0x01, 0x01, 0x04, 0x0b, 0x4d, 0x79, 0x43, 0x6f,
0x6d, 0x6d, 0x75, 0x6e, 0x69, 0x74, 0x79, 0xa0, 0x0b, 0x02, 0x01,
0x00, 0x02, 0x01, 0x00, 0x02, 0x01, 0x00, 0x30, 0x00,
}
buf, err := msg.Marshal()
if err != nil {
t.Fatal("Marshal() : %v", err)
}
if !bytes.Equal(expBuf, buf) {
t.Errorf("Marshal() - expected [%s], actual [%s]",
snmpgo.ToHexStr(expBuf, " "), snmpgo.ToHexStr(buf, " "))
}
expStr := `{"Version": "2c", "Community": "MyCommunity", ` +
`"Pdu": {"Type": "GetRequest", "RequestId": "0", "ErrorStatus": ` +
`"NoError", "ErrorIndex": "0", "VarBinds": []}}`
m := snmpgo.NewMessage(snmpgo.V2c, pdu)
rest, err := m.Unmarshal(buf)
if len(rest) != 0 || err != nil {
t.Errorf("Unmarshal() - len[%d] err[%v]", len(rest), err)
}
if expStr != m.String() {
t.Errorf("Unmarshal() - expected [%s], actual [%s]", expStr, m.String())
}
}
func TestMessageV3(t *testing.T) {
pdu := snmpgo.NewPdu(snmpgo.V3, snmpgo.GetRequest)
msg := snmpgo.ToMessageV3(snmpgo.NewMessage(snmpgo.V3, pdu))
b, _ := pdu.Marshal()
msg.SetPduBytes(b)
msg.MessageId = 123
msg.MessageMaxSize = 321
msg.SetReportable(true)
msg.SetPrivacy(true)
msg.SetAuthentication(true)
msg.SecurityModel = 3
msg.AuthEngineId = []byte{0x80, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07}
msg.AuthEngineBoots = 456
msg.AuthEngineTime = 654
msg.UserName = []byte("User")
msg.AuthParameter = []byte{0xaa, 0xbb, 0xcc}
msg.PrivParameter = []byte{0xdd, 0xee, 0xff}
expBuf := []byte{
0x30, 0x4b, 0x02, 0x01, 0x03, 0x30, 0x0d, 0x02, 0x01, 0x7b,
0x02, 0x02, 0x01, 0x41, 0x04, 0x01, 0x07, 0x02, 0x01, 0x03,
0x04, 0x24, 0x30, 0x22, 0x04, 0x08, 0x80, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
0x02, 0x02, 0x01, 0xc8, 0x02, 0x02, 0x02, 0x8e, 0x04, 0x04, 0x55, 0x73, 0x65, 0x72,
0x04, 0x03, 0xaa, 0xbb, 0xcc, 0x04, 0x03, 0xdd, 0xee, 0xff,
0x30, 0x11, 0x04, 0x00, 0x04, 0x00, 0xa0, 0x0b, 0x02, 0x01,
0x00, 0x02, 0x01, 0x00, 0x02, 0x01, 0x00, 0x30, 0x00,
}
buf, err := msg.Marshal()
if err != nil {
t.Fatal("Marshal() : %v", err)
}
if !bytes.Equal(expBuf, buf) {
t.Errorf("Marshal() - expected [%s], actual [%s]",
snmpgo.ToHexStr(expBuf, " "), snmpgo.ToHexStr(buf, " "))
}
expStr := `{"Version": "3", "GlobalData": {"MessageId": "123", "MessageMaxSize": "321", ` +
`"MessageFlags": "apr", "SecurityModel": "USM"}, "SecurityParameter": ` +
`{"AuthEngineId": "8001020304050607", "AuthEngineBoots": "456", ` +
`"AuthEngineTime": "654", "UserName": "******", "AuthParameter": "aa:bb:cc", ` +
`"PrivParameter": "dd:ee:ff"}, "Pdu": {"Type": "GetRequest", "RequestId": "0", ` +
`"ErrorStatus": "NoError", "ErrorIndex": "0", "ContextEngineId": "", ` +
`"ContextName": "", "VarBinds": []}}`
m := snmpgo.NewMessage(snmpgo.V3, pdu)
rest, err := m.Unmarshal(buf)
if len(rest) != 0 || err != nil {
t.Errorf("Unmarshal() - len[%d] err[%v]", len(rest), err)
}
if expStr != m.String() {
t.Errorf("Unmarshal() - expected [%s], actual [%s]", expStr, m.String())
}
}
func TestScopedPdu(t *testing.T) {
pdu := snmpgo.NewPdu(snmpgo.V3, snmpgo.GetRequest)
pdu.SetRequestId(123)
pdu.SetErrorStatus(snmpgo.TooBig)
pdu.SetErrorIndex(2)
sp := pdu.(*snmpgo.ScopedPdu)
sp.ContextEngineId = []byte{0x80, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07}
sp.ContextName = []byte("MyContext")
oid, _ := snmpgo.NewOid("1.3.6.1.2.1.1.1.0")
pdu.AppendVarBind(oid, snmpgo.NewOctetString([]byte("MyHost")))
oid, _ = snmpgo.NewOid("1.3.6.1.2.1.1.2.0")
pdu.AppendVarBind(oid, snmpgo.NewNull())
oid, _ = snmpgo.NewOid("1.3.6.1.2.1.1.3.0")
pdu.AppendVarBind(oid, snmpgo.NewTimeTicks(uint32(11111)))
expBuf := []byte{
0x30, 0x54, 0x04, 0x08, 0x80, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
0x04, 0x09, 0x4d, 0x79, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x78, 0x74,
0xa0, 0x3d, 0x02, 0x01, 0x7b, 0x02, 0x01, 0x01, 0x02, 0x01, 0x02,
0x30, 0x32, 0x30, 0x12, 0x06, 0x08, 0x2b, 0x06, 0x01, 0x02, 0x01,
0x01, 0x01, 0x00, 0x04, 0x06, 0x4d, 0x79, 0x48, 0x6f, 0x73, 0x74,
0x30, 0x0c, 0x06, 0x08, 0x2b, 0x06, 0x01, 0x02, 0x01, 0x01, 0x02,
0x00, 0x05, 0x00, 0x30, 0x0e, 0x06, 0x08, 0x2b, 0x06, 0x01, 0x02,
0x01, 0x01, 0x03, 0x00, 0x43, 0x02, 0x2b, 0x67,
}
buf, err := pdu.Marshal()
if err != nil {
t.Fatal("Marshal() : %v", err)
}
if !bytes.Equal(expBuf, buf) {
t.Errorf("Marshal() - expected [%s], actual [%s]",
snmpgo.ToHexStr(expBuf, " "), snmpgo.ToHexStr(buf, " "))
}
expStr := `{"Type": "GetRequest", "RequestId": "123", ` +
`"ErrorStatus": "TooBig", "ErrorIndex": "2", ` +
`"ContextEngineId": "8001020304050607", "ContextName": "MyContext", ` +
`"VarBinds": [` +
`{"Oid": "1.3.6.1.2.1.1.1.0", "Variable": {"Type": "OctetString", "Value": "MyHost"}}, ` +
`{"Oid": "1.3.6.1.2.1.1.2.0", "Variable": {"Type": "Null", "Value": ""}}, ` +
`{"Oid": "1.3.6.1.2.1.1.3.0", "Variable": {"Type": "TimeTicks", "Value": "11111"}}]}`
var w snmpgo.ScopedPdu
rest, err := (&w).Unmarshal(buf)
if len(rest) != 0 || err != nil {
t.Errorf("Unmarshal() - len[%d] err[%v]", len(rest), err)
}
if expStr != w.String() {
t.Errorf("Unmarshal() - expected [%s], actual [%s]", expStr, w.String())
}
}
func TestCommunity(t *testing.T) {
expCom := "public"
snmp, _ := snmpgo.NewSNMP(snmpgo.SNMPArguments{
Version: snmpgo.V2c,
Community: expCom,
})
sec := snmpgo.NewCommunity()
pdu := snmpgo.NewPdu(snmpgo.V2c, snmpgo.GetRequest)
smsg := snmpgo.ToMessageV1(snmpgo.NewMessage(snmpgo.V2c, pdu))
err := sec.GenerateRequestMessage(snmp, smsg)
if err != nil {
t.Errorf("GenerateRequestMessage() - has error %v", err)
}
if !bytes.Equal(smsg.Community, []byte(expCom)) {
t.Errorf("GenerateRequestMessage() - expected [%s], actual [%s]", expCom, smsg.Community)
}
if len(smsg.PduBytes()) == 0 {
t.Error("GenerateRequestMessage() - pdu marshal")
}
pdu = snmpgo.NewPdu(snmpgo.V2c, snmpgo.GetResponse)
rmsg := snmpgo.ToMessageV1(snmpgo.NewMessage(snmpgo.V2c, pdu))
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - community check")
}
rmsg.Community = []byte(expCom)
rmsg.SetPduBytes(smsg.PduBytes())
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err != nil {
t.Errorf("ProcessIncomingMessage() - has error %v", err)
}
}
func TestSNMP(t *testing.T) {
snmp, _ := snmpgo.NewSNMP(snmpgo.SNMPArguments{
Version: snmpgo.V3,
UserName: "******",
SecurityLevel: snmpgo.AuthPriv,
AuthPassword: "******",
AuthProtocol: snmpgo.Md5,
PrivPassword: "******",
PrivProtocol: snmpgo.Des,
})
pdu := snmpgo.NewPdu(snmpgo.V3, snmpgo.Report)
err := snmpgo.SnmpCheckPdu(snmp, pdu)
if err != nil {
t.Errorf("checkPdu() - has error %v", err)
}
oids, _ := snmpgo.NewOids([]string{"1.3.6.1.6.3.11.2.1.1.0"})
pdu = snmpgo.NewPduWithOids(snmpgo.V3, snmpgo.Report, oids)
err = snmpgo.SnmpCheckPdu(snmp, pdu)
if err == nil {
t.Error("checkPdu() - report oid")
}
}
func TestMessageProcessingV3(t *testing.T) {
snmp, _ := snmpgo.NewSNMP(snmpgo.SNMPArguments{
Version: snmpgo.V3,
UserName: "******",
SecurityLevel: snmpgo.AuthPriv,
AuthPassword: "******",
AuthProtocol: snmpgo.Md5,
PrivPassword: "******",
PrivProtocol: snmpgo.Des,
})
mp := snmpgo.NewMessageProcessing(snmpgo.V3)
usm := snmpgo.ToUsm(mp.Security())
usm.AuthKey = []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}
usm.PrivKey = []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}
pdu := snmpgo.NewPdu(snmpgo.V3, snmpgo.GetRequest)
msg, err := mp.PrepareOutgoingMessage(snmp, pdu)
if err != nil {
t.Errorf("PrepareOutgoingMessage() - has error %v", err)
}
if len(msg.PduBytes()) == 0 {
t.Error("PrepareOutgoingMessage() - pdu bytes")
}
if pdu.RequestId() == 0 {
t.Error("PrepareOutgoingMessage() - request id")
}
msgv3 := snmpgo.ToMessageV3(msg)
if msgv3.MessageId == 0 {
t.Error("PrepareOutgoingMessage() - message id")
}
if !msgv3.Reportable() || !msgv3.Authentication() || !msgv3.Privacy() {
t.Error("PrepareOutgoingMessage() - security flag")
}
msgv3.SetAuthentication(false)
msgv3.SetPrivacy(false)
msgv3.AuthEngineId = []byte{0, 0, 0, 0, 0}
requestId := pdu.RequestId()
messageId := msgv3.MessageId
_, err = mp.PrepareDataElements(snmp, msg, []byte{0x00, 0x00})
if err == nil {
t.Error("PrepareDataElements() - message unmarshal error")
}
b, _ := msg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err == nil {
t.Error("PrepareDataElements() - pdu type check")
}
pdu = snmpgo.NewPdu(snmpgo.V3, snmpgo.GetResponse)
rmsg := snmpgo.ToMessageV3(snmpgo.NewMessage(snmpgo.V3, pdu))
b, _ = rmsg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err == nil {
t.Error("PrepareDataElements() - message id check")
}
rmsg = snmpgo.ToMessageV3(snmpgo.NewMessage(snmpgo.V3, pdu))
rmsg.AuthEngineId = []byte{0, 0, 0, 0, 0}
rmsg.MessageId = messageId
b, _ = rmsg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err == nil {
t.Error("PrepareDataElements() - security model check")
}
pdu.(*snmpgo.ScopedPdu).ContextEngineId = rmsg.AuthEngineId
pduBytes, _ := pdu.Marshal()
rmsg.SetPduBytes(pduBytes)
rmsg.SecurityModel = 3
b, _ = rmsg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err == nil {
t.Error("PrepareDataElements() - request id check")
}
pdu.SetRequestId(requestId)
pduBytes, _ = pdu.Marshal()
rmsg.SetPduBytes(pduBytes)
b, _ = rmsg.Marshal()
_, err = mp.PrepareDataElements(snmp, msg, b)
if err != nil {
t.Errorf("PrepareDataElements() - has error %v", err)
}
}
func TestUsm(t *testing.T) {
expUser := []byte("myUser")
expEngId := []byte{0x80, 0x00, 0x00, 0x00, 0x01}
expCtxId := []byte{0x80, 0x00, 0x00, 0x00, 0x05}
expCtxName := "myName"
snmp, _ := snmpgo.NewSNMP(snmpgo.SNMPArguments{
Version: snmpgo.V3,
UserName: string(expUser),
SecurityLevel: snmpgo.AuthPriv,
AuthPassword: "******",
AuthProtocol: snmpgo.Md5,
PrivPassword: "******",
PrivProtocol: snmpgo.Des,
ContextEngineId: hex.EncodeToString(expCtxId),
ContextName: expCtxName,
})
sec := snmpgo.NewUsm()
pdu := snmpgo.NewPdu(snmpgo.V3, snmpgo.GetRequest)
spdu := pdu.(*snmpgo.ScopedPdu)
smsg := snmpgo.ToMessageV3(snmpgo.NewMessage(snmpgo.V3, pdu))
smsg.SetAuthentication(false)
smsg.SetPrivacy(false)
// Discovery
err := sec.GenerateRequestMessage(snmp, smsg)
if err != nil {
t.Errorf("GenerateRequestMessage() - has error %v", err)
}
if !bytes.Equal(spdu.ContextEngineId, expCtxId) {
t.Errorf("GenerateRequestMessage() - expected [%s], actual [%s]",
expCtxId, spdu.ContextEngineId)
}
if string(spdu.ContextName) != expCtxName {
t.Errorf("GenerateRequestMessage() - expected [%s], actual [%s]",
expCtxName, string(spdu.ContextName))
}
if len(smsg.PduBytes()) == 0 {
t.Error("GenerateRequestMessage() - pdu marshal")
}
pdu = snmpgo.NewPdu(snmpgo.V3, snmpgo.Report)
rmsg := snmpgo.ToMessageV3(snmpgo.NewMessage(snmpgo.V3, pdu))
rmsg.SetPduBytes(smsg.PduBytes())
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - engineId check")
}
rmsg.AuthEngineId = expEngId
rmsg.AuthEngineBoots = -1
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - boots check")
}
rmsg.AuthEngineBoots = 1
rmsg.AuthEngineTime = -1
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - time check")
}
rmsg.AuthEngineTime = 1
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err != nil {
t.Errorf("ProcessIncomingMessage() - has error %v", err)
}
if !bytes.Equal(sec.AuthEngineId, expEngId) {
t.Errorf("ProcessIncomingMessage() - expected [%s], actual [%s]",
sec.AuthEngineId, expEngId)
}
if len(sec.AuthKey) == 0 {
t.Error("ProcessIncomingMessage() - authKey")
}
if len(sec.PrivKey) == 0 {
t.Error("ProcessIncomingMessage() - privKey")
}
// Synchronize
smsg.SetAuthentication(true)
smsg.SetPrivacy(true)
err = sec.GenerateRequestMessage(snmp, smsg)
if err != nil {
t.Errorf("GenerateRequestMessage() - has error %v", err)
}
if !bytes.Equal(smsg.UserName, expUser) {
t.Errorf("GenerateRequestMessage() - expected [%s], actual [%s]",
expUser, smsg.UserName)
}
if !bytes.Equal(smsg.AuthEngineId, expEngId) {
t.Errorf("GenerateRequestMessage() - expected [%s], actual [%s]",
expEngId, smsg.AuthEngineId)
}
if len(smsg.PrivParameter) == 0 {
t.Error("GenerateRequestMessage() - privParameter")
}
if len(smsg.AuthParameter) == 0 {
t.Error("GenerateRequestMessage() - authParameter")
}
pdu = snmpgo.NewPdu(snmpgo.V3, snmpgo.Report)
rmsg = snmpgo.ToMessageV3(snmpgo.NewMessage(snmpgo.V3, pdu))
rmsg.SetAuthentication(true)
rmsg.SetPrivacy(true)
rmsg.SetPduBytes(smsg.PduBytes())
rmsg.AuthEngineId = []byte("foobar")
rmsg.AuthEngineBoots = smsg.AuthEngineBoots
rmsg.AuthEngineTime = smsg.AuthEngineTime
rmsg.PrivParameter = smsg.PrivParameter
rmsg.AuthParameter = smsg.AuthParameter
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - userName check")
}
rmsg.UserName = expUser
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - authEngine check")
}
rmsg.AuthEngineId = expEngId
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err != nil {
t.Errorf("ProcessIncomingMessage() - has error %v", err)
}
if sec.AuthEngineBoots != rmsg.AuthEngineBoots {
t.Error("ProcessIncomingMessage() - engineBoots")
}
if sec.AuthEngineTime != rmsg.AuthEngineTime {
t.Error("ProcessIncomingMessage() - engineTime")
}
// Request
sec.AuthEngineBoots = 1
sec.AuthEngineTime = 1
err = sec.GenerateRequestMessage(snmp, smsg)
if err != nil {
t.Errorf("GenerateRequestMessage() - has error %v", err)
}
if smsg.AuthEngineBoots != sec.AuthEngineBoots {
t.Errorf("GenerateRequestMessage() - expected [%d], actual [%d]",
sec.AuthEngineBoots, smsg.AuthEngineBoots)
}
if smsg.AuthEngineTime != sec.AuthEngineTime {
t.Errorf("GenerateRequestMessage() - expected [%d], actual [%d]",
sec.AuthEngineTime, smsg.AuthEngineTime)
}
pdu = snmpgo.NewPdu(snmpgo.V3, snmpgo.GetResponse)
spdu = pdu.(*snmpgo.ScopedPdu)
rmsg = snmpgo.ToMessageV3(snmpgo.NewMessage(snmpgo.V3, pdu))
rmsg.AuthEngineId = expEngId
rmsg.AuthEngineBoots = smsg.AuthEngineBoots
rmsg.AuthEngineTime = smsg.AuthEngineTime
rmsg.UserName = expUser
// set PduBytes with GetResponse
b, _ := spdu.Marshal()
rmsg.SetPduBytes(b)
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - contextEngineId check")
}
// set PduBytes with ContextEngineId
spdu.ContextEngineId = expCtxId
b, _ = spdu.Marshal()
rmsg.SetPduBytes(b)
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - contextName check")
}
// set PduBytes with ContextName
spdu.ContextName = []byte(expCtxName)
b, _ = spdu.Marshal()
rmsg.SetPduBytes(b)
err = sec.ProcessIncomingMessage(snmp, smsg, rmsg)
if err == nil {
t.Error("ProcessIncomingMessage() - response authenticate check")
}
}