]> git.stg.codes - ssmd.git/blob - src/acl.cpp
Control dumping snmp scripts.
[ssmd.git] / src / acl.cpp
1 #include <boost/lexical_cast.hpp>
2 #include <boost/format.hpp>
3
4 #include "snmp_pp/snmp_pp.h"
5
6 #include "acl.h"
7 #include "oids.h"
8 #include "logger.h"
9
10 using SSMD::ACL;
11
12 ACL::ACL(unsigned id,
13          unsigned profileId,
14          const std::string & mac,
15          unsigned port,
16          unsigned shape,
17          unsigned burst,
18          bool isUpload)
19     : _id(id),
20       _profileId(profileId),
21       _mac(mac),
22       _shape(shape),
23       _burst(burst),
24       _isUpload(isUpload)
25 {
26     port = 1 << (32 - port);
27     _port = (boost::format("%|08x|") % port).str();
28 }
29
30 ACL::ACL(const ACL & rvalue)
31     : _id(rvalue._id),
32       _profileId(rvalue._profileId),
33       _mac(rvalue._mac),
34       _port(rvalue._port),
35       _shape(rvalue._shape),
36       _burst(rvalue._burst),
37       _isUpload(rvalue._isUpload)
38 {
39 }
40
41 ACL::~ACL()
42 {
43 }
44
45 ACL & ACL::operator=(const ACL & rvalue)
46 {
47     _id = rvalue._id;
48     _profileId = rvalue._profileId;
49     _mac = rvalue._mac;
50     _port = rvalue._port;
51     _shape = rvalue._shape;
52     _burst = rvalue._burst;
53     _isUpload = rvalue._isUpload;
54
55     return *this;
56 }
57
58 void ACL::appendPdu(Pdu & pdu) const
59 {
60     std::string oidValue;
61     // MAC
62     if (_isUpload) {
63         oidValue = swACLEtherRuleSrcMacAddress;
64         oidValue += getSuffix();
65     } else {
66         oidValue = swACLEtherRuleDstMacAddress;
67         oidValue += getSuffix();
68     }
69     Vb vb(Oid(oidValue.c_str()));
70     vb.set_value(OctetStr::from_hex_string(_mac.c_str()));
71     pdu += vb;
72
73     // Permit rule
74     oidValue = swACLEtherRulePermit;
75     oidValue += getSuffix();
76     vb.set_oid(Oid(oidValue.c_str()));
77     vb.set_value(int(2));
78     pdu += vb;
79
80     // Port
81     oidValue = swACLEtherRulePort;
82     oidValue += getSuffix();
83     vb.set_oid(Oid(oidValue.c_str()));
84     vb.set_value(OctetStr::from_hex_string(_port.c_str()));
85     pdu += vb;
86
87     // Shape
88     oidValue = swACLEtherRuleRxRate;
89     oidValue += getSuffix();
90     vb.set_oid(Oid(oidValue.c_str()));
91     vb.set_value(int(_shape));
92     pdu += vb;
93
94     // Create ACL
95     oidValue = swACLEtherRuleRowStatus;
96     oidValue += getSuffix();
97     vb.set_oid(Oid(oidValue.c_str()));
98     vb.set_value(int(4));
99     pdu += vb;
100
101     // Burst
102     /*oidValue = swACLMeterBurstSize;
103     oidValue += getSuffix();
104     vb.set_oid(Oid(oidValue.c_str()));
105     vb.set_value(int(_burst));
106     pdu += vb;*/
107 }
108
109 std::ostream & SSMD::operator<<(std::ostream & stream, const ACL & acl)
110 {
111     std::string oidValue;
112     // MAC
113     if (acl._isUpload) {
114         oidValue = swACLEtherRuleSrcMacAddress;
115         oidValue += acl.getSuffix();
116     } else {
117         oidValue = swACLEtherRuleDstMacAddress;
118         oidValue += acl.getSuffix();
119     }
120     stream << oidValue << " x " << acl._mac << " ";
121
122     // Permit rule
123     oidValue = swACLEtherRulePermit;
124     oidValue += acl.getSuffix();
125     stream << oidValue << " i " << 2 << " ";
126
127     // Port
128     oidValue = swACLEtherRulePort;
129     oidValue += acl.getSuffix();
130     stream << oidValue << " x " << acl._port << " ";
131
132     // Shape
133     oidValue = swACLEtherRuleRxRate;
134     oidValue += acl.getSuffix();
135     stream << oidValue << " i " << acl._shape << " ";
136
137     // Create ACL
138     oidValue = swACLEtherRuleRowStatus;
139     oidValue += acl.getSuffix();
140     stream << oidValue << " i " << 4;
141
142     return stream;
143 }
144
145 inline
146 std::string ACL::getSuffix() const
147 {
148     return std::string(".") +
149            boost::lexical_cast<std::string>(_profileId) +
150            "." +
151            boost::lexical_cast<std::string>(_id);
152 }