Compare commits

...

18 commits
master ... dev

Author SHA1 Message Date
1f3730ac87 reformat 2016-02-25 19:46:35 +01:00
f15abc71be lookup clean up 2016-02-25 01:47:14 +01:00
69198cd6fe 1.2 2016-02-23 21:43:35 +01:00
e494a7db91 trying out stuff 2016-02-23 20:51:11 +01:00
507cfe2140 trying out stuff 2016-02-23 15:54:56 +01:00
a90401ddeb trying out stuff 2016-02-23 12:24:49 +01:00
49ab8a307e interactive shell wip 2016-02-06 02:35:47 +01:00
5b7014737a interactive shell wip 2016-02-06 02:32:17 +01:00
59796d600b clean up 2016-01-28 00:38:40 +01:00
164edcdad9 implementing -I, -s and reset 2016-01-27 23:54:19 +01:00
a42a0aa8ce repair 2016-01-23 10:05:59 +01:00
2952953f5d repair 2016-01-23 09:55:37 +01:00
0c289aff9c clean up 2016-01-23 09:53:18 +01:00
8200103514 get&set 2016-01-23 09:22:46 +01:00
d3b89e3b11 clean up 2016-01-23 04:25:55 +01:00
693d8d42ff Program.cpp update 2016-01-22 23:51:17 +01:00
922ac9f88e updates 2016-01-21 05:20:21 +01:00
057a7b4344 updates 2016-01-20 20:56:00 +01:00
37 changed files with 1602 additions and 1153 deletions

View file

@ -15,8 +15,7 @@ $(BUILDDIR):
$(TARGET): $(OBJECTS) $(TARGET): $(OBJECTS)
$(CC) $^ -o $(BUILDDIR)/$@ -lboost_filesystem -lboost_system $(CC) $^ -o $(BUILDDIR)/$@ -lboost_filesystem -lboost_system -lreadline -lboost_program_options
$(OBJECTS): $(BUILDDIR)/%.o : $(SOURCEDIR)/%.cpp $(OBJECTS): $(BUILDDIR)/%.o : $(SOURCEDIR)/%.cpp
$(CC) $(CFLAGS) $< -o $@ $(CC) $(CFLAGS) $< -o $@

View file

@ -3,8 +3,10 @@ Command-Line Tool which might in the future be able to configure TP-Link Easy Sm
The usage is loosely based on the swconfig utility. The usage is loosely based on the swconfig utility.
### dependencys ### dependencys
libboost-filesystem-dev libboost-filesystem
libboost-system-dev libboost-system
libboost-program-options
libreadline
### compatible devices ### compatible devices
+ TL-SG105E + TL-SG105E

View file

@ -11,41 +11,48 @@
#include "Types.h" #include "Types.h"
#define VERSION "smrtlink (v1.1 Linux)\n" #define VERSION "smrtlink (v1.1 Linux)\n"
#define USAGE "usage: %s [-bdhjrvswx] [-i interface] [-u [password:]username]\n\ #define USAGE "usage: %s [-bdhjrvVstwx] [-i interface] [-U [password:]username]\n\
[-p password] <command>\n\n" [-P password] <command>\n\n"
#define HELP "\ #define HELP "\
Option Summary:\n\ Option Summary:\n\
-h --help This help text\n\ -h --help This help text\n\
-v --version Display version of this tool\n\ -V --version Display version of this tool\n\
-d --debug [n] Show debugging messages\n\ -d --debug [n] Show debugging messages\n\
-r switch ports to emulate switch while sniffing\n\ -v --verbose [n] Show debugging messages\n\
-b --header Show header\n\ \
-x --hex Display Packets as Hex String\n\ -r --reverse switch ports to emulate switch while sniffing\n\
-j --json Display Packets as JSON\n\ -i --interface <iface> only use one Interface\n\
-i --interface <iface> only use one Interface\n\ -t --timeout <n> Timeout in milliseconds. Default: 180\n\
-u --user <[password:]username>\n\ -w --wait Not yet implemented: blocking until operation is completed\n\
-p --password <password>\n\ -s --permanent make changes immediately permanent\n\
Login with Username and Password\n\ \
-f --file <path> choose a settings file\n\n\ -b --header Show header\n\
-t --timeout <n> Timeout in milliseconds. Default: 180\n\ -t --text Display Packets as readable Text (default)\n\
-I --interactive Not yet implemented\n\ -x --hex Display Packets as Hex String\n\
-O --stdout Not yet implemented\n\ -j --json Display Packets as JSON\n\n\
-w --wait Not yet implemented: blocking until operation is completed\n\ \
-s --permanent Not yet implemented: make changes immediately permanent\n\n\ -U --user <[password:]username>\n\
Command Summary:\n\ -P --password <password>\n\
help This help text\n\ Login with Username and Password\n\
list list all connected switches\n\ -f --file <path> choose a settings file\n\
sniff [type:<type>] [<filter>]\n\ -I --stdin Not yet implemented\n\
capture and display all incoming or outgoing packets\n\ -O --stdout Not yet implemented\n\
depending on the --reverse option\n\ -X --interactive gives in interactive Shell. Ignores all commands\n\n\
encode use encoding algorithm on hex data separated by colon\n\ Command Summary:\n\
get Not yet implemented\n\ list list all connected switches\n\
set Not yet implemented\n\ sniff [type:<type>] [<filter>]\n\
get|save Not yet implemented: save config to file\n\ capture and display all incoming or outgoing packets\n\
set|restore Not yet implemented: restore onfig from file\n\ depending on the --reverse option\n\
flash Not yet implemented: replace firmware\n\ encode use encoding algorithm on hex data separated by colon\n\
reboot Not yet implemented\n\ get <property1> [<property1> [...]]\n\
reset Not yet implemented\n\n\ Not yet implemented\n\
set <property1>=<value> [<property2>=<value> [...]]\n\
Not yet implemented\n\
save Not yet implemented: save config to file\n\
restore Not yet implemented: restore onfig from file\n\
flash Not yet implemented: replace firmware\n\
reboot Not yet implemented\n\
reset Not yet implemented\n\n\
### for questions please contact <smrtlink@jdi.li> ###\n\n" ### for questions please contact <smrtlink@jdi.li> ###\n\n"
//TODO //TODO
/* /*
@ -54,13 +61,6 @@
* *
*/ */
#define FLAG_HEX 1
#define FLAG_REVERSE 2
#define FLAG_HEADER 4
#define FLAG_PERMANENT 8
#define FLAG_WAIT 16
#define FLAG_DEBUG 32
extern Options options; extern Options options;
#endif /* OPTIONS_H_ */ #endif /* OPTIONS_H_ */

View file

@ -10,31 +10,32 @@
#include "Constant.h" #include "Constant.h"
std::string File::read() { std::string File::read() {
if (!fs::exists(home)) { if (!fs::exists(home)) {
fs::create_directory(home); fs::create_directory(home);
} }
fs::ifstream in(((options.file=="")?home / DEFAULT_FILE:options.file), std::ios::in | std::ios::binary); fs::ifstream in(((options.file == "") ? home / DEFAULT_FILE : options.file),
if (in) { std::ios::in | std::ios::binary);
std::string contents; if (in) {
in.seekg(0, std::ios::end); std::string contents;
contents.resize(in.tellg()); in.seekg(0, std::ios::end);
in.seekg(0, std::ios::beg); contents.resize(in.tellg());
in.read(&contents[0], contents.size()); in.seekg(0, std::ios::beg);
in.close(); in.read(&contents[0], contents.size());
return (contents); in.close();
} return (contents);
return ""; }
return "";
} }
int File::write(std::string content) { int File::write(std::string content) {
if (!fs::exists(home)) { if (!fs::exists(home)) {
fs::create_directory(home); fs::create_directory(home);
} }
fs::path p = (options.file=="")?home / DEFAULT_FILE:options.file; fs::path p = (options.file == "") ? home / DEFAULT_FILE : options.file;
fs::ofstream file(p); fs::ofstream file(p);
file << content<<"\n"; file << content << "\n";
file.close(); file.close();
return 0; return 0;
} }

View file

@ -17,14 +17,14 @@ namespace fs = boost::filesystem;
#define DEFAULT_FILE "config.json" #define DEFAULT_FILE "config.json"
class File { class File {
public: public:
File() { File() {
home = fs::path(getenv("HOME")) / ".smrtlink"; home = fs::path(getenv("HOME")) / ".smrtlink";
} }
std::string read(); std::string read();
int write(std::string); int write(std::string);
private: private:
fs::path home; fs::path home;
}; };
#endif /* FILE_H_ */ #endif /* FILE_H_ */

54
src/Filter.cpp Normal file
View file

@ -0,0 +1,54 @@
/*
* Filter.cpp
*
* Created on: 23.02.2016
* Author: jedi
*/
#include "Filter.h"
Filter::Filter() {
}
Filter::Filter(macAddr a) {
switchMac = a;
}
Filter::Filter(Packet::OpCode o) {
oc = o;
}
Filter Filter::opcode(Packet::OpCode o) {
oc = o;
return *this;
}
Filter Filter::mac(macAddr a) {
switchMac = a;
return *this;
}
Filter Filter::tokenid(short i) {
tokenId = i;
return *this;
}
bool Filter::pass(const Packet p) const {
macAddr none {
0, 0, 0, 0, 0, 0 };
if (this->oc != Packet::NONE && this->oc != p.getOpCode())
return false;
if (this->switchMac != none && this->switchMac != p.getSwitchMac())
return false;
if (this->tokenId != -1 && this->tokenId != p.getTokenId())
return false;
/* if (fragmentOffset != p.getF && fragmentOffset != -1)
return false;*/
return true;
}
bool Filter::operator<(const Filter f) const {
if (this->oc != f.oc)
return this->oc > f.oc;
if (this->switchMac != f.switchMac)
return this->switchMac > f.switchMac;
if (this->tokenId != f.tokenId)
return this->tokenId > f.tokenId;
return false;
}

31
src/Filter.h Normal file
View file

@ -0,0 +1,31 @@
/*
* Filter.h
*
* Created on: 23.02.2016
* Author: jedi
*/
#ifndef FILTER_H_
#define FILTER_H_
#include "Packet.h"
class Filter {
public:
Filter();
Filter(Packet::OpCode);
Filter(macAddr);
Filter opcode(Packet::OpCode);
Filter mac(macAddr);
Filter tokenid(short);
bool pass(const Packet) const;
bool operator<(const Filter) const;
private:
Packet::OpCode oc = Packet::NONE;
macAddr switchMac = {
0, 0, 0, 0, 0, 0 };
short tokenId = -1;
short fragmentOffset = -1;
};
#endif /* FILTER_H_ */

View file

@ -26,77 +26,79 @@
#include "Types.h" #include "Types.h"
macAddr Host::getMac() { macAddr Host::getMac() {
int s; int s;
struct ifreq buffer; struct ifreq buffer;
macAddr data { 0, 0, 0, 0, 0, 0 }; macAddr data {
if (options.interface != "") { 0, 0, 0, 0, 0, 0 };
s = socket(PF_INET, SOCK_DGRAM, 0); if (options.interface != "") {
memset(&buffer, 0x00, sizeof(buffer)); s = socket(PF_INET, SOCK_DGRAM, 0);
strcpy(buffer.ifr_name, options.interface.c_str()); memset(&buffer, 0x00, sizeof(buffer));
ioctl(s, SIOCGIFHWADDR, &buffer); strcpy(buffer.ifr_name, options.interface.c_str());
close(s); ioctl(s, SIOCGIFHWADDR, &buffer);
memcpy(&data[0], &buffer.ifr_hwaddr.sa_data[0], 6); close(s);
} memcpy(&data[0], &buffer.ifr_hwaddr.sa_data[0], 6);
return data; }
return data;
} }
ipAddr Host::getIp() { ipAddr Host::getIp() {
struct ifaddrs *ifaddr, *ifa; struct ifaddrs *ifaddr, *ifa;
int n; int n;
ipAddr data { 0, 0, 0, 0 }; ipAddr data {
if (getifaddrs(&ifaddr) == -1) { 0, 0, 0, 0 };
perror("getifaddrs"); if (getifaddrs(&ifaddr) == -1) {
exit(EXIT_FAILURE); perror("getifaddrs");
} exit(EXIT_FAILURE);
for (ifa = ifaddr, n = 0; ifa != NULL; ifa = ifa->ifa_next, n++) { }
if (ifa->ifa_addr == NULL) for (ifa = ifaddr, n = 0; ifa != NULL; ifa = ifa->ifa_next, n++) {
continue; if (ifa->ifa_addr == NULL)
if (ifa->ifa_addr->sa_family == AF_INET) continue;
if (options.interface.compare(ifa->ifa_name) == 0) { if (ifa->ifa_addr->sa_family == AF_INET)
memcpy(&data[0], &ifa->ifa_addr->sa_data[2], 4); if (options.interface.compare(ifa->ifa_name) == 0) {
return data; memcpy(&data[0], &ifa->ifa_addr->sa_data[2], 4);
} return data;
} }
/* }
for (ifa = ifaddr, n = 0; ifa != NULL; ifa = ifa->ifa_next, n++) { /*
if (ifa->ifa_addr == NULL) for (ifa = ifaddr, n = 0; ifa != NULL; ifa = ifa->ifa_next, n++) {
continue; if (ifa->ifa_addr == NULL)
if (ifa->ifa_addr->sa_family == AF_INET) continue;
if (getIface().compare(ifa->ifa_name) == 0) { if (ifa->ifa_addr->sa_family == AF_INET)
memcpy(&data[0], &ifa->ifa_addr->sa_data[2], 4); if (getIface().compare(ifa->ifa_name) == 0) {
return data; memcpy(&data[0], &ifa->ifa_addr->sa_data[2], 4);
} return data;
} }
*/ }
freeifaddrs(ifaddr); */
return data; freeifaddrs(ifaddr);
return data;
} }
std::string Host::getIface() { std::string Host::getIface() {
std::string defaultInterface; std::string defaultInterface;
std::ifstream routeFile("/proc/net/route", std::ios_base::in); std::ifstream routeFile("/proc/net/route", std::ios_base::in);
if (!routeFile.good()) if (!routeFile.good())
return ""; return "";
std::string line; std::string line;
std::vector<std::string> tokens; std::vector<std::string> tokens;
if (std::getline(routeFile, line)) if (std::getline(routeFile, line))
while (std::getline(routeFile, line)) { while (std::getline(routeFile, line)) {
std::istringstream stream(line); std::istringstream stream(line);
std::string delimiter = "\t"; std::string delimiter = "\t";
size_t pos = 0; size_t pos = 0;
std::string token; std::string token;
while ((pos = line.find(delimiter)) != std::string::npos) { while ((pos = line.find(delimiter)) != std::string::npos) {
token = line.substr(0, pos); token = line.substr(0, pos);
tokens.push_back(token); tokens.push_back(token);
line.erase(0, pos + delimiter.length()); line.erase(0, pos + delimiter.length());
} }
if ((tokens.size() >= 2) if ((tokens.size() >= 2)
&& (tokens[1] == std::string("00000000"))) { && (tokens[1] == std::string("00000000"))) {
defaultInterface = tokens[0]; defaultInterface = tokens[0];
break; break;
} }
tokens.clear(); tokens.clear();
} }
routeFile.close(); routeFile.close();
return defaultInterface; return defaultInterface;
} }

View file

@ -11,11 +11,12 @@
#include "Types.h" #include "Types.h"
class Host { class Host {
public: public:
Host(){}; Host() {
macAddr getMac(); }
ipAddr getIp(); macAddr getMac();
std::string getIface(); ipAddr getIp();
std::string getIface();
}; };
#endif /* HOST_H_ */ #endif /* HOST_H_ */

46
src/Interactive.cpp Normal file
View file

@ -0,0 +1,46 @@
/*
* Interactive.cpp
*
* Created on: 27.01.2016
* Author: jedi
*/
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include<boost/program_options/parsers.hpp>
#include<readline/readline.h>
#include<readline/history.h>
#include "Program.h"
#include "Interactive.h"
using namespace std;
int Interactive::loop() {
string cmd;
vector<string> v;
Program p = Program();
p.init();
while (1) {
cmd = readline("smrtlink> ");
if (!cmd.compare("quit") || !cmd.compare("q"))
return 0;
if (!cmd.empty()) {
add_history(cmd.c_str());
v = boost::program_options::split_unix(cmd);
p.run(v);
}
}
return 0;
}
int Interactive::single(vector<string> v) {
Program p = Program();
p.init();
p.run(v);
return 0;
}

21
src/Interactive.h Normal file
View file

@ -0,0 +1,21 @@
/*
* Interactive.h
*
* Created on: 27.01.2016
* Author: jedi
*/
#ifndef INTERACTIVE_H_
#define INTERACTIVE_H_
class Interactive {
public:
Interactive() {
}
virtual ~Interactive() {
}
int loop();
int single(std::vector<std::string> v);
};
#endif /* INTERACTIVE_H_ */

View file

@ -1,66 +0,0 @@
/*
* Options.h
*
* Created on: 11.09.2015
* Author: jdi
*/
#ifndef OPTIONS_H_
#define OPTIONS_H_
#include "Types.h"
#define VERSION "smrtlink (v1 Linux)\n"
#define USAGE "usage: %s [-bdhjrvswx] [-i interface] [-u [password:]username]\n\
[-p password] <command>\n\n"
#define HELP "\
Option Summary:\n\
-h --help This help text\n\
-v --version Display version of this tool\n\
-d --debug [n] Show debugging messages\n\
-r switch ports to emulate switch while sniffing\n\
-b --header Show header\n\
-x --hex Display Packets as Hex String\n\
-j --json Display Packets as JSON\n\
-i --interface <iface> only use one Interface\n\
-u --user <[password:]username>\n\
-p --password <password>\n\
Login with Username and Password\n\
-f --file <path> choose a settings file\n\n\
-t --timeout <n> Timeout in milliseconds. Default: 180\n\
-I --interactive Not yet implemented\n\
-O --stdout Not yet implemented\n\
-w --wait Not yet implemented: blocking until operation is completed\n\
-s --permanent Not yet implemented: make changes immediately permanent\n\n\
Command Summary:\n\
help This help text\n\
list list all connected switches\n\
sniff [type:<type>] [<filter>]\n\
capture and display all incoming or outgoing packets\n\
depending on the --reverse option\n\
encode use encoding algorithm on hex data separated by colon\n\
get Not yet implemented\n\
set Not yet implemented\n\
get|save Not yet implemented: save config to file\n\
set|restore Not yet implemented: restore onfig from file\n\
flash Not yet implemented: replace firmware\n\
reboot Not yet implemented\n\
reset Not yet implemented\n\n\
### for questions please contact <smrtlink@jdi.li> ###\n\n"
//TODO
/*
* Stdin
* socketmode
*
*/
#define FLAG_HEX 1
#define FLAG_REVERSE 2
#define FLAG_HEADER 4
#define FLAG_PERMANENT 8
#define FLAG_WAIT 16
#define FLAG_DEBUG 32
extern Options options;
#endif /* OPTIONS_H_ */

View file

@ -150,42 +150,43 @@ void Packet::setTokenId(short tokenId) {
std::string Packet::opCodeToString() { std::string Packet::opCodeToString() {
switch (opCode) { switch (opCode) {
case DISCOVERY: case DISCOVERY:
return "DISCOVERY"; return "DISCOVERY";
case GET: case GET:
return "GET"; return "GET";
case SET: case SET:
return "SET"; return "SET";
case CONFIRM: case CONFIRM:
return "CONFIRM"; return "CONFIRM";
case RETURN: case REPLY:
return "RETURN"; return "REPLY";
default: default:
return "NONE"; return "NONE";
} }
return "NONE"; return "NONE";
} }
void Packet::encode(bytes &data) { void Packet::encode(bytes &data) {
int len = data.size(); int len = data.size();
bytes key = { 191, 155, 227, 202, 99, 162, 79, 104, 49, 18, 190, 164, 30, bytes key = {
76, 189, 131, 23, 52, 86, 106, 207, 125, 126, 169, 196, 28, 172, 58, 191, 155, 227, 202, 99, 162, 79, 104, 49, 18, 190, 164, 30, 76, 189,
188, 132, 160, 3, 36, 120, 144, 168, 12, 231, 116, 44, 41, 97, 108, 131, 23, 52, 86, 106, 207, 125, 126, 169, 196, 28, 172, 58, 188,
213, 42, 198, 32, 148, 218, 107, 247, 112, 204, 14, 66, 68, 91, 224, 132, 160, 3, 36, 120, 144, 168, 12, 231, 116, 44, 41, 97, 108, 213,
206, 235, 33, 130, 203, 178, 1, 134, 199, 78, 249, 123, 7, 145, 73, 42, 198, 32, 148, 218, 107, 247, 112, 204, 14, 66, 68, 91, 224, 206,
208, 209, 100, 74, 115, 72, 118, 8, 22, 243, 147, 64, 96, 5, 87, 60, 235, 33, 130, 203, 178, 1, 134, 199, 78, 249, 123, 7, 145, 73, 208,
113, 233, 152, 31, 219, 143, 174, 232, 153, 245, 158, 254, 70, 170, 209, 100, 74, 115, 72, 118, 8, 22, 243, 147, 64, 96, 5, 87, 60, 113,
75, 77, 215, 211, 59, 71, 133, 214, 157, 151, 6, 46, 81, 94, 136, 233, 152, 31, 219, 143, 174, 232, 153, 245, 158, 254, 70, 170, 75,
166, 210, 4, 43, 241, 29, 223, 176, 67, 63, 186, 137, 129, 40, 248, 77, 215, 211, 59, 71, 133, 214, 157, 151, 6, 46, 81, 94, 136, 166,
255, 55, 15, 62, 183, 222, 105, 236, 197, 127, 54, 179, 194, 229, 210, 4, 43, 241, 29, 223, 176, 67, 63, 186, 137, 129, 40, 248, 255,
185, 37, 90, 237, 184, 25, 156, 173, 26, 187, 220, 2, 225, 0, 240, 55, 15, 62, 183, 222, 105, 236, 197, 127, 54, 179, 194, 229, 185,
50, 251, 212, 253, 167, 17, 193, 205, 177, 21, 181, 246, 82, 226, 37, 90, 237, 184, 25, 156, 173, 26, 187, 220, 2, 225, 0, 240, 50,
38, 101, 163, 182, 242, 92, 20, 11, 95, 13, 230, 16, 121, 124, 109, 251, 212, 253, 167, 17, 193, 205, 177, 21, 181, 246, 82, 226, 38,
195, 117, 39, 98, 239, 84, 56, 139, 161, 47, 201, 51, 135, 250, 10, 101, 163, 182, 242, 92, 20, 11, 95, 13, 230, 16, 121, 124, 109, 195,
19, 150, 45, 111, 27, 24, 142, 80, 85, 83, 234, 138, 216, 57, 93, 117, 39, 98, 239, 84, 56, 139, 161, 47, 201, 51, 135, 250, 10, 19,
65, 154, 141, 122, 34, 140, 128, 238, 88, 89, 9, 146, 171, 149, 53, 150, 45, 111, 27, 24, 142, 80, 85, 83, 234, 138, 216, 57, 93, 65,
102, 61, 114, 69, 217, 175, 103, 228, 35, 180, 252, 200, 192, 165, 154, 141, 122, 34, 140, 128, 238, 88, 89, 9, 146, 171, 149, 53, 102,
159, 221, 244, 110, 119, 48 }; 61, 114, 69, 217, 175, 103, 228, 35, 180, 252, 200, 192, 165, 159,
221, 244, 110, 119, 48 };
bytes s = key; bytes s = key;
int i, j = 0; int i, j = 0;
for (int k = 0; k < len; k++) { for (int k = 0; k < len; k++) {
@ -199,7 +200,8 @@ void Packet::encode(bytes &data) {
void Packet::push(bytes &arr, int &index, byte data) { void Packet::push(bytes &arr, int &index, byte data) {
if (arr.size() > (unsigned) index) { if (arr.size() > (unsigned) index) {
arr[index++] = data; arr[index++] = data;
} else { }
else {
arr.push_back(data); arr.push_back(data);
index++; index++;
} }
@ -277,6 +279,14 @@ void Packet::pull(bytes &arr, int &index, int &ret) {
ret |= arr[index++] & 0xFF; ret |= arr[index++] & 0xFF;
} }
byte Packet::getOpCode() const {
return opCode;
}
void Packet::setOpCode(byte opCode) {
this->opCode = opCode;
}
void Packet::pull(bytes &arr, int &index, dataset &ret) { void Packet::pull(bytes &arr, int &index, dataset &ret) {
pull(arr, index, ret.type); pull(arr, index, ret.type);
pull(arr, index, ret.len); pull(arr, index, ret.len);

View file

@ -13,65 +13,71 @@
#include "Types.h" #include "Types.h"
static short sequenceId=0; static short sequenceId = 0;
class Packet class Packet {
{ public:
public: enum OpCode {
enum OpCode {DISCOVERY, GET, RETURN, SET, CONFIRM, NONE}; DISCOVERY, GET, REPLY, SET, CONFIRM, NONE
Packet(OpCode); };
void encode(bytes&); Packet(OpCode);
bytes getBytes(); static void encode(bytes&);
void parse(bytes); bytes getBytes();
void printHeader(); void parse(bytes);
std::string opCodeToString(); void printHeader();
short getLength() const; std::string opCodeToString();
int getCheckSum() const; short getLength() const;
int getErrorCode() const; int getCheckSum() const;
short getSequenceId() const; int getErrorCode() const;
macAddr getSwitchMac() const; short getSequenceId() const;
const bytes& getBody() const; macAddr getSwitchMac() const;
const bytes& getHead() const; const bytes& getBody() const;
const datasets& getPayload() const; const bytes& getHead() const;
void setBody(bytes); const datasets& getPayload() const;
void setHostMac(macAddr); void setBody(bytes);
void setSwitchMac(macAddr); void setHostMac(macAddr);
void setCheckSum(int); void setSwitchMac(macAddr);
void setSequenceId(short ); void setCheckSum(int);
void setPayload(datasets payload); void setSequenceId(short);
short getTokenId() const; void setPayload(datasets payload);
void setTokenId(short tokenId = 0); short getTokenId() const;
void setTokenId(short tokenId = 0);
byte getOpCode() const;
void setOpCode(byte opCode);
private: private:
bytes head = bytes(32); bytes head = bytes(32);
bytes body = bytes(0); bytes body = bytes(0);
datasets payload; datasets payload;
byte version = 1; byte version = 1;
byte opCode; byte opCode;
macAddr switchMac { {0, 0, 0, 0, 0, 0}}; macAddr switchMac {
macAddr hostMac { {0, 0, 0, 0, 0, 0}}; { 0, 0, 0, 0, 0, 0 } };
macAddr broadcastMac { {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF}}; macAddr hostMac {
short tokenId = 0; { 0, 0, 0, 0, 0, 0 } };
short fragmentOffset = 0; macAddr broadcastMac {
int errorCode = 0; { 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF } };
int checkSum = 0; short tokenId = 0;
short flag = 0; short fragmentOffset = 0;
void buildHead(); int errorCode = 0;
void buildBody(); int checkSum = 0;
void push(bytes&, int&, short ); short flag = 0;
void push(bytes&, int&, int); void buildHead();
void push(bytes&, int&, byte); void buildBody();
void push(bytes&, int&, bytes); void push(bytes&, int&, short);
void push(bytes&, int&, ipAddr); void push(bytes&, int&, int);
void push(bytes&, int&, macAddr); void push(bytes&, int&, byte);
void push(bytes&, int&, dataset); void push(bytes&, int&, bytes);
void pull(bytes&, int&, short &); void push(bytes&, int&, ipAddr);
void pull(bytes&, int&, int&); void push(bytes&, int&, macAddr);
void pull(bytes&, int&, byte&); void push(bytes&, int&, dataset);
void pull(bytes&, int&, bytes&, unsigned ); void pull(bytes&, int&, short &);
void pull(bytes&, int&, ipAddr&); void pull(bytes&, int&, int&);
void pull(bytes&, int&, macAddr&); void pull(bytes&, int&, byte&);
void pull(bytes&, int&, dataset&); void pull(bytes&, int&, bytes&, unsigned);
void pull(bytes&, int&, ipAddr&);
void pull(bytes&, int&, macAddr&);
void pull(bytes&, int&, dataset&);
}; };
#endif /* PACKET_H_ */ #endif /* PACKET_H_ */

View file

@ -6,6 +6,9 @@
*/ */
#include <iostream> #include <iostream>
#include <algorithm> #include <algorithm>
#include <regex>
#include <string>
#include <cstring>
#include "Constant.h" #include "Constant.h"
#include "Program.h" #include "Program.h"
@ -14,14 +17,120 @@
#include "Socket.h" #include "Socket.h"
#include "Switch.h" #include "Switch.h"
#include "Packet.h" #include "Packet.h"
#include "Filter.h"
#include "Types.h"
#include "lookup.h" #include "lookup.h"
#include "table.h" #include "table.h"
using namespace smrtlink;
using namespace std;
int Program::run(vector<string> arg) {
int optind = 0;
std::vector<std::string> vect;
std::map<std::string, std::string> ll;
std::cmatch sm;
std::string cmd = arg[optind++];
io_service->reset();
sock->clear();
switch (caseArg(cmd.c_str())) {
case caseArg("reboot"):
if (reboot())
return 1;
break;
case caseArg("reset"):
if (!reset())
return 0;
break;
case caseArg("save"):
if (save())
return 1;
break;
case caseArg("restore"):
if (restore())
return 1;
break;
case caseArg("flash"):
if (flash())
return 1;
break;
case caseArg("list"):
if (list())
return 1;
break;
case caseArg("sniff"):
if (sniff())
return 1;
break;
case caseArg("encode"):
if (optind < arg.size()) {
std::string s(arg[optind]);
optind++;
if (!encode(s))
return 1;
}
else {
fprintf(stderr, "Argument expected after encode\n");
return 1;
}
break;
case caseArg("set"):
while (optind < arg.size()) {
if (regex_match(arg[optind].c_str(), sm,
std::regex("^([a-z]+)=(.*)$"))) {
if (!lookup.exists(sm[1])) {
cerr << "Unknown argument " << arg[optind] << endl;
return 1;
}
ll.insert(
std::pair<std::string, std::string>(sm[1], sm[2]));
}
else {
cerr << "Invalid Syntax " << arg[optind] << endl;
return 1;
}
optind++;
}
if (setProperty(ll))
return 1;
break;
case caseArg("get"):
while (optind < arg.size()) {
if (regex_match(arg[optind].c_str(), sm,
std::regex("^([a-z]+)$"))) {
if (!lookup.exists(sm[1])) {
cerr << "Unknown argument " << arg[optind] << endl;
return 1;
}
vect.push_back(sm[1]);
}
else {
cerr << "Invalid argument " << arg[optind] << endl;
return 1;
}
optind++;
}
if (getProperty(vect))
return 1;
break;
default:
printf("Unknown command: %s\n", cmd.c_str());
return 1;
}
io_service->run();
return 0;
}
int printHeader(Packet p) { int printHeader(Packet p) {
if (options.flags & FLAG_HEADER) { if (options.flags.HEADER) {
if (options.flags & FLAG_HEX) { if (options.flags.HEX) {
std::cout << "Received Header:\n\t" << p.getHead() << "\n"; cout << "Received Header:\n\t" << p.getHead() << "\n";
} else { }
else {
p.printHeader(); p.printHeader();
printf("\n"); printf("\n");
} }
@ -30,56 +139,56 @@ int printHeader(Packet p) {
} }
int printPacket(Packet p) { int printPacket(Packet p) {
if (options.flags & FLAG_HEX) { if (options.flags.HEX) {
std::cout << "Received Payload:\n\t" << p.getBody() << "\n"; cout << "Received Payload:\n\t" << p.getBody() << "\n";
} else { }
else {
for (dataset d : p.getPayload()) { for (dataset d : p.getPayload()) {
auto lookup = //auto lookup = (options.flags.REVERSE) ? snd_lookup : rcv_lookup;
(options.flags & FLAG_REVERSE) ? snd_lookup : rcv_lookup;
if (lookup.exists(d.type)) { if (lookup.exists(d.type)) {
table::set s = lookup[d.type]; table::set s = lookup[d.type];
if (d.len > 0) { if (d.len > 0) {
switch (s.format) { switch (s.format) {
case table::STRING: case table::STRING:
std::cout << "+\t" << s.name << " = " << &d.value[0] cout << "+\t" << s.name << " = " << &d.value[0]
<< "\n"; << "\n";
break; break;
case table::BOOL: case table::BOOL:
std::cout << "+\t" << s.name << " = " cout << "+\t" << s.name << " = "
<< (d.value[0] ? "YES" : "NO") << "\n"; << (d.value[0] ? "YES" : "NO") << "\n";
break; break;
case table::HEX: case table::HEX:
std::cout << "+\t" << s.name << " = " << d.value cout << "+\t" << s.name << " = " << d.value << "\n";
<< "\n";
break; break;
case table::DEC: case table::DEC:
std::cout << "+\t" << s.name << " = "; cout << "+\t" << s.name << " = ";
if (d.value.size() > 0) if (d.value.size() > 0)
std::cout << std::dec << (unsigned) d.value[0]; cout << dec << (unsigned) d.value[0];
for (unsigned i = 1; i < d.value.size(); i++) for (unsigned i = 1; i < d.value.size(); i++)
std::cout << std::dec << "." cout << dec << "." << (unsigned) d.value[i];
<< (unsigned) d.value[i]; cout << "\n";
std::cout << "\n";
break; break;
case table::ACTION: case table::ACTION:
std::cout << "Error:" << s.name cout << "Error:" << s.name
<< " is marked as 'action' but carries payload." << " is marked as 'action' but carries payload."
<< d.value << "\n"; << d.value << "\n";
break; break;
default: default:
std::cout << "+\t" << s.name << " = " << d.value cout << "+\t" << s.name << " = " << d.value << "\n";
<< "\n";
break; break;
} }
} else { //empty
std::cout << std::dec << ">\t" << s.name << "\n";
} }
} else { //unknown id else { //empty
cout << dec << ">\t" << s.name << "\n";
}
}
else { //unknown id
if (d.len > 0) { if (d.len > 0) {
std::cout << "##\t" << d.type << ":\n\t"; cout << "##\t" << d.type << ":\n\t";
std::cout << std::hex << d.value << std::dec << "\n"; cout << hex << d.value << dec << "\n";
} else { //empty }
std::cout << "#>\t" << d.type << "\n"; else { //empty
cout << "#>\t" << d.type << "\n";
} }
} }
} }
@ -88,144 +197,130 @@ int printPacket(Packet p) {
} }
int Program::list() { int Program::list() {
std::cout << "List:\n";
Packet p = Packet(Packet::DISCOVERY);
p.setHostMac(host.getMac());
p.setPayload( { });
bytes b = p.getBytes();
p.encode(b);
try { try {
sock->setHostIp(host.getIp()); cout << "List:\n";
sock->init(DST_PORT, SRC_PORT); discover([this](Packet a) {
sock->callback = printHeader(a);
[this](Packet a) { if (options.flags.HEX) {
printHeader(a); cout <<"Received Payload:\n"<<a.getBody()<<"\n";
if (options.flags & FLAG_HEX) { }
std::cout <<"Received Payload:\n"<<a.getBody()<<"\n"; else {
} else { datasets d =a.getPayload();
datasets d =a.getPayload();
Switch sw = Switch(); int b = a.getSwitchMac().hash();
sw.parse(d); if (devices.empty()||devices.find(b) == devices.end())
File f; devices.insert(pair<int,Switch>(b,Switch()));
f.write(sw.toString()); devices[a.getSwitchMac().hash()].parse(d);
std::cout <<"\t"<<sw.settings.hostname<<" ("<< sw.device.type<<")\tMAC: "<<sw.device.mac<<"\tIP: "<<sw.settings.ip_addr<<"\n"; devices[a.getSwitchMac().hash()].print();
}
return 1; File f;
}; f.write(devices[a.getSwitchMac().hash()].toString());
sock->send(b); }
io_service->run(); return 0;
} catch (std::exception& e) { });
std::cerr << "Exception: " << e.what() << "\n"; } catch (exception& e) {
cerr << "Exception: " << e.what() << "\n";
} }
return 1; return 0;
} }
int Program::sniff() { int Program::sniff() {
printf("Listening:\n"); printf("Listening:\n");
try { try {
boost::asio::io_service io_service; sock->listen([](Packet p) {
Socket s(io_service); cout << p.opCodeToString() << "\n";
s.setHostIp(host.getIp());
s.init(DST_PORT, SRC_PORT);
s.callback = [](Packet p) {
std::cout << p.opCodeToString() << "\n";
printHeader(p); printHeader(p);
printPacket(p); printPacket(p);
return 0; return 0;
}; });
s.listen(); } catch (exception& e) {
io_service.run(); cerr << "Exception: " << e.what() << "\n";
} catch (std::exception& e) {
std::cerr << "Exception: " << e.what() << "\n";
} }
return 0;
return 1;
} }
int Program::encode(std::string s) { int Program::encode(string s) {
bytes d(s); bytes d;
Packet p = Packet(Packet::DISCOVERY); d = d.readHex(s);
Packet p = Packet(Packet::NONE);
p.encode(d); p.encode(d);
std::cout << d << std::endl; cout << d << endl;
return 0; return 0;
} }
int Program::setProperty() { int Program::setProperty(map<string, string> prop) {
return 0;
}
int Program::getProperty() {
std::cout << "List:\n";
Packet p = Packet(Packet::DISCOVERY);
p.setHostMac(host.getMac());
p.setPayload( { });
bytes b = p.getBytes();
p.encode(b);
auto s = sock;
try { try {
sock->setHostIp(host.getIp()); datasets data = { };
sock->init(DST_PORT, SRC_PORT); for (auto p : prop) {
sock->callback = string s = p.second;
[this](Packet a) { dataset d;
auto s = sock; if (lookup.exists(p.first)) {
datasets d =a.getPayload(); d.type = lookup.type(p.first);
Switch sw = Switch(); if (lookup[p.first].format == table::STRING) {
sw.parse(d); d.len = s.size() + 1;
std::cout <<"\t"<<sw.settings.hostname<<" ("<< sw.device.type<<")\tMAC: "<<sw.device.mac<<"\tIP: "<<sw.settings.ip_addr<<"\n"; d.value = s;
data.push_back(d);
}
}
}
Packet p = Packet(Packet::GET); cout << "List:\n";
p.setSwitchMac(a.getSwitchMac()); discover([this,data](Packet a) {
p.setHostMac(host.getMac()); datasets d =a.getPayload();
datasets t = { {SND_PING, 0, {}}}; Switch sw = Switch();
p.setPayload(t); sw.parse(d);
bytes c = p.getBytes(); sw.print();
p.encode(c);
sock->callback = datasets t = { {PING, 0, {}}};
[this](Packet a) { get(a, t, [this,data](Packet a) {
auto s = sock;
datasets d =a.getPayload(); datasets d =a.getPayload();
Switch sw = Switch(); Switch sw = Switch();
sw.parse(d); sw.parse(d);
Packet p = Packet(Packet::SET); set(a,data,
p.setSwitchMac(a.getSwitchMac());
p.setTokenId(a.getTokenId());
p.setHostMac(host.getMac());
bytes n = options.user;
bytes w = options.password;
n.push_back('\0');
w.push_back('\0');
datasets t = {
{ LOGIN_USER, (short)(n.size()), n},
{ LOGIN_PASSWORD, (short)(w.size()), w},
{ REBOOT, 1, {0}}
};
p.setPayload(t);
bytes c = p.getBytes();
p.encode(c);
sock->callback = [this](Packet a) {
[this](Packet a) { cout << a.opCodeToString() << "\n";
std::cout << a.opCodeToString() << "\n"; printHeader(a);
printHeader(a); printPacket(a);
printPacket(a); return 0;
return 0; });
};
sock->send(c);
return 0; return 0;
}; });
return 0;
sock->send(c); });
return 0; } catch (exception& e) {
}; cerr << "Exception: " << e.what() << "\n";
sock->send(b);
io_service->run();
} catch (std::exception& e) {
std::cerr << "Exception: " << e.what() << "\n";
} }
return 1; return 0;
}
int Program::getProperty(vector<string> prop) {
try {
datasets data = { };
for (string s : prop) {
dataset d;
d.type = lookup.type(s);
data.push_back(d);
}
cout << "List:\n";
discover([this,data](Packet a) {
datasets d =a.getPayload();
Switch sw = Switch();
sw.parse(d);
sw.print();
get(a, data, [this](Packet a) {
cout << a.opCodeToString() << "\n";
printHeader(a);
printPacket(a);
return 0;
});
return 0;
});
} catch (exception& e) {
cerr << "Exception: " << e.what() << "\n";
}
return 0;
} }
int Program::save() { int Program::save() {
@ -233,40 +328,120 @@ int Program::save() {
sw.settings.hostname = "testname.lan"; sw.settings.hostname = "testname.lan";
File f; File f;
f.write(sw.toString()); f.write(sw.toString());
return 1; return 0;
} }
int Program::restore() { int Program::restore() {
File f; File f;
Switch sw; Switch sw;
sw.parse(f.read()); sw.parse(f.read());
std::cout << "Devices:\n\t" << sw.settings.hostname << " (" sw.print();
<< sw.device.type << ")\tMAC: " << sw.device.mac << "\tIP: " return 0;
<< sw.settings.ip_addr << "\n";
return 1;
} }
int Program::flash() { int Program::flash() {
return 0; return 1;
} }
int Program::reboot() { int Program::reboot() {
try {
discover(
[this](Packet a) {
datasets d =a.getPayload();
int b = a.getSwitchMac().hash();
if (devices.empty()||devices.find(b) == devices.end())
devices.insert(pair<int,Switch>(b,Switch()));
devices[a.getSwitchMac().hash()].parse(d);
datasets t = { {PING, 0, {}}};
get(a, t, [this](Packet a) {
datasets d =a.getPayload();
cout <<devices[a.getSwitchMac().hash()].settings.hostname<<"\t";
devices[a.getSwitchMac().hash()].parse(d);
datasets t = { {REBOOT, 1, {options.flags.PERMANENT?(byte)1:(byte)0}}};
set(a,t,[this](Packet a) {
cout<< "rebooting now.\n";
return 0;
});
return 0;
});
return 0;
});
} catch (exception& e) {
cerr << "Exception: " << e.what() << "\n";
}
return 0; return 0;
} }
int Program::reset() { int Program::reset() {
try {
discover([this](Packet a) {
datasets d =a.getPayload();
Switch sw = Switch();
sw.parse(d);
cout <<sw.settings.hostname<<"\t";
datasets t = { {PING, 0, {}}};
get(a, t, [this](Packet a) {
datasets d =a.getPayload();
Switch sw = Switch();
sw.parse(d);
datasets t = { {RESET, 0, {}}};
set(a,t,[this](Packet a) {
if( a.getOpCode()==Packet::CONFIRM)
cout<< "reseting config.\n";
return 0;
});
return 0;
});
return 0;
});
} catch (exception& e) {
cerr << "Exception: " << e.what() << "\n";
}
return 0; return 0;
} }
int Program::ping(std::function<int(Packet)>) { int Program::discover(Listener c) {
Packet p = Packet(Packet::DISCOVERY);
p.setHostMac(host.getMac());
p.setPayload( { });
sock->listen(c, Filter(Packet::REPLY));
sock->send(p);
return 0;
}
int Program::get(Packet l, datasets t, Listener c) {
Packet p = Packet(Packet::GET);
p.setSwitchMac(l.getSwitchMac());
p.setHostMac(host.getMac());
p.setPayload(t);
sock->listen(c, Filter(Packet::REPLY).mac(l.getSwitchMac()));
sock->send(p);
return 0;
}
int Program::set(Packet l, datasets t, Listener c) {
Packet p = Packet(Packet::SET);
p.setSwitchMac(l.getSwitchMac());
p.setTokenId(l.getTokenId());
p.setHostMac(host.getMac());
bytes n = options.user;
bytes w = options.password;
datasets ld = {
{ LOGIN_USER, (short) (n.size()), n }, {
LOGIN_PASSWORD, (short) (w.size()), w } };
p.setPayload(ld + t);
sock->listen(c, Filter(Packet::CONFIRM).mac(l.getSwitchMac()));
sock->send(p);
return 0; return 0;
} }
void Program::init() { void Program::init() {
io_service = std::make_shared<boost::asio::io_service>();
sock = std::make_shared < Socket > (*io_service);
if (options.interface.compare("") == 0) if (options.interface.compare("") == 0)
options.interface = host.getIface(); options.interface = host.getIface();
sock->setHostIp(host.getIp());
sock->init(DST_PORT, SRC_PORT);
} }

View file

@ -9,34 +9,42 @@
#define PROGRAM_H_ #define PROGRAM_H_
#include <memory> #include <memory>
#include <vector>
#include "Types.h" #include "Types.h"
#include "Host.h" #include "Host.h"
#include "Switch.h"
#include "Socket.h" #include "Socket.h"
class Program { class Program {
private: private:
std::shared_ptr<boost::asio::io_service> io_service; std::shared_ptr<boost::asio::io_service> io_service;
std::shared_ptr<Socket> sock; std::shared_ptr<Socket> sock;
Host host = Host(); Host host = Host();
int ping(std::function<int(Packet)>); std::map<int, Switch> devices;
public: int get(Packet, datasets, std::function<int(Packet)>);
Program() { int set(Packet, datasets, std::function<int(Packet)>);
io_service = std::make_shared<boost::asio::io_service>(); int discover(std::function<int(Packet)>);
sock = std::make_shared<Socket>(*io_service); public:
} Program() {
void init(); }
int list(); void init();
int sniff(); int run(std::vector<std::string>);
int encode(std::string); std::function<int()> callback = []() {
int setProperty(); return 0;
int getProperty(); };
int save();
int restore(); int list();
int flash(); int sniff();
int reboot(); int encode(std::string);
int reset(); int getProperty(std::vector<std::string>);
std::string input; int setProperty(std::map<std::string, std::string>);
int save();
int restore();
int flash();
int reboot();
int reset();
std::string input;
}; };
#endif /* PROGRAM_H_ */ #endif /* PROGRAM_H_ */

View file

@ -5,29 +5,33 @@
* Author: jdi * Author: jdi
*/ */
#include <cstdlib> #include <cstdlib>
#include <array> #include <map>
#include <unistd.h> #include <unistd.h>
#include "Socket.h" #include "Socket.h"
#include "Packet.h" #include "Packet.h"
#include "Constant.h" #include "Constant.h"
#include "Filter.h"
#include "Host.h" #include "Host.h"
#include "Types.h" #include "Types.h"
using namespace std;
Socket::Socket(boost::asio::io_service& io_service) : Socket::Socket(boost::asio::io_service& io_service) :
send_socket_(io_service), receive_socket_(io_service), timer(io_service) { send_socket_(io_service), receive_socket_(io_service), timer(io_service) {
} }
//, resolver( io_service) //, resolver( io_service)
void Socket::init(short dst_port, short src_port) {
if (initialized)
return;
if (options.flags & FLAG_REVERSE) { Socket::~Socket() {
}
void Socket::init(short dst_port, short src_port) {
if (options.flags.REVERSE) {
short p = dst_port; short p = dst_port;
dst_port = src_port; dst_port = src_port;
src_port = p; src_port = p;
} }
if (options.flags & FLAG_DEBUG) if (options.debug_level >= 1)
std::cout << "Local IP:\t" << local_ip << "\n"; std::cout << "Local IP:\t" << local_ip << "\n";
wildcard_endpoint_ = boost::asio::ip::udp::endpoint( wildcard_endpoint_ = boost::asio::ip::udp::endpoint(
@ -47,37 +51,40 @@ void Socket::init(short dst_port, short src_port) {
receive_socket_.set_option(boost::asio::socket_base::broadcast(true)); receive_socket_.set_option(boost::asio::socket_base::broadcast(true));
receive_socket_.set_option(boost::asio::socket_base::reuse_address(true)); receive_socket_.set_option(boost::asio::socket_base::reuse_address(true));
receive_socket_.bind(wildcard_endpoint_); receive_socket_.bind(wildcard_endpoint_);
if (options.timeout != 0) {
timer.expires_from_now(
boost::posix_time::milliseconds(options.timeout));
timer.async_wait([this](const boost::system::error_code& error)
{
if (!error)
{
receive_socket_.close();
}
});
}
initialized = 1;
} }
void Socket::setHostIp(ipAddr ip) { void Socket::setHostIp(ipAddr ip) {
local_ip = ip; local_ip = ip;
} }
void Socket::send(bytes data) { void Socket::clear() {
callback.clear();
}
void Socket::listen(Listener l, Filter f) {
if (callback.find(f) == callback.end()) {
callback.insert(ListenerPair(f, l));
}
else {
callback[f] = l;
}
receive();
}
void Socket::send(Packet p) {
bytes data = p.getBytes();
p.encode(data);
unsigned char * a = &data[0]; unsigned char * a = &data[0];
send_socket_.async_send_to(boost::asio::buffer(a, data.size()), send_socket_.async_send_to(boost::asio::buffer(a, data.size()),
broadcast_endpoint_, broadcast_endpoint_,
[this](boost::system::error_code ec, std::size_t bytes_sent) [this](boost::system::error_code ec, std::size_t bytes_sent)
{ {
listen(); receive();
settimeout();
}); });
} }
void Socket::listen() { void Socket::receive() {
data.resize(MAX_LENGTH); data.resize(MAX_LENGTH);
receive_socket_.async_receive_from(boost::asio::buffer(data, MAX_LENGTH), receive_socket_.async_receive_from(boost::asio::buffer(data, MAX_LENGTH),
remote_endpoint_, remote_endpoint_,
@ -86,17 +93,43 @@ void Socket::listen() {
if (ec || bytes_recvd == 0) { if (ec || bytes_recvd == 0) {
//listen(); //listen();
// TODO distinguish error codes // TODO distinguish error codes
} else { }
else {
data.resize(bytes_recvd); data.resize(bytes_recvd);
Packet p = Packet(Packet::NONE); Packet p = Packet(Packet::NONE);
p.encode(data); p.encode(data);
// std::cout << "err" << p.getErrorCode() <<std::endl;
p.parse(data); p.parse(data);
//std::cout << "err" << p.getErrorCode() <<std::endl; for(auto r : callback) {
if(!callback(p)) { if(r.first.pass(p)) {
//TODO do something r.second(p);
// std::cout<<"pass"<<std::endl;
break;
}
else {
// std::cout<<"no pass"<<std::endl;
}
} }
listen(); receive();
settimeout();
} }
}); });
} }
void Socket::settimeout() {
if (options.timeout != 0) {
timer.expires_from_now(
boost::posix_time::milliseconds(options.timeout));
timer.async_wait(
[this](const boost::system::error_code& error)
{
if (!error)
{
receive_socket_.close();
receive_socket_.open(boost::asio::ip::udp::v4());
receive_socket_.set_option(boost::asio::socket_base::broadcast(true));
receive_socket_.set_option(boost::asio::socket_base::reuse_address(true));
receive_socket_.bind(wildcard_endpoint_);
}
});
}
}

View file

@ -8,8 +8,10 @@
#ifndef SOCKET_H_ #ifndef SOCKET_H_
#define SOCKET_H_ #define SOCKET_H_
#include <map>
#include <boost/asio.hpp> #include <boost/asio.hpp>
#include "Packet.h" #include "Packet.h"
#include "Filter.h"
#include "Types.h" #include "Types.h"
#define SRC_PORT 29809 #define SRC_PORT 29809
@ -17,30 +19,32 @@
#define MAX_LENGTH 1024 #define MAX_LENGTH 1024
class Socket { typedef std::function<int(Packet)> Listener;
public: typedef std::pair<Filter, Listener> ListenerPair;
Socket(boost::asio::io_service&);
virtual ~Socket() {
}
void init(short, short);
void send(bytes);
void listen();
void setHostIp(ipAddr);
std::function<int(Packet)> callback = [](Packet a) {
return 0;
};
private: class Socket {
boost::asio::ip::udp::socket send_socket_; public:
boost::asio::ip::udp::socket receive_socket_; Socket(boost::asio::io_service&);
boost::asio::ip::udp::endpoint broadcast_endpoint_; virtual ~Socket();
boost::asio::ip::udp::endpoint remote_endpoint_; void init(short, short);
boost::asio::ip::udp::endpoint wildcard_endpoint_; void clear();
boost::asio::ip::udp::endpoint local_endpoint_; void send(Packet);
boost::asio::deadline_timer timer; void setHostIp(ipAddr);
bytes data = bytes(MAX_LENGTH); void listen(Listener l, Filter f = Filter());
ipAddr local_ip;
int initialized = 0; private:
void receive();
void settimeout();
boost::asio::ip::udp::socket send_socket_;
boost::asio::ip::udp::socket receive_socket_;
boost::asio::ip::udp::endpoint broadcast_endpoint_;
boost::asio::ip::udp::endpoint remote_endpoint_;
boost::asio::ip::udp::endpoint wildcard_endpoint_;
boost::asio::ip::udp::endpoint local_endpoint_;
boost::asio::deadline_timer timer;
bytes data = bytes(MAX_LENGTH);
ipAddr local_ip;
std::map<Filter, Listener> callback = { };
}; };

31
src/Switch-Cmd.cpp Normal file
View file

@ -0,0 +1,31 @@
/*
* Switch.cpp
*
* Created on: 29.09.2015
* Author: jdi
*/
#include <string>
#include "Types.h"
#include "Switch.h"
#include "Constant.h"
#include "table.h"
using namespace smrtlink;
using namespace std;
int Switch::set(pair<string, string> str) {
return 0;
}
std::string Switch::get(std::string str) {
std::string ret;
switch (caseArg(str.c_str())) {
case caseArg("ip"):
ret = "0.0.0.0";
break;
}
return ret;
}

View file

@ -14,85 +14,85 @@
int Switch::parse(std::string str) { int Switch::parse(std::string str) {
if (json.Parse(str.c_str()).HasParseError()) if (json.Parse(str.c_str()).HasParseError())
return 1; return 1;
if (options.flags & FLAG_DEBUG) if (options.debug_level >= 1)
std::cout << "\nParsing to document succeeded.\n"; std::cout << "\nParsing to document succeeded.\n";
if (json.IsObject()) { if (json.IsObject()) {
if (json.HasMember("hostname")) if (json.HasMember("hostname"))
settings.hostname = json["hostname"].GetString(); settings.hostname = json["hostname"].GetString();
if (json.HasMember("type")) if (json.HasMember("type"))
device.type = json["type"].GetString(); device.type = json["type"].GetString();
if (json.HasMember("hardware_version")) if (json.HasMember("hardware_version"))
device.hardware_version = json["hardware_version"].GetString(); device.hardware_version = json["hardware_version"].GetString();
if (json.HasMember("firmware_version")) if (json.HasMember("firmware_version"))
device.hardware_version = json["firmware_version"].GetString(); device.hardware_version = json["firmware_version"].GetString();
if (json.HasMember("ports") && json["ports"].IsArray()) { if (json.HasMember("ports") && json["ports"].IsArray()) {
const rapidjson::Value& a = json["ports"]; const rapidjson::Value& a = json["ports"];
for (rapidjson::SizeType i = 0; i < a.Size(); i++) for (rapidjson::SizeType i = 0; i < a.Size(); i++)
if (a[i].IsObject()) { if (a[i].IsObject()) {
port p; port p;
if (a[i].HasMember("id") && a[i]["id"].IsInt()) { if (a[i].HasMember("id") && a[i]["id"].IsInt()) {
p.id = a[i]["id"].GetInt(); p.id = a[i]["id"].GetInt();
std::cout << a[i]["id"].GetInt() << "\n"; std::cout << a[i]["id"].GetInt() << "\n";
} }
ports.push_back(p); ports.push_back(p);
} }
} }
if (json.HasMember("vlans") && json["vlans"].IsArray()) { if (json.HasMember("vlans") && json["vlans"].IsArray()) {
const rapidjson::Value& a = json["vlans"]; const rapidjson::Value& a = json["vlans"];
for (rapidjson::SizeType i = 0; i < a.Size(); i++) for (rapidjson::SizeType i = 0; i < a.Size(); i++)
if (a[i].IsObject()) { if (a[i].IsObject()) {
vlan v; vlan v;
if (a[i].HasMember("name") && a[i]["name"].IsString()) { if (a[i].HasMember("name") && a[i]["name"].IsString()) {
v.name = a[i]["name"].GetString(); v.name = a[i]["name"].GetString();
std::cout << a[i]["name"].GetString() << "\n"; std::cout << a[i]["name"].GetString() << "\n";
} }
vlans.push_back(v); vlans.push_back(v);
} }
} }
} }
/* /*
{ {
const rapidjson::Value& a = json["a"]; const rapidjson::Value& a = json["a"];
assert(a.IsArray()); assert(a.IsArray());
for (rapidjson::SizeType i = 0; i < a.Size(); i++) for (rapidjson::SizeType i = 0; i < a.Size(); i++)
printf("a[%d] = %d\n", i, a[i].GetInt()); printf("a[%d] = %d\n", i, a[i].GetInt());
int y = a[0].GetInt(); int y = a[0].GetInt();
(void) y; (void) y;
*/ */
return 0; return 0;
} }
std::string Switch::toString() { std::string Switch::toString() {
if (!json.IsObject()) { if (!json.IsObject()) {
json.SetObject(); json.SetObject();
} }
rapidjson::Document::AllocatorType& allocator = json.GetAllocator(); rapidjson::Document::AllocatorType& allocator = json.GetAllocator();
json.AddMember("hostname", jsonNode(settings.hostname, json), allocator); json.AddMember("hostname", jsonNode(settings.hostname, json), allocator);
json.AddMember("ip", jsonNode(settings.ip_addr, json), allocator); json.AddMember("ip", jsonNode(settings.ip_addr, json), allocator);
json.AddMember("netmask", jsonNode(settings.ip_mask, json), allocator); json.AddMember("netmask", jsonNode(settings.ip_mask, json), allocator);
json.AddMember("gateway", jsonNode(settings.gateway, json), allocator); json.AddMember("gateway", jsonNode(settings.gateway, json), allocator);
json.AddMember("type", jsonNode(device.type, json), allocator); json.AddMember("type", jsonNode(device.type, json), allocator);
json.AddMember("hardware_version", jsonNode(device.hardware_version, json), json.AddMember("hardware_version", jsonNode(device.hardware_version, json),
allocator); allocator);
json.AddMember("firmware_version", jsonNode(device.firmware_version, json), json.AddMember("firmware_version", jsonNode(device.firmware_version, json),
allocator); allocator);
json.AddMember("ports", jsonNode(ports, json), allocator); json.AddMember("ports", jsonNode(ports, json), allocator);
json.AddMember("vlans", jsonNode(vlans, json), allocator); json.AddMember("vlans", jsonNode(vlans, json), allocator);
rapidjson::StringBuffer sb; rapidjson::StringBuffer sb;
rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb); rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
json.Accept(writer); json.Accept(writer);
return sb.GetString(); return sb.GetString();
} }

View file

@ -19,48 +19,53 @@ int Switch::parse(datasets arr) {
return 0; return 0;
} }
int Switch::print() {
std::cout << "\t" << settings.hostname << " (" << device.type << ")\tMAC: "
<< device.mac << "\tIP: " << settings.ip_addr << "\n";
return 0;
}
int Switch::parse(dataset d) { int Switch::parse(dataset d) {
switch (d.type) { switch (d.type) {
case RCV_TYPE: case TYPE:
device.type = d.value; device.type = d.value;
break; break;
case RCV_MAC: case MAC:
device.mac = d.value; device.mac = d.value;
break; break;
case FIRMWARE_VERSION: case FIRMWARE_VERSION:
device.firmware_version = d.value; device.firmware_version = d.value;
break; break;
case HARDWARE_VERSION: case HARDWARE_VERSION:
device.hardware_version = d.value; device.hardware_version = d.value;
break; break;
case PORTS: case PORTS:
device.ports = d.value[0]; device.ports = d.value[0];
break; break;
case HOSTNAME: case HOSTNAME:
settings.hostname = d.value; settings.hostname = d.value;
break; break;
case IP_ADDR: case IP_ADDR:
settings.ip_addr = d.value; settings.ip_addr = d.value;
break; break;
case IP_MASK: case IP_MASK:
settings.ip_mask = d.value; settings.ip_mask = d.value;
break; break;
case GATEWAY: case GATEWAY:
settings.gateway = d.value; settings.gateway = d.value;
break; break;
case DHCP_ENABLED: case DHCP_ENABLED:
settings.dhcp = d.value[0]; settings.dhcp = d.value[0];
break; break;
case LOOP_PREVENTION: case LOOP_PREVENTION:
settings.loop_prevention = d.value[0]; settings.loop_prevention = d.value[0];
break; break;
case QOS_BASIC_ENABLED: case QOS_BASIC_ENABLED:
settings.qos_enabled = d.value[0]; settings.qos_enabled = d.value[0];
break; break;
case VLAN_ENABLED: case VLAN_ENABLED:
settings.vlan_enabled = d.value[0]; settings.vlan_enabled = d.value[0];
break; break;
} }
return 0; return 0;
} }

View file

@ -18,80 +18,87 @@
#define DEFAULT_PASS "admin" #define DEFAULT_PASS "admin"
enum CntStatus { enum CntStatus {
OPEN, SHORT OPEN, SHORT
}; };
struct vlan { struct vlan {
int vlan_id; int vlan_id;
std::string name; std::string name;
std::vector<byte> tagged_member; std::vector<byte> tagged_member;
std::vector<byte> untagged_member; std::vector<byte> untagged_member;
}; };
struct trunk { struct trunk {
int trunk_id; int trunk_id;
std::vector<byte> member; std::vector<byte> member;
}; };
struct port { struct port {
byte id; byte id;
byte status; byte status;
int pvid; int pvid;
struct { struct {
//port_settings //port_settings
} settings; } settings;
struct { struct {
//port_statistics //port_statistics
} stats; } stats;
struct { struct {
//bandwidth_control_ingress //bandwidth_control_ingress
//bandwidth_control_egress //bandwidth_control_egress
} bw_control; } bw_control;
struct { struct {
//storm_control //storm_control
} storm_control; } storm_control;
struct { struct {
CntStatus status; CntStatus status;
int fault_distace; int fault_distace;
} test; } test;
//qos_basic //qos_basic
}; };
class Switch { class Switch {
public: public:
Switch() { Switch() {
} }
int parse(datasets); int parse(datasets);
int parse(dataset); int parse(dataset);
int parse(std::string); int parse(std::string);
std::string toString(); int set(std::pair<std::string, std::string>);
struct { std::string get(std::string);
std::string type; std::string toString();
std::string hardware_version;
std::string firmware_version;
macAddr mac { 0x0, 0x0, 0x0, 0x0, 0x0, 0x0 };
int ports;
} device;
struct {
std::string password = DEFAULT_PASS;
std::string username = DEFAULT_USER;
std::string hostname;
ipAddr ip_addr { 0, 0, 0, 0, };
ipAddr ip_mask;
ipAddr gateway;
bool dhcp;
bool loop_prevention;
bool qos_enabled;
bool vlan_enabled;
//mtu_vlan
} settings;
private: int print();
rapidjson::Document json;
std::vector<vlan> vlans; struct {
std::vector<port> ports; std::string type;
std::string hardware_version;
std::string firmware_version;
macAddr mac {
0x0, 0x0, 0x0, 0x0, 0x0, 0x0 };
int ports;
} device;
struct {
std::string password = DEFAULT_PASS;
std::string username = DEFAULT_USER;
std::string hostname;
ipAddr ip_addr {
0, 0, 0, 0, };
ipAddr ip_mask;
ipAddr gateway;
bool dhcp;
bool loop_prevention;
bool qos_enabled;
bool vlan_enabled;
//mtu_vlan
} settings;
private:
rapidjson::Document json;
std::vector<vlan> vlans;
std::vector<port> ports;
}; };
#endif /* SWITCH_H_ */ #endif /* SWITCH_H_ */

View file

@ -20,106 +20,146 @@
#include "table.h" #include "table.h"
class macAddr: public std::array<byte, 6> { class macAddr: public std::array<byte, 6> {
public: public:
friend std::ostream& operator<<(std::ostream& out, const macAddr& arr) { friend std::ostream& operator<<(std::ostream& out, const macAddr& arr) {
out << std::hex << std::setw(2) << std::setfill('0') out << std::hex << std::setw(2) << std::setfill('0')
<< (unsigned) arr[0]; << (unsigned) arr[0];
for (unsigned i = 1; i < 6; i++) { for (unsigned i = 1; i < 6; i++) {
out << ":" << std::setw(2) << std::setfill('0') out << ":" << std::setw(2) << std::setfill('0')
<< (unsigned) arr[i]; << (unsigned) arr[i];
} }
return out; return out;
}
macAddr() {
*this= {0,0,0,0,0,0};
}
macAddr(std::initializer_list<byte> s) {
int i = 0;
for (byte b : s) {
if(i<6) (*this)[i++]=b;
else break;
} }
}
macAddr(bytes bts) { macAddr() {
int i = 0; *this= {0,0,0,0,0,0};
for (byte b : bts) {
if(i<6) (*this)[i++]=b;
else break;
} }
}
};
/* macAddr(std::initializer_list<byte> s) {
class mac_addr : public std::array<unsigned char, 6> { int i = 0;
public: for (byte b : s) {
typedef std::array<unsigned char, 6> super; if(i<6) (*this)[i++]=b;
else break;
}
}
using super::super; macAddr(bytes bts) {
int i = 0;
for (byte b : bts) {
if(i<6) (*this)[i++]=b;
else break;
}
}
mac_addr{00, 00, 00, 000}; int hash() {
}; int ret=0;
*/ for (unsigned i = 0; i < 6; i++) {
ret = (ret*33) ^ (*this)[i];
}
return ret;
}
bool operator==(const macAddr &A) {
for (unsigned i = 0; i < 6; i++) {
if(A[i]!=(*this)[i])return false;
}
return true;
}
bool operator!=(const macAddr &A) {
for (unsigned i = 0; i < 6; i++) {
if(A[i]!=(*this)[i])return true;
}
return false;
}
};
/*
class mac_addr : public std::array<unsigned char, 6> {
public:
typedef std::array<unsigned char, 6> super;
using super::super;
mac_addr{00, 00, 00, 000};
};
*/
class ipAddr: public std::array<byte, 4> { class ipAddr: public std::array<byte, 4> {
public: public:
ipAddr() { ipAddr() {
*this= {0,0,0,0,0,0}; *this= {0,0,0,0,0,0};
}
ipAddr(std::initializer_list<byte> s) {
int i = 0;
for (byte b : s) {
if(i<4) (*this)[i++]=b;
else break;
} }
}
ipAddr(bytes bts) { ipAddr(std::initializer_list<byte> s) {
int i = 0; int i = 0;
for (byte b : bts) { for (byte b : s) {
if(i<4) (*this)[i++]=b; if(i<4) (*this)[i++]=b;
else break; else break;
}
} }
}
friend std::ostream& operator<<(std::ostream& out, ipAddr& arr) { ipAddr(bytes bts) {
out << std::dec << (unsigned) arr[0]; int i = 0;
for (unsigned i = 1; i < 4; i++) { for (byte b : bts) {
out << "." << (unsigned) arr[i]; if(i<4) (*this)[i++]=b;
} else break;
return out; }
} }
};
friend std::ostream& operator<<(std::ostream& out, ipAddr& arr) {
out << std::dec << (unsigned) arr[0];
for (unsigned i = 1; i < 4; i++) {
out << "." << (unsigned) arr[i];
}
return out;
}
};
namespace smrtlink {
constexpr unsigned int caseArg(const char* str, int h = 0) {
return !str[h] ? 5381 : (caseArg(str, h + 1) * 33) ^ str[h];
}
}
template<typename T> template<typename T>
std::vector<T> operator+(const std::vector<T> &A, const std::vector<T> &B) { std::vector<T> operator+(const std::vector<T> &A, const std::vector<T> &B) {
std::vector<T> AB; std::vector<T> AB;
AB.reserve(A.size() + B.size()); // preallocate memory AB.reserve(A.size() + B.size()); // preallocate memory
AB.insert(AB.end(), A.begin(), A.end()); // add A; AB.insert(AB.end(), A.begin(), A.end()); // add A;
AB.insert(AB.end(), B.begin(), B.end()); // add B; AB.insert(AB.end(), B.begin(), B.end()); // add B;
return AB; return AB;
} }
template<typename T> template<typename T>
std::vector<T> &operator+=(std::vector<T> &A, const std::vector<T> &B) { std::vector<T> &operator+=(std::vector<T> &A, const std::vector<T> &B) {
A.reserve(A.size() + B.size()); A.reserve(A.size() + B.size());
A.insert(A.end(), B.begin(), B.end()); A.insert(A.end(), B.begin(), B.end());
return A; return A;
} }
struct Options { struct Options {
unsigned flags = 0x00; struct {
std::string user; bool HEX;
std::string password; bool JSON;
std::string interface; bool PLAIN;
std::string file; bool REVERSE;
int debug_level=0;
long timeout = 180U; bool HEADER;
bool PERMANENT;
bool WAIT;
bool INTERACTIVE;
} flags;
std::string user;
std::string password;
std::string interface;
std::string file;
int debug_level = 0;
int verbosity = 0;
long timeout = 250U;
}; };
#endif /* TYPES_H_ */ #endif /* TYPES_H_ */

View file

@ -7,23 +7,23 @@
#include "bytes.h" #include "bytes.h"
bytes bytes::readHex(std::string s){ bytes bytes::readHex(std::string s) {
vector ret; vector ret;
std::string delimiter = ":"; std::string delimiter = ":";
std::string token; std::string token;
size_t pos = 0; size_t pos = 0;
int hex; int hex;
byte b; byte b;
ret.resize(0); ret.resize(0);
while ((pos = s.find(delimiter)) != std::string::npos) { while ((pos = s.find(delimiter)) != std::string::npos) {
token = s.substr(0, pos); token = s.substr(0, pos);
sscanf(token.c_str(), "%x", &hex); sscanf(token.c_str(), "%x", &hex);
s.erase(0, pos + delimiter.length()); s.erase(0, pos + delimiter.length());
b = hex & 0xFF; b = hex & 0xFF;
ret.push_back(b); ret.push_back(b);
} }
sscanf(s.c_str(), "%x", &hex); sscanf(s.c_str(), "%x", &hex);
b = hex & 0xFF; b = hex & 0xFF;
ret.push_back(b); ret.push_back(b);
return ret; return ret;
} }

View file

@ -19,66 +19,69 @@
typedef unsigned char byte; typedef unsigned char byte;
class bytes: public std::vector<unsigned char> { class bytes: public std::vector<unsigned char> {
typedef std::vector<unsigned char> vector; typedef std::vector<unsigned char> vector;
public: public:
using vector::operator[]; using vector::operator[];
bytes() { bytes() {
} }
bytes(int n) : vector(n){ bytes(int n) :
} vector(n) {
}
bytes(std::string d) : vector(d.begin(), d.end()){ bytes(std::string d) :
} vector(d.begin(), d.end()) {
this->push_back('\0');
}
bytes(std::initializer_list<uint8_t> s) bytes(std::initializer_list<uint8_t> s) {
{ for (uint8_t b : s) {
for (uint8_t b : s) { this->push_back(b);
this->push_back(b); }
} }
}
bytes(const vector &B) { bytes(const vector &B) {
this->reserve(B.size()); this->reserve(B.size());
this->insert(this->begin(), B.begin(), B.end()); this->insert(this->begin(), B.begin(), B.end());
} }
bytes readHex(std::string); bytes readHex(std::string);
bytes operator=(const vector &B) { bytes operator=(const vector &B) {
this->reserve(B.size()); this->reserve(B.size());
this->insert(this->begin(), B.begin(), B.end()); this->insert(this->begin(), B.begin(), B.end());
return *this; return *this;
} }
bytes &operator+=(const bytes &B) { bytes &operator+=(const bytes &B) {
this->reserve(this->size() + B.size()); this->reserve(this->size() + B.size());
this->insert(this->end(), B.begin(), B.end()); this->insert(this->end(), B.begin(), B.end());
return *this; return *this;
} }
bytes operator+(const bytes &B) { bytes operator+(const bytes &B) {
bytes AB; bytes AB;
AB.reserve(this->size() + B.size()); AB.reserve(this->size() + B.size());
AB.insert(AB.end(), this->begin(), this->end()); AB.insert(AB.end(), this->begin(), this->end());
AB.insert(AB.end(), B.begin(), B.end()); AB.insert(AB.end(), B.begin(), B.end());
return AB; return AB;
} }
friend std::ostream& operator<<(std::ostream& out, const bytes& arr) { friend std::ostream& operator<<(std::ostream& out, const bytes& arr) {
if (arr.size() > 0) { if (arr.size() > 0) {
out << std::hex << std::setw(2) << std::setfill('0') out << std::hex << std::setw(2) << std::setfill('0')
<< (unsigned) arr[0]; << (unsigned) arr[0];
} }
for (unsigned i = 1; i < arr.size(); i++) { for (unsigned i = 1; i < arr.size(); i++) {
out << ":" << std::setw(2) << std::setfill('0') << (unsigned) arr[i]; out << ":" << std::setw(2) << std::setfill('0')
} << (unsigned) arr[i];
return out; }
} return out;
}
operator std::string() { operator std::string() {
std::string s(this->begin(),this->end()); std::string s(this->begin(), this->end());
return s; return s;
} }
}; };
#endif /* BYTES_H_ */ #endif /* BYTES_H_ */

View file

@ -12,21 +12,29 @@
#include "bytes.h" #include "bytes.h"
struct dataset { struct dataset {
short type; short type;
short len; short len;
bytes value; bytes value;
}; };
class datasets : public std::vector<dataset> { class datasets: public std::vector<dataset> {
public: public:
datasets(){}; datasets() {
datasets(std::initializer_list<dataset> s) }
{ datasets(std::initializer_list<dataset> s) {
for (dataset b : s) { for (dataset b : s) {
//(*this)[b.type]=b; //(*this)[b.type]=b;
push_back(b); push_back(b);
} }
} }
datasets operator+(const datasets &B) {
datasets AB;
AB.reserve(this->size() + B.size());
AB.insert(AB.end(), this->begin(), this->end());
AB.insert(AB.end(), B.begin(), B.end());
return AB;
}
}; };
#endif /* DATASETS_H_ */ #endif /* DATASETS_H_ */

View file

@ -11,43 +11,43 @@
#include "jsonNode.h" #include "jsonNode.h"
jsonNode::jsonNode(const std::string &x, doc &root) { jsonNode::jsonNode(const std::string &x, doc &root) {
super(rapidjson::kStringType); super(rapidjson::kStringType);
char buffer[30]; char buffer[30];
int len = sprintf(buffer, "%s", x.c_str()); int len = sprintf(buffer, "%s", x.c_str());
this->SetString(buffer, static_cast<size_t>(len), root.GetAllocator()); this->SetString(buffer, static_cast<size_t>(len), root.GetAllocator());
memset(buffer, 0, sizeof(buffer)); memset(buffer, 0, sizeof(buffer));
} }
jsonNode::jsonNode(const ipAddr &x, doc &root) { jsonNode::jsonNode(const ipAddr &x, doc &root) {
super(rapidjson::kStringType); super(rapidjson::kStringType);
char buffer[16]; char buffer[16];
int len = sprintf(buffer, "%d.%d.%d.%d", x[0], x[1], x[2], x[3]); int len = sprintf(buffer, "%d.%d.%d.%d", x[0], x[1], x[2], x[3]);
this->SetString(buffer, static_cast<size_t>(len), root.GetAllocator()); this->SetString(buffer, static_cast<size_t>(len), root.GetAllocator());
memset(buffer, 0, sizeof(buffer)); memset(buffer, 0, sizeof(buffer));
} }
jsonNode::jsonNode(const macAddr &x, doc &root) { jsonNode::jsonNode(const macAddr &x, doc &root) {
super(rapidjson::kStringType); super(rapidjson::kStringType);
char buffer[18]; char buffer[18];
int len = sprintf(buffer, "%02x:%02x:%02x:%02x:%02x:%02x", x[0], x[1], x[2], int len = sprintf(buffer, "%02x:%02x:%02x:%02x:%02x:%02x", x[0], x[1], x[2],
x[3], x[4], x[5]); x[3], x[4], x[5]);
this->SetString(buffer, static_cast<size_t>(len), root.GetAllocator()); this->SetString(buffer, static_cast<size_t>(len), root.GetAllocator());
memset(buffer, 0, sizeof(buffer)); memset(buffer, 0, sizeof(buffer));
} }
jsonNode::jsonNode(const vlan &x, doc &root) { jsonNode::jsonNode(const vlan &x, doc &root) {
super(rapidjson::kObjectType); super(rapidjson::kObjectType);
AddMember("id", x.vlan_id, root.GetAllocator()); AddMember("id", x.vlan_id, root.GetAllocator());
AddMember("name", jsonNode(x.name, root), root.GetAllocator()); AddMember("name", jsonNode(x.name, root), root.GetAllocator());
AddMember("tagged-members", jsonNode(x.tagged_member, root), AddMember("tagged-members", jsonNode(x.tagged_member, root),
root.GetAllocator()); root.GetAllocator());
AddMember("untagged-members", jsonNode(x.untagged_member, root), AddMember("untagged-members", jsonNode(x.untagged_member, root),
root.GetAllocator()); root.GetAllocator());
} }
jsonNode::jsonNode(const port &x, doc &root) { jsonNode::jsonNode(const port &x, doc &root) {
super(rapidjson::kObjectType); super(rapidjson::kObjectType);
AddMember("id", x.id, root.GetAllocator()); AddMember("id", x.id, root.GetAllocator());
AddMember("status", x.status, root.GetAllocator()); AddMember("status", x.status, root.GetAllocator());
AddMember("pvid", x.pvid, root.GetAllocator()); AddMember("pvid", x.pvid, root.GetAllocator());
} }

View file

@ -13,26 +13,26 @@
#include "Switch.h" #include "Switch.h"
class jsonNode: public rapidjson::Value { class jsonNode: public rapidjson::Value {
typedef rapidjson::Value super; typedef rapidjson::Value super;
typedef rapidjson::Document doc; typedef rapidjson::Document doc;
public: public:
jsonNode(const std::string&, doc&); jsonNode(const std::string&, doc&);
jsonNode(const macAddr&, doc&); jsonNode(const macAddr&, doc&);
jsonNode(const ipAddr&, doc&); jsonNode(const ipAddr&, doc&);
jsonNode(const vlan&, doc&); jsonNode(const vlan&, doc&);
jsonNode(const port&, doc&); jsonNode(const port&, doc&);
template<class T> template<class T>
jsonNode(const T &templ_arg, doc &root){ jsonNode(const T &templ_arg, doc &root) {
std::cerr<<"Not serializable Type: "<<typeid(T).name()<<"\n"; std::cerr << "Not serializable Type: " << typeid(T).name() << "\n";
} }
template<class T> template<class T>
jsonNode(const std::vector<T> &x, doc &root) { jsonNode(const std::vector<T> &x, doc &root) {
super(rapidjson::kArrayType); super(rapidjson::kArrayType);
for (T y : x) for (T y : x)
PushBack(jsonNode(y, root), root.GetAllocator()); PushBack(jsonNode(y, root), root.GetAllocator());
} }
}; };
#endif /* JSONNODE_H_ */ #endif /* JSONNODE_H_ */

View file

@ -1,30 +1,12 @@
enum { enum {
#define LOOKUP_SET(id, num, type) RCV_ ## id = num, #define LOOKUP_SET(id, name, num, type, write) id = num,
#include "lookup/rcv.lst" #include "lookup/map.lst"
#include "lookup/general.lst"
#undef LOOKUP_SET
#define LOOKUP_SET(id, num, type) SND_ ## id = num,
#include "lookup/snd.lst"
#include "lookup/general.lst"
#undef LOOKUP_SET
#define LOOKUP_SET(id, num, type) id = num,
#include "lookup/rcv.lst"
#include "lookup/snd.lst"
#include "lookup/general.lst"
#undef LOOKUP_SET #undef LOOKUP_SET
}; };
static table rcv_lookup { static table lookup {
#define LOOKUP_SET(id, num, type) { RCV_ ## id, table::type, #id }, #define LOOKUP_SET(id, name, num, type, write) { id, table::type, table::write, #name, #id},
#include "lookup/rcv.lst" #include "lookup/map.lst"
#include "lookup/general.lst"
#undef LOOKUP_SET
};
static table snd_lookup {
#define LOOKUP_SET(id, num, type) { SND_ ## id, table::type, #id },
#include "lookup/snd.lst"
#include "lookup/general.lst"
#undef LOOKUP_SET #undef LOOKUP_SET
}; };

View file

@ -1,25 +1,27 @@
LOOKUP_SET(PORTS, 10, DEC) //+byte, maybe number of ports
LOOKUP_SET(IGMP_SNOOPING, 4352, HEX) //???
LOOKUP_SET(PORTS_SETTINGS, 4096, HEX) //+per port
LOOKUP_SET(PORT_TRUNK, 4608, HEX) //byte[5] last byte bitmask??
LOOKUP_SET(MTU_VLAN, 8192, HEX) //byte[2] first byte bool,second byte port id LOOKUP_SET(PORTS, ports, 10, DEC) //+byte, maybe number of ports
LOOKUP_SET(PORT_VLAN_ENABLED, 8448, BOOL) //open page
LOOKUP_SET(PORT_VLAN, 8449, HEX)
LOOKUP_SET(PORT_VLAN_MAX, 8450, DEC)
LOOKUP_SET(VLAN_ENABLED, 8704, BOOL) //+bool byte
LOOKUP_SET(VLAN, 8705, HEX) //+one set per vlan
LOOKUP_SET(VLAN_PVID, 8706, HEX) //+per port
LOOKUP_SET(VLAN_FOOOO, 8707, DEC) //????
LOOKUP_SET(QOS_BASIC_ENABLED, 12288, BOOL) //+bool = QoS Mod LOOKUP_SET(IGMP_SNOOPING, igmp, 4352, HEX) //???
LOOKUP_SET(QOS_BASIC, 12289, HEX) //+per port ??? LOOKUP_SET(PORTS_SETTINGS, psetings, 4096, HEX) //+per port
LOOKUP_SET(BW_CONTROL_INGRESS, 12544, HEX) //+per port ??? LOOKUP_SET(PORT_TRUNK, trunk, 4608, HEX) //byte[5] last byte bitmask??
LOOKUP_SET(BW_CONTROL_EGRESS, 12545, HEX) //+per port ???
LOOKUP_SET(STORM_CONTROL, 12800, HEX) //+per port ??? LOOKUP_SET(MTU_VLAN, 8192,8192, HEX) //byte[2] first byte bool,second byte port id
LOOKUP_SET(PORT_MIRROR, 16640, HEX) //byte[10] second byte port id??? LOOKUP_SET(PORT_VLAN_ENABLED, 8448,8448, BOOL) //open page
LOOKUP_SET(PORT_STATISTICS, 16384, HEX) //+per port ??? LOOKUP_SET(PORT_VLAN, 8449,8449, HEX)
LOOKUP_SET(CABLE_TEST, 16896, HEX) //+per port ??? LOOKUP_SET(PORT_VLAN_MAX, 8450,8450, DEC)
LOOKUP_SET(LOOP_PREVENTION, 17152, BOOL) //+bool byte LOOKUP_SET(VLAN_ENABLED, 8704, 8704, BOOL) //+bool byte
LOOKUP_SET(VLAN, 8705,8705, HEX) //+one set per vlan
LOOKUP_SET(VLAN_PVID, 8706,8706, HEX) //+per port
LOOKUP_SET(VLAN_FOOOO, 8707,8707, DEC) //????
LOOKUP_SET(QOS_BASIC_ENABLED, 12288,12288, BOOL) //+bool = QoS Mod
LOOKUP_SET(QOS_BASIC, 12289,12289, HEX) //+per port ???
LOOKUP_SET(BW_CONTROL_INGRESS, 12544,12544, HEX) //+per port ???
LOOKUP_SET(BW_CONTROL_EGRESS, 12545,12545, HEX) //+per port ???
LOOKUP_SET(STORM_CONTROL, 12800,12800, HEX) //+per port ???
LOOKUP_SET(PORT_MIRROR, 16640,16640, HEX) //byte[10] second byte port id???
LOOKUP_SET(PORT_STATISTICS, 16384,16384, HEX) //+per port ???
LOOKUP_SET(CABLE_TEST, 16896,16896, HEX) //+per port ???
LOOKUP_SET(LOOP_PREVENTION, 17152,17152, BOOL) //+bool byte

49
src/lookup/map.lst Normal file
View file

@ -0,0 +1,49 @@
LOOKUP_SET(TYPE, type, 1, STRING, RO) //+string
LOOKUP_SET(HOSTNAME, host, 2, STRING, RW) //+string //SYSTEM_INFO,SYSTEM_INFO, 2, HEX) //page sysinfo
LOOKUP_SET(MAC, mac, 3, HEX, RO) //+byte[6]
LOOKUP_SET(IP_ADDR, ip, 4, DEC, RW) //+byte[4]
LOOKUP_SET(IP_MASK, mask, 5, DEC, RW) //+byte[4]
LOOKUP_SET(GATEWAY, gateway, 6, DEC, RW) //+byte[4]
LOOKUP_SET(FIRMWARE_VERSION,firmware, 7, STRING, RO)
LOOKUP_SET(HARDWARE_VERSION,hardware, 8, STRING, RO)
LOOKUP_SET(DHCP_ENABLED, dhcp, 9, BOOL, RW) //+bool byte //IP_CONFIG,IP_CONFIG, 9, HEX) //page sysinfo
LOOKUP_SET(PORTS, ports, 10, DEC, RO) //+byte, maybe number of ports
LOOKUP_SET(LOGIN_USER, luser, 512, STRING, NONE) //string
LOOKUP_SET(NEW_USER, user, 513, STRING, WO) //string
LOOKUP_SET(LOGIN_PASSWORD, lpw, 514, STRING, NONE) //string
LOOKUP_SET(NEW_PASSWORD, password, 515, STRING, WO) //string
LOOKUP_SET(REBOOT, reboot, 773, BOOL, NONE) //bool byte = save config
LOOKUP_SET(RESET, reset, 1280, ACTION, NONE) //
LOOKUP_SET(FLASH, flash, 1536, ACTION, NONE) //update firmware
LOOKUP_SET(PORT_VLAN_FOOOOO,vlan21, 2101, HEX, RO)
LOOKUP_SET(VLAN_FOOOOOO, vlan22, 2200, HEX, RO) //vlan
LOOKUP_SET(SAVE, save, 2304, ACTION, NONE) //save
LOOKUP_SET(PING, ping, 2305, ACTION, NONE) //sent before SET ???
LOOKUP_SET(IGMP_SNOOPING, igmp, 4352, HEX, RW) //???
LOOKUP_SET(PORTS_SETTINGS, psetings, 4096, HEX, RW) //+per port
LOOKUP_SET(PORT_TRUNK, trunk, 4608, HEX, RW) //byte[5] last byte bitmask??
LOOKUP_SET(MTU_VLAN, 8192, 8192, HEX, UNKNOWN) //byte[2] first byte bool,second byte port id
LOOKUP_SET(PORT_VLAN_ENABLED,8448, 8448, BOOL, UNKNOWN) //open page
LOOKUP_SET(PORT_VLAN, 8449, 8449, HEX, UNKNOWN)
LOOKUP_SET(PORT_VLAN_MAX, 8450, 8450, DEC, UNKNOWN)
LOOKUP_SET(VLAN_ENABLED, 8704, 8704, BOOL, UNKNOWN) //+bool byte
LOOKUP_SET(VLAN, 8705, 8705, HEX, UNKNOWN) //+one set per vlan
LOOKUP_SET(VLAN_PVID, 8706, 8706, HEX, UNKNOWN) //+per port
LOOKUP_SET(VLAN_FOOOO, 8707, 8707, DEC, UNKNOWN) //????
LOOKUP_SET(QOS_BASIC_ENABLED,12288, 12288, BOOL, UNKNOWN) //+bool = QoS Mod
LOOKUP_SET(QOS_BASIC, 12289, 12289, HEX, UNKNOWN) //+per port ???
LOOKUP_SET(BW_CONTROL_INGRESS,12544, 12544, HEX, UNKNOWN) //+per port ???
LOOKUP_SET(BW_CONTROL_EGRESS,12545, 12545, HEX, UNKNOWN) //+per port ???
LOOKUP_SET(STORM_CONTROL, 12800, 12800, HEX, UNKNOWN) //+per port ???
LOOKUP_SET(PORT_MIRROR, 16640, 16640, HEX, UNKNOWN) //byte[10] second byte port id???
LOOKUP_SET(PORT_STATISTICS, 16384, 16384, HEX, UNKNOWN) //+per port ???
LOOKUP_SET(CABLE_TEST, 16896, 16896, HEX, UNKNOWN) //+per port ???
LOOKUP_SET(LOOP_PREVENTION, 17152, 17152, BOOL, UNKNOWN) //+bool byte

View file

@ -1,13 +1,13 @@
LOOKUP_SET(TYPE, 1, STRING) //+string LOOKUP_SET(TYPE,type, 1, STRING) //+string
LOOKUP_SET(HOSTNAME, 2, STRING) //+string LOOKUP_SET(HOSTNAME,host, 2, STRING) //+string
LOOKUP_SET(MAC, 3, HEX) //+byte[6] LOOKUP_SET(MAC,MAC, 3, HEX) //+byte[6]
LOOKUP_SET(IP_ADDR, 4, DEC) //+byte[4] LOOKUP_SET(IP_ADDR,ip, 4, DEC) //+byte[4]
LOOKUP_SET(IP_MASK, 5, DEC) //+byte[4] LOOKUP_SET(IP_MASK,mask, 5, DEC) //+byte[4]
LOOKUP_SET(GATEWAY, 6, DEC) //+byte[4] LOOKUP_SET(GATEWAY,gateway, 6, DEC) //+byte[4]
LOOKUP_SET(FIRMWARE_VERSION, 7, STRING) LOOKUP_SET(FIRMWARE_VERSION,FIRMWARE_VERSION, 7, STRING)
LOOKUP_SET(HARDWARE_VERSION, 8, STRING) LOOKUP_SET(HARDWARE_VERSION,HARDWARE_VERSION, 8, STRING)
LOOKUP_SET(DHCP_ENABLED, 9, BOOL) //+bool byte LOOKUP_SET(DHCP_ENABLED,dhcp, 9, BOOL) //+bool byte
LOOKUP_SET(PORT_VLAN_FOOOOO, 2101, HEX) LOOKUP_SET(PORT_VLAN_FOOOOO, 2101,2101, HEX)

View file

@ -1,17 +1,17 @@
LOOKUP_SET(SYSTEM_INFO, 2, HEX) //page sysinfo LOOKUP_SET(SYSTEM_INFO,SYSTEM_INFO, 2, HEX) //page sysinfo
LOOKUP_SET(IP_CONFIG, 9, HEX) //page sysinfo LOOKUP_SET(IP_CONFIG,IP_CONFIG, 9, HEX) //page sysinfo
LOOKUP_SET(LOGIN_USER, 512, STRING) //string LOOKUP_SET(LOGIN_USER,LOGIN_USER, 512, STRING) //string
LOOKUP_SET(NEW_USER, 513, STRING) //string LOOKUP_SET(NEW_USER,NEW_USER, 513, STRING) //string
LOOKUP_SET(LOGIN_PASSWORD, 514, STRING) //string LOOKUP_SET(LOGIN_PASSWORD,LOGIN_PASSWORD, 514, STRING) //string
LOOKUP_SET(NEW_PASSWORD, 515, STRING) //string LOOKUP_SET(NEW_PASSWORD,NEW_PASSWORD, 515, STRING) //string
LOOKUP_SET(REBOOT, 773, BOOL) //bool byte = save config LOOKUP_SET(REBOOT,REBOOT, 773, BOOL) //bool byte = save config
LOOKUP_SET(RESET, 1280, ACTION) // LOOKUP_SET(RESET,RESET, 1280, ACTION) //
LOOKUP_SET(FLASH, 1536, ACTION) //update firmware LOOKUP_SET(FLASH,FLASH, 1536, ACTION) //update firmware
LOOKUP_SET(VLAN_FOOOOOO, 2200, HEX) //vlan LOOKUP_SET(VLAN_FOOOOOO, 2200,2200, HEX) //vlan
LOOKUP_SET(SAVE, 2304, ACTION) //save LOOKUP_SET(SAVE,SAVE, 2304, ACTION) //save
LOOKUP_SET(PING, 2305, ACTION) //sent before SET ??? LOOKUP_SET(PING,PING, 2305, ACTION) //sent before SET ???

View file

@ -1,221 +1,169 @@
//============================================================================ //============================================================================
// Name : smrtlink.cpp // Name : smrtlink.cpp
// Author : jdi // Author : jdi
// Version : // Version : 1.2
// Copyright : GPL v2 // Copyright : GPL v2
// Description : SmrtLink in C++, Ansi-style // Description : SmrtLink in C++, Ansi-style
//============================================================================ //============================================================================
#include <cstring>
#include <iostream> #include <iostream>
#include <sstream>
#include <string> #include <string>
#include <cstring>
#include <sstream>
#include <cstdlib> #include <cstdlib>
#include <cstdio>
#include <getopt.h> #include <getopt.h>
#include <unistd.h> #include <unistd.h>
#include "Constant.h" #include "Constant.h"
#include "Interactive.h"
#include "Host.h" #include "Host.h"
#include "Program.h" #include "Program.h"
#include "Types.h"
#include "Switch.h" #include "Switch.h"
#include "lookup.h"
#define no_argument 0 #define no_argument 0
#define required_argument 1 #define required_argument 1
#define optional_argument 2 #define optional_argument 2
using namespace std;
Options options; Options options;
constexpr unsigned int caseArg(const char* str, int h = 0) {
return !str[h] ? 5381 : (caseArg(str, h + 1) * 33) ^ str[h];
}
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
int opt, index; int index, opt;
options.user = DEFAULT_USER; options.user = DEFAULT_USER;
options.password = DEFAULT_PASS; options.password = DEFAULT_PASS;
const struct option longopts[] = { { "version", no_argument, 0, 'v' }, { const struct option longopts[] = {
"help", no_argument, 0, 'h' }, { "reverse", no_argument, 0, 'r' }, { { "version", no_argument, 0, 'V' }, {
"permanent", no_argument, 0, 's' }, { "debug", optional_argument, 0, "verbose", no_argument, 0, 'v' },
'd' }, { "password", required_argument, 0, 'p' }, { "user", { "help", no_argument, 0, 'h' }, {
required_argument, 0, 'u' }, { "interface", required_argument, 0, 'i' }, { "reverse", no_argument, 0, 'r' },
"header", required_argument, 0, 'b' }, { "hex", required_argument, { "permanent", no_argument, 0, 's' }, {
0, 'x' }, { "file", required_argument, 0, 'f' }, { "timeout", "debug", optional_argument, 0, 'd' },
required_argument, 0, 't' }, { "wait", { "password", required_argument, 0, 'P' }, {
required_argument, 0, 'w' }, { 0, 0, 0, 0 }, }; "user", required_argument, 0, 'U' },
{ "interface", required_argument, 0, 'i' }, {
"header", required_argument, 0, 'b' },
{ "hex", required_argument, 0, 'x' }, {
"file", required_argument, 0, 'f' },
{ "timeout", required_argument, 0, 't' }, {
"wait", required_argument, 0, 'w' },
{ 0, 0, 0, 0 }, };
Program p = Program(); while ((opt = getopt_long(argc, argv, "bhrvVXIswxP:U:i:f:t:d::", longopts,
&index)) != -1) {
switch (opt) {
while ((opt = getopt_long(argc, argv, "bhrvswxp:u:i:f:t:d::", longopts, case 'h':
&index)) != -1) { fprintf(stderr, VERSION);
switch (opt) { fprintf(stderr, USAGE, argv[0]);
fprintf(stderr, HELP);
exit(EXIT_SUCCESS);
break;
case 'h': case 'V':
fprintf(stderr, VERSION); fprintf(stderr, VERSION);
fprintf(stderr, USAGE, argv[0]); exit(EXIT_SUCCESS);
fprintf(stderr, HELP); break;
exit(EXIT_SUCCESS);
break;
case 'v': case 'r':
fprintf(stderr, VERSION); options.flags.REVERSE = true;
exit(EXIT_SUCCESS); break;
break;
case 'r': case 'b':
options.flags |= FLAG_REVERSE; options.flags.HEADER = true;
break; break;
case 'b': case 'x':
options.flags |= FLAG_HEADER; options.flags.HEX = true;
break; break;
case 'x': case 's':
options.flags |= FLAG_HEX; options.flags.PERMANENT = true;
break; break;
case 's': case 'w':
options.flags |= FLAG_PERMANENT; options.flags.WAIT = true;
break; break;
case 'w': case 'X':
options.flags |= FLAG_WAIT; options.flags.INTERACTIVE = true;
break; break;
case 'd': case 'v':
options.flags |= FLAG_DEBUG; if (optarg != NULL)
if (optarg != NULL) options.verbosity = atoi(optarg);
options.debug_level = atoi(optarg); else
break; options.verbosity++;
break;
case 't': case 'd':
options.timeout = atol(optarg); if (optarg != NULL)
break; options.debug_level = atoi(optarg);
else
options.debug_level++;
break;
case 'f': case 't':
options.file = std::string(optarg); options.timeout = atol(optarg);
break; break;
case 'p': case 'f':
options.password = std::string(optarg); options.file = std::string(optarg);
break; break;
case 'u': case 'P':
options.user = std::string(optarg); options.password = std::string(optarg);
break; break;
case 'i': case 'U':
options.interface = std::string(optarg); options.user = std::string(optarg);
break; break;
default: /* '?' */ case 'i':
fprintf(stderr, "Unknown option\n"); options.interface = std::string(optarg);
fprintf(stderr, USAGE, argv[0]); break;
exit(EXIT_FAILURE);
}
}
/*//TODO stdin default: /* '?' */
std::ostringstream bucket; fprintf(stderr, "Unknown option\n");
bucket << std::cin.rdbuf(); fprintf(stderr, USAGE, argv[0]);
p.input = bucket.str(); exit(EXIT_FAILURE);
*/ }
}
if (optind >= argc) { /*//TODO stdin
fprintf(stderr, "Command expected\n"); std::ostringstream bucket;
fprintf(stderr, USAGE, argv[0]); bucket << std::cin.rdbuf();
exit(EXIT_FAILURE); p.input = bucket.str();
} */
p.init(); if (optind >= argc && !options.flags.INTERACTIVE) {
cerr << "Command expected\n";
cerr << USAGE;
exit(EXIT_FAILURE);
}
if (optind < argc) { Interactive p = Interactive();
std::string cmd = std::string(argv[optind++]); if (options.flags.INTERACTIVE) {
if (optind < argc) {
switch (caseArg(cmd.c_str())) { cerr << "Command is ignored in interactive mode\n";
case caseArg("set"): }
if (p.setProperty()) if (!p.loop())
exit(EXIT_SUCCESS); exit(EXIT_SUCCESS);
fprintf(stderr, "Not yet implemented.\n"); fprintf(stderr, "Not yet implemented.\n");
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
break; }
case caseArg("reboot"): else if (optind < argc) {
if (p.reboot()) vector<string> v;
exit(EXIT_SUCCESS); while (optind < argc)
fprintf(stderr, "Not yet implemented.\n"); v.push_back(argv[optind++]);
exit(EXIT_FAILURE); p.single(v);
break; }
case caseArg("reset"): exit(EXIT_FAILURE);
if (p.reset())
exit(EXIT_SUCCESS);
fprintf(stderr, "Not yet implemented.\n");
exit(EXIT_FAILURE);
break;
case caseArg("save"):
if (p.save())
exit(EXIT_SUCCESS);
fprintf(stderr, "Not yet implemented.\n");
exit(EXIT_FAILURE);
break;
case caseArg("restore"):
if (p.restore())
exit(EXIT_SUCCESS);
fprintf(stderr, "Not yet implemented.\n");
exit(EXIT_FAILURE);
break;
case caseArg("flash"):
if (p.flash())
exit(EXIT_SUCCESS);
fprintf(stderr, "Not yet implemented.\n");
exit(EXIT_FAILURE);
break;
case caseArg("get"):
if (p.getProperty())
exit(EXIT_SUCCESS);
break;
case caseArg("list"):
if (p.list())
exit(EXIT_SUCCESS);
break;
case caseArg("sniff"):
if (p.sniff())
exit(EXIT_SUCCESS);
break;
case caseArg("encode"):
if (optind < argc) {
std::string s(argv[optind]);
optind++;
if (p.encode(s))
exit(EXIT_SUCCESS);
} else {
fprintf(stderr, "Argument expected after encode\n");
exit(EXIT_FAILURE);
}
break;
case caseArg("help"):
fprintf(stderr, VERSION);
fprintf(stderr, USAGE, argv[0]);
fprintf(stderr, HELP);
exit(EXIT_SUCCESS);
break;
default:
printf("Unknown command: %s\n", cmd.c_str());
while (optind < argc) {
printf("->%s\n", argv[optind]);
optind++;
}
exit(EXIT_FAILURE);
}
}
exit(EXIT_FAILURE);
} }

View file

@ -9,17 +9,33 @@
#include "table.h" #include "table.h"
table::table(std::initializer_list<set> l) { table::table(std::initializer_list<set> l) {
int i = 0; int i = 0;
for (set s : l) { this->data.resize(l.size());
this->left[s.type] = s; for (set s : l) {
} this->data[i] = s;
this->left[s.type] = &this->data[i];
this->right[s.name] = &this->data[i];
i++;
}
} }
table::set table::operator[](short n){ const table::set table::operator[](std::string s) {
return this->left[n]; return *this->right[s];
} }
bool table::exists(short n){ const table::set table::operator[](short n) {
return !(left.find(n) == left.end()); return *this->left[n];
} }
std::string table::name(short n){ bool table::exists(std::string s) {
return this->left[n].name; return !(right.find(s) == right.end());
}
bool table::exists(short n) {
return !(left.find(n) == left.end());
}
short table::type(std::string s) {
return this->right[s]->type;
}
std::string table::id(short n) {
return this->left[n]->id;
}
std::string table::name(short n) {
return this->left[n]->name;
} }

View file

@ -12,19 +12,32 @@
#include <vector> #include <vector>
class table { class table {
public: public:
enum F {STRING,HEX,DEC,ACTION,BOOL,EMPTY}; enum A {
struct set { RW, RO, WO, NONE, UNKNOWN
short type; };
F format; enum F {
std::string name; STRING, HEX, DEC, ACTION, BOOL, EMPTY
}; };
table(std::initializer_list<set> l); struct set {
set operator[](short); short type;
bool exists(short); F format;
std::string name(short); A action;
private: std::string name;
std::map<short, set> left; std::string id;
};
table(std::initializer_list<set> l);
const table::set operator[](std::string);
const table::set operator[](short);
bool exists(std::string);
bool exists(short);
short type(std::string);
std::string id(short);
std::string name(short);
private:
std::vector<set> data;
std::map<short, set*> left;
std::map<std::string, set*> right;
}; };
#endif /* LOOKUPTABLE_H_ */ #endif /* LOOKUPTABLE_H_ */

18
test.sh Executable file
View file

@ -0,0 +1,18 @@
#!/bin/sh
clear
echo SmrtLink Test
echo bin/smrtlink encode 00:11:22:33:44:55:66:77:88:99
bin/smrtlink encode 00:11:22:33:44:55:66:77:88:9
echo bin/smrtlink -i eth0 list
bin/smrtlink -i eth0 list
echo bin/smrtlink -i eth6 list
bin/smrtlink -i eth6 list
echo bin/smrtlink list
bin/smrtlink list
echo bin/smrtlink -i wlan0 list
bin/smrtlink -i wlan0 list