Compare commits
18 commits
Author | SHA1 | Date | |
---|---|---|---|
1f3730ac87 | |||
f15abc71be | |||
69198cd6fe | |||
e494a7db91 | |||
507cfe2140 | |||
a90401ddeb | |||
49ab8a307e | |||
5b7014737a | |||
59796d600b | |||
164edcdad9 | |||
a42a0aa8ce | |||
2952953f5d | |||
0c289aff9c | |||
8200103514 | |||
d3b89e3b11 | |||
693d8d42ff | |||
922ac9f88e | |||
057a7b4344 |
37 changed files with 1602 additions and 1153 deletions
3
Makefile
3
Makefile
|
@ -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 $@
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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_ */
|
||||||
|
|
45
src/File.cpp
45
src/File.cpp
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
16
src/File.h
16
src/File.h
|
@ -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
54
src/Filter.cpp
Normal 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
31
src/Filter.h
Normal 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_ */
|
136
src/Host.cpp
136
src/Host.cpp
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
11
src/Host.h
11
src/Host.h
|
@ -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
46
src/Interactive.cpp
Normal 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
21
src/Interactive.h
Normal 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_ */
|
|
@ -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_ */
|
|
|
@ -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);
|
||||||
|
|
120
src/Packet.h
120
src/Packet.h
|
@ -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_ */
|
||||||
|
|
497
src/Program.cpp
497
src/Program.cpp
|
@ -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);
|
||||||
}
|
}
|
||||||
|
|
|
@ -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_ */
|
||||||
|
|
|
@ -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_);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
50
src/Socket.h
50
src/Socket.h
|
@ -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
31
src/Switch-Cmd.cpp
Normal 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;
|
||||||
|
}
|
||||||
|
|
|
@ -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();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
125
src/Switch.h
125
src/Switch.h
|
@ -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_ */
|
||||||
|
|
192
src/Types.h
192
src/Types.h
|
@ -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_ */
|
||||||
|
|
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
105
src/bytes.h
105
src/bytes.h
|
@ -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_ */
|
||||||
|
|
|
@ -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_ */
|
||||||
|
|
|
@ -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());
|
||||||
}
|
}
|
||||||
|
|
|
@ -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_ */
|
||||||
|
|
28
src/lookup.h
28
src/lookup.h
|
@ -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
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
|
@ -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
49
src/lookup/map.lst
Normal 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
|
|
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -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 ???
|
||||||
|
|
||||||
|
|
||||||
|
|
290
src/smrtlink.cpp
290
src/smrtlink.cpp
|
@ -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);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
39
src/table.h
39
src/table.h
|
@ -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
18
test.sh
Executable 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
|
Loading…
Reference in a new issue