Compare commits

..

1 Commits

Author SHA1 Message Date
TIP Automation User
3eb45c219c Chg: update image tag in helm values to v3.0.0-RC1 2023-11-27 17:38:10 +00:00
95 changed files with 1630 additions and 7895 deletions

View File

@@ -21,7 +21,7 @@ defaults:
jobs: jobs:
docker: docker:
runs-on: ubuntu-latest runs-on: ubuntu-20.04
env: env:
DOCKER_REGISTRY_URL: tip-tip-wlan-cloud-ucentral.jfrog.io DOCKER_REGISTRY_URL: tip-tip-wlan-cloud-ucentral.jfrog.io
DOCKER_REGISTRY_USERNAME: ucentral DOCKER_REGISTRY_USERNAME: ucentral

View File

@@ -11,7 +11,7 @@ defaults:
jobs: jobs:
helm-package: helm-package:
runs-on: ubuntu-latest runs-on: ubuntu-20.04
env: env:
HELM_REPO_URL: https://tip.jfrog.io/artifactory/tip-wlan-cloud-ucentral-helm/ HELM_REPO_URL: https://tip.jfrog.io/artifactory/tip-wlan-cloud-ucentral-helm/
HELM_REPO_USERNAME: ucentral HELM_REPO_USERNAME: ucentral

4
.gitignore vendored
View File

@@ -21,12 +21,10 @@ _deps
/docker-compose/.env /docker-compose/.env
/docker-compose/.env_* /docker-compose/.env_*
/cmake-build/ /cmake-build/
/uploads/
test_scripts/curl/token.json test_scripts/curl/token.json
.vscode/c_cpp_properties.json .vscode/c_cpp_properties.json
test_scripts/curl/result.json test_scripts/curl/result.json
*.swp *.swp
helm/charts/* helm/charts/*
!helm/charts/.gitkeep !helm/charts/.gitkeep
/portal-test/
/src/ow_version.h

2
.idea/.gitignore generated vendored
View File

@@ -6,5 +6,3 @@
/dataSources.local.xml /dataSources.local.xml
# Editor-based HTTP Client requests # Editor-based HTTP Client requests
/httpRequests/ /httpRequests/
# GitHub Copilot persisted chat sessions
/copilot/chatSessions

3
.idea/misc.xml generated
View File

@@ -1,8 +1,5 @@
<?xml version="1.0" encoding="UTF-8"?> <?xml version="1.0" encoding="UTF-8"?>
<project version="4"> <project version="4">
<component name="Black">
<option name="sdkName" value="Python 3.9 (wlan-cloud-ucentralgw)" />
</component>
<component name="CMakeWorkspace" PROJECT_DIR="$PROJECT_DIR$" /> <component name="CMakeWorkspace" PROJECT_DIR="$PROJECT_DIR$" />
<component name="CidrRootsConfiguration"> <component name="CidrRootsConfiguration">
<excludeRoots> <excludeRoots>

View File

@@ -1,5 +1,5 @@
# Building from source # Building from source
In order to build OWGW, you will need to install its dependencies, which includes the following: In order to build the OWGW, you will need to install its dependencies, which includes the following:
- cmake - cmake
- boost - boost
- POCO 1.10.1 or later - POCO 1.10.1 or later
@@ -12,43 +12,43 @@ In order to build OWGW, you will need to install its dependencies, which include
The build is done in 2 parts. The first part is to build a local copy of the framework tailored to your environment. This The build is done in 2 parts. The first part is to build a local copy of the framework tailored to your environment. This
framework is called [Poco](https://github.com/pocoproject/poco). The version used in this project has a couple of fixes framework is called [Poco](https://github.com/pocoproject/poco). The version used in this project has a couple of fixes
from the master copy needed for cmake. Please use the version of this [Poco fix](https://github.com/Telecominfraproject/wlan-cloud-lib-poco). Building from the master copy needed for cmake. Please use the version of this [Poco fix](https://github.com/AriliaWireless/poco). Building
Poco may take several minutes depending on the platform you are building on. Poco may take several minutes depending on the platform you are building on.
## Ubuntu ## Ubuntu
These instructions have proven to work on Ubuntu 20.4. These instructions have proven to work on Ubuntu 20.4.
```bash ```bash
sudo apt install git cmake g++ libssl-dev libmariadb-dev \ sudo apt install git cmake g++ libssl-dev libmariadb-dev
libpq-dev libaprutil1-dev apache2-dev libboost-all-dev \ sudo apt install libpq-dev libaprutil1-dev apache2-dev libboost-all-dev
librdkafka-dev // default-libmysqlclient-dev \ sudo apt install librdkafka-dev // default-libmysqlclient-dev
nlohmann-json-dev sudo apt install nlohmann-json-dev
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-poco --branch poco-tip-v1 poco cd ~
git clone https://github.com/AriliaWireless/poco --branch poco-tip-v2
cd poco cd poco
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
cd ../..
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-cppkafka --branch tip-v1 cppkafka cd ~
git clone https://github.com/AriliaWireless/cppkafka --branch tip-v1
cd cppkafka cd cppkafka
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
cd ../..
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-valijson --branch tip-v1 valijson cd ~
git clone https://github.com/AriliaWireless/valijson --branch tip-v1
cd valijson cd valijson
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
cd ../..
git clone https://github.com/fmtlib/fmt --branch 9.0.0 /fmtlib git clone https://github.com/fmtlib/fmt --branch 9.0.0 /fmtlib
cd fmtlib cd fmtlib
@@ -57,59 +57,56 @@ cd cmake-build
cmake .. cmake ..
make make
make install make install
cd ../..
cd ~
git clone https://github.com/Telecominfraproject/wlan-cloud-ucentralgw git clone https://github.com/Telecominfraproject/wlan-cloud-ucentralgw
cd wlan-cloud-ucentralgw cd wlan-cloud-ucentralgw
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
make -j 8 make -j 8
cd ../..
``` ```
## Fedora ## Fedora
The following instructions have proven to work on Fedora 33 The following instructions have proven to work on Fedora 33
```bash ```bash
sudo yum install cmake g++ openssl-devel mysql-devel mysql apr-util-devel boost boost-devel \ sudo yum install cmake g++ openssl-devel mysql-devel mysql apr-util-devel boost boost-devel
yaml-cpp-devel lua-devel sudo yum install yaml-cpp-devel lua-devel
sudo dnf install postgresql.x86_64 librdkafka-devel sudo dnf install postgresql.x86_64 librdkafka-devel
sudo dnf install postgresql-devel json-devel sudo dnf install postgresql-devel json-devel
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-poco --branch poco-tip-v1 poco git clone https://github.com/AriliaWireless/poco --branch poco-tip-v2
cd poco cd poco
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
cd ../..
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-cppkafka --branch tip-v1 cppkafka git clone https://github.com/AriliaWireless/cppkafka --branch tip-v1
cd cppkafka cd cppkafka
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
cd ../..
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-valijson --branch tip-v1 valijson cd ~
git clone https://github.com/AriliaWireless/valijson --branch tip-v1
cd valijson cd valijson
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
cd ../..
cd ~
git clone https://github.com/Telecominfraproject/wlan-cloud-ucentralgw git clone https://github.com/Telecominfraproject/wlan-cloud-ucentralgw
cd wlan-cloud-ucentralgw cd wlan-cloud-ucentralgw
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
make make
cd ../..
``` ```
## macOS Build ## macOS Build
@@ -128,7 +125,7 @@ brew install openssl \
nlohmann-json \ nlohmann-json \
fmt fmt
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-poco --branch poco-tip-v1 poco git clone https://github.com/AriliaWireless/poco --branch poco-tip-v2
pushd poco pushd poco
mkdir cmake-build mkdir cmake-build
push cmake-build push cmake-build
@@ -138,7 +135,7 @@ sudo cmake --build . --target install
popd popd
popd popd
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-cppkafka --branch tip-v1 cppkafka git clone https://github.com/AriliaWireless/cppkafka --branch tip-v1
pushd cppkafka pushd cppkafka
mkdir cmake-build mkdir cmake-build
pushd cmake-build pushd cmake-build
@@ -148,10 +145,10 @@ sudo cmake --build . --target install
popd popd
popd popd
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-valijson --branch tip-v1 valijson git clone https://github.com/AriliaWireless/valijson --branch tip-v1
pushd valijson cd valijson
mkdir cmake-build mkdir cmake-build
pushd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
@@ -174,23 +171,20 @@ support. You can build with only SQLite support by not installing the packages f
adding -DSMALL_BUILD=1 on the cmake build line. adding -DSMALL_BUILD=1 on the cmake build line.
```bash ```bash
sudo apt install git cmake g++ libssl-dev libaprutil1-dev apache2-dev \ sudo apt install git cmake g++ libssl-dev libaprutil1-dev apache2-dev libboost-all-dev libyaml-cpp-dev
libboost-all-dev libyaml-cpp-dev git clone https://github.com/stephb9959/poco
git clone https://github.com/Telecominfraproject/wlan-cloud-lib-poco --branch poco-tip-v1 poco
cd poco cd poco
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake .. cmake ..
cmake --build . --config Release cmake --build . --config Release
sudo cmake --build . --target install sudo cmake --build . --target install
cd ../..
cd ~
git clone https://github.com/Telecominfraproject/wlan-cloud-ucentralgw git clone https://github.com/Telecominfraproject/wlan-cloud-ucentralgw
cd wlan-cloud-ucentralgw cd wlan-cloud-ucentralgw
mkdir cmake-build mkdir cmake-build
cd cmake-build cd cmake-build
cmake -DSMALL_BUILD=1 .. cmake -DSMALL_BUILD=1 ..
make make
cd ../..
``` ```

View File

@@ -1,8 +1,7 @@
cmake_minimum_required(VERSION 3.13) cmake_minimum_required(VERSION 3.13)
project(owgw VERSION 4.1.0) project(owgw VERSION 3.0.0)
set(CMAKE_CXX_STANDARD 20) set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED True)
if(UNIX AND APPLE) if(UNIX AND APPLE)
set(OPENSSL_ROOT_DIR /usr/local/opt/openssl) set(OPENSSL_ROOT_DIR /usr/local/opt/openssl)
@@ -149,7 +148,6 @@ add_executable( owgw
src/RESTAPI/RESTAPI_script_handler.cpp src/RESTAPI/RESTAPI_script_handler.h src/RESTAPI/RESTAPI_script_handler.cpp src/RESTAPI/RESTAPI_script_handler.h
src/RESTAPI/RESTAPI_regulatory.cpp src/RESTAPI/RESTAPI_regulatory.h src/RESTAPI/RESTAPI_regulatory.cpp src/RESTAPI/RESTAPI_regulatory.h
src/RESTAPI/RESTAPI_radiussessions_handler.cpp src/RESTAPI/RESTAPI_radiussessions_handler.h src/RESTAPI/RESTAPI_radiussessions_handler.cpp src/RESTAPI/RESTAPI_radiussessions_handler.h
src/storage/storage_blacklist.cpp src/storage/storage_tables.cpp src/storage/storage_logs.cpp src/storage/storage_blacklist.cpp src/storage/storage_tables.cpp src/storage/storage_logs.cpp
src/storage/storage_command.cpp src/storage/storage_healthcheck.cpp src/storage/storage_statistics.cpp src/storage/storage_command.cpp src/storage/storage_healthcheck.cpp src/storage/storage_statistics.cpp
src/storage/storage_device.cpp src/storage/storage_capabilities.cpp src/storage/storage_defconfig.cpp src/storage/storage_device.cpp src/storage/storage_capabilities.cpp src/storage/storage_defconfig.cpp
@@ -177,7 +175,7 @@ add_executable( owgw
src/SDKcalls.cpp src/SDKcalls.cpp
src/SDKcalls.h src/SDKcalls.h
src/StateUtils.cpp src/StateUtils.h src/StateUtils.cpp src/StateUtils.h
src/AP_WS_Reactor_Pool.h src/AP_WS_ReactorPool.h
src/AP_WS_Connection.h src/AP_WS_Connection.h
src/AP_WS_Connection.cpp src/AP_WS_Connection.cpp
src/TelemetryClient.h src/TelemetryClient.cpp src/TelemetryClient.h src/TelemetryClient.cpp
@@ -213,8 +211,7 @@ add_executable( owgw
src/RegulatoryInfo.cpp src/RegulatoryInfo.h src/RegulatoryInfo.cpp src/RegulatoryInfo.h
src/RADIUSSessionTracker.cpp src/RADIUSSessionTracker.h src/RADIUSSessionTracker.cpp src/RADIUSSessionTracker.h
src/libs/Scheduler.h src/libs/InterruptableSleep.h src/libs/ctpl_stl.h src/libs/Cron.h src/libs/Scheduler.h src/libs/InterruptableSleep.h src/libs/ctpl_stl.h src/libs/Cron.h
src/GenericScheduler.cpp src/GenericScheduler.h src/framework/default_device_types.h src/AP_WS_Process_rebootLog.cpp src/AP_WS_ConfigAutoUpgrader.cpp src/AP_WS_ConfigAutoUpgrader.h src/RESTAPI/RESTAPI_default_firmwares.cpp src/RESTAPI/RESTAPI_default_firmwares.h src/RESTAPI/RESTAPI_default_firmware.cpp src/RESTAPI/RESTAPI_default_firmware.h src/storage/storage_def_firmware.cpp src/firmware_revision_cache.h src/sdks/sdk_fms.h src/GenericScheduler.cpp src/GenericScheduler.h src/framework/default_device_types.h src/AP_WS_Process_rebootLog.cpp src/AP_WS_ConfigAutoUpgrader.cpp src/AP_WS_ConfigAutoUpgrader.h src/RESTAPI/RESTAPI_default_firmwares.cpp src/RESTAPI/RESTAPI_default_firmwares.h src/RESTAPI/RESTAPI_default_firmware.cpp src/RESTAPI/RESTAPI_default_firmware.h src/storage/storage_def_firmware.cpp src/firmware_revision_cache.h src/sdks/sdk_fms.h)
src/AP_WS_LookForUpgrade.cpp)
if(NOT SMALL_BUILD) if(NOT SMALL_BUILD)

View File

@@ -1,7 +1,7 @@
ARG DEBIAN_VERSION=11.5-slim ARG DEBIAN_VERSION=11.5-slim
ARG POCO_VERSION=poco-tip-v2 ARG POCO_VERSION=poco-tip-v2
ARG CPPKAFKA_VERSION=tip-v1 ARG CPPKAFKA_VERSION=tip-v1
ARG VALIJASON_VERSION=tip-v1.0.2 ARG VALIJASON_VERSION=tip-v1
ARG APP_NAME=owgw ARG APP_NAME=owgw
ARG APP_HOME_DIR=/openwifi ARG APP_HOME_DIR=/openwifi
@@ -17,8 +17,8 @@ FROM build-base AS poco-build
ARG POCO_VERSION ARG POCO_VERSION
ADD https://api.github.com/repos/Telecominfraproject/wlan-cloud-lib-poco/git/refs/tags/${POCO_VERSION} version.json ADD https://api.github.com/repos/AriliaWireless/poco/git/refs/tags/${POCO_VERSION} version.json
RUN git clone https://github.com/Telecominfraproject/wlan-cloud-lib-poco --branch ${POCO_VERSION} /poco RUN git clone https://github.com/AriliaWireless/poco --branch ${POCO_VERSION} /poco
WORKDIR /poco WORKDIR /poco
RUN mkdir cmake-build RUN mkdir cmake-build
@@ -31,8 +31,8 @@ FROM build-base AS cppkafka-build
ARG CPPKAFKA_VERSION ARG CPPKAFKA_VERSION
ADD https://api.github.com/repos/Telecominfraproject/wlan-cloud-lib-cppkafka/git/refs/tags/${CPPKAFKA_VERSION} version.json ADD https://api.github.com/repos/AriliaWireless/cppkafka/git/refs/tags/${CPPKAFKA_VERSION} version.json
RUN git clone https://github.com/Telecominfraproject/wlan-cloud-lib-cppkafka --branch ${CPPKAFKA_VERSION} /cppkafka RUN git clone https://github.com/AriliaWireless/cppkafka --branch ${CPPKAFKA_VERSION} /cppkafka
WORKDIR /cppkafka WORKDIR /cppkafka
RUN mkdir cmake-build RUN mkdir cmake-build
@@ -45,8 +45,8 @@ FROM build-base AS valijson-build
ARG VALIJASON_VERSION ARG VALIJASON_VERSION
ADD https://api.github.com/repos/Telecominfraproject/wlan-cloud-lib-valijson/git/refs/tags/${VALIJASON_VERSION} version.json ADD https://api.github.com/repos/AriliaWireless/valijson/git/refs/tags/${VALIJASON_VERSION} version.json
RUN git clone https://github.com/Telecominfraproject/wlan-cloud-lib-valijson --branch ${VALIJASON_VERSION} /valijson RUN git clone https://github.com/AriliaWireless/valijson --branch ${VALIJASON_VERSION} /valijson
WORKDIR /valijson WORKDIR /valijson
RUN mkdir cmake-build RUN mkdir cmake-build

View File

@@ -306,54 +306,8 @@ The device should answer:
}, },
"id" : <same number> "id" : <same number>
} }
```
#### Controller wants the device to apply a given fixed configuration
Controller sends this command when it requires the device to apply fixed configuration, eg. country code. The device
should respond with message indicating failure or success.
```json
{ "jsonrpc" : "2.0",
"method" : "fixedconfig",
"params" : {
"serial" : <serial number>,
"when" : Optional - <UTC time when to apply this config, 0 means immediate, this is a suggestion>
"country" : "<country-code>"
},
}
```
If AP supports compressed configuration feature by inidcating `compress_cmd=true` in its capabilities, controller
will send a compressed configuration message where configuration payload (i.e. contents of `params`) is compressed
and encoded in base64 format:
```json
{ "jsonrpc" : "2.0",
"method" : "configure",
"params" : {
"compress_64" : "<b64 encoded zlib compressed payload>",
"compress_sz" : "<size of uncompressed data in bytes>"
},
"id" : <some number>
}
```
The device should answer:
```json
{ "jsonrpc" : "2.0",
"result" : {
"serial": <serial number>,
"status": {
"error": 0 or an error number,
"text": <description of the error or success, eg. "Applied fixed config, rebooting">
},
"uuid": <UUID>
}
}
``` ```
##### The Answer ##### The Answer
The device can answer and tell the controller it has rejected certain parts of the config and potentially replaced them with The device can answer and tell the controller it has rejected certain parts of the config and potentially replaced them with
appropriate values. This could be used to allow a device to replace frequencies for the regions it is located in. The device appropriate values. This could be used to allow a device to replace frequencies for the regions it is located in. The device
@@ -401,39 +355,6 @@ The device should answer:
- 1 : the device is busy but will reboot soon. `text` may indicate why. - 1 : the device is busy but will reboot soon. `text` may indicate why.
- 2 : the device will not reboot. `text` contains information as to why. - 2 : the device will not reboot. `text` contains information as to why.
#### Controller wants to power-cycle PoE port(s)
Controller sends this command to power-cycle 1 or more PoE ports
```json
{ "jsonrpc" : "2.0" ,
"method" : "powercycle" ,
"params" : {
"serial" : <serial number> ,
"ports" : [ { "name" : "Ethernet1", "cycle" : 5000}, { "name" : "Ethernet8", "cycle" : 10000 } ],
"when" : Optional - <UTC time when to reboot, 0 mean immediately, this is a suggestion>
},
"id" : <some number>
}
```
The device should answer:
```json
{ "jsonrpc" : "2.0" ,
"result" : {
"serial" : <serial number> ,
"status" : {
"error" : 0 or an error number,
"text" : [ "Error 1" , "Error 2" ],
"when" : <time when this will be performed as UTC seconds>,
},
"id" : <same id from request>
}
```
###### Error codes
- 0 : is rebooting at `when` seconds.
- 1 : the device is busy but will reboot soon. `text` may indicate why.
- 2 : the device will not reboot. `text` contains information as to why.
#### Controller wants the device to upgrade its firmware #### Controller wants the device to upgrade its firmware
Controller sends this command when it believes the device should upgrade its firmware. Controller sends this command when it believes the device should upgrade its firmware.
```json ```json
@@ -880,32 +801,6 @@ The device should answer:
} }
``` ```
#### Controller wants the device to perform re-enrollment
Controller sends this command to trigger re-enrollment, i.e. update of operational certificate. Extreme care must be taken.
```json
{ "jsonrpc" : "2.0" ,
"method" : "reenroll" ,
"params" : {
"serial" : <serial number>,
"when" : Optional - <UTC time when to apply this config, 0 mean immediate, this is a suggestion>
},
"id" : <some number>
}
```
The device should answer:
```json
{ "jsonrpc" : "2.0" ,
"result" : {
"serial" : <serial number> ,
"status" : {
"error" : <0 or the value of $? from the shell running the command, 255 signifies a timeout>,
"txt" : <text describing the error or success>
},
"id" : <same number as request>
}
```
#### Controller wants the device to switch to another controller #### Controller wants the device to switch to another controller
Controller sends this when the device should change the controller it connects to without looking up a new redirector. Controller sends this when the device should change the controller it connects to without looking up a new redirector.

2
build
View File

@@ -1 +1 @@
3 10

View File

@@ -9,7 +9,7 @@ fullnameOverride: ""
images: images:
owgw: owgw:
repository: tip-tip-wlan-cloud-ucentral.jfrog.io/owgw repository: tip-tip-wlan-cloud-ucentral.jfrog.io/owgw
tag: master tag: v3.0.0-RC1
pullPolicy: Always pullPolicy: Always
# regcred: # regcred:
# registry: tip-tip-wlan-cloud-ucentral.jfrog.io # registry: tip-tip-wlan-cloud-ucentral.jfrog.io

View File

@@ -1,85 +0,0 @@
{
"ethernet": [
{
"select-ports": [
"Ethernet0",
"Ethernet1",
"Ethernet2",
"Ethernet3",
"Ethernet4",
"Ethernet5",
"Ethernet6",
"Ethernet7"
],
"speed": 2500,
"duplex": "full",
"enabled": true,
"poe": {
"admin-mode": true,
"power-limit": 60000
}
},
{
"select-ports": [
"Ethernet8",
"Ethernet9"
],
"speed": 10000,
"duplex": "full",
"media": "sfp-forced-1000sfp"
}
],
"interfaces": [
{
"name": "VLAN1",
"vlan": {
"id": 1
},
"ipv4": {
"addressing": "dynamic"
},
"ethernet": [
{
"select-ports": [
"Ethernet0",
"Ethernet1",
"Ethernet2",
"Ethernet3",
"Ethernet4",
"Ethernet5",
"Ethernet6",
"Ethernet7",
"Ethernet8",
"Ethernet9"
],
"vlan-tag": "un-tagged"
}
]
}
],
"metrics": {
"dhcp-snooping": {
"filters": [
"ack",
"discover",
"offer",
"request",
"solicit",
"reply",
"renew"
]
},
"health": {
"interval": 60
},
"statistics": {
"interval": 120,
"types": []
}
},
"unit": {
"leds-active": true,
"usage-threshold": 95
},
"uuid": 1678263900
}

View File

@@ -12,7 +12,7 @@ info:
url: https://www.ucentral.info/support url: https://www.ucentral.info/support
servers: servers:
- url: 'https://localhost:16002/api/v1' - url: 'https://localhost:16001/api/v1'
security: security:
- bearerAuth: [] - bearerAuth: []
@@ -42,10 +42,12 @@ components:
schemas: schemas:
DeviceType: DeviceType:
type: string type: string
default: ap default: AP
enum: enum:
- ap - AP
- switch - SWITCH
- IOT
- MESH
DeviceRestrictionsKeyInfo: DeviceRestrictionsKeyInfo:
type: object type: object
@@ -155,9 +157,6 @@ components:
lastRecordedContact: lastRecordedContact:
type: integer type: integer
format: int64 format: int64
blackListed:
type: boolean
readOnly: true
DeviceWithStatus: DeviceWithStatus:
type: object type: object
@@ -282,9 +281,6 @@ components:
format: float format: float
connectReason: connectReason:
type: string type: string
blackListed:
type: boolean
readOnly: true
DeviceList: DeviceList:
type: object type: object
@@ -549,12 +545,6 @@ components:
lastModified: lastModified:
type: integer type: integer
format: int64 format: int64
platform:
type: string
enum:
- ap
- switch
default: ap
DefaultConfigurationList: DefaultConfigurationList:
properties: properties:
@@ -1576,39 +1566,6 @@ components:
format: base64 format: base64
description: This is a base64 encoded string of the certificate bundle (the current bundle .tar.gz file from the PKI portal) description: This is a base64 encoded string of the certificate bundle (the current bundle .tar.gz file from the PKI portal)
ReenrollRequest:
type: object
properties:
serialNumber:
type: string
when:
type: integer
format: int64
PowerCycleRequest:
type: object
properties:
serial:
type: string
when:
type: integer
format: int64
ports:
type: array
items:
type: object
properties:
name:
type: string
example:
- Ethernet0
cycle:
type: integer
default: 10000
minimum: 1
maximum: 60000
description: off time in milliseconds
paths: paths:
/devices: /devices:
get: get:
@@ -1700,22 +1657,6 @@ paths:
type: integer type: integer
default: 70 default: 70
required: false required: false
- in: query
description: return only devices matching a certain platform of AP or SWITCH
name: platform
schema:
type: string
default: ALL
enum:
- all
- ap
- switch
required: false
- in: query
description: only devices which are not provisioned
name: includeProvisioned
schema:
type: boolean
responses: responses:
200: 200:
description: List devices description: List devices
@@ -3065,60 +3006,6 @@ paths:
404: 404:
$ref: '#/components/responses/NotFound' $ref: '#/components/responses/NotFound'
/device/{serialNumber}/reenroll:
post:
tags:
- Commands
summary: Reenroll operational certificate for the device.
operationId: reenrollCertificate
parameters:
- in: path
name: serialNumber
schema:
type: string
required: true
requestBody:
description: Reenroll operational certificate for the device
content:
application/json:
schema:
$ref: '#/components/schemas/ReenrollRequest'
responses:
200:
$ref: '#/components/responses/Success'
403:
$ref: '#/components/responses/Unauthorized'
404:
$ref: '#/components/responses/NotFound'
/device/{serialNumber}/powercycle:
post:
tags:
- Commands
summary: Perform PoE power cycle for some PoE ports.
operationId: performPowerCycle
parameters:
- in: path
name: serialNumber
schema:
type: string
required: true
requestBody:
description: Certificate update details
content:
application/json:
schema:
type: array
items:
$ref: '#/components/schemas/PowerCycleRequest'
responses:
200:
$ref: '#/components/responses/Success'
403:
$ref: '#/components/responses/Unauthorized'
404:
$ref: '#/components/responses/NotFound'
/ouis: /ouis:
get: get:
tags: tags:

View File

@@ -145,7 +145,7 @@ storage.type.sqlite.db = devices.db
storage.type.sqlite.idletime = 120 storage.type.sqlite.idletime = 120
storage.type.sqlite.maxsessions = 128 storage.type.sqlite.maxsessions = 128
storage.type.postgresql.maxsessions = 250 storage.type.postgresql.maxsessions = 64
storage.type.postgresql.idletime = 60 storage.type.postgresql.idletime = 60
storage.type.postgresql.host = ${STORAGE_TYPE_POSTGRESQL_HOST} storage.type.postgresql.host = ${STORAGE_TYPE_POSTGRESQL_HOST}
storage.type.postgresql.username = ${STORAGE_TYPE_POSTGRESQL_USERNAME} storage.type.postgresql.username = ${STORAGE_TYPE_POSTGRESQL_USERNAME}

View File

@@ -9,14 +9,14 @@
namespace OpenWifi { namespace OpenWifi {
int AP_WS_ConfigAutoUpgradeAgent::Start() { int AP_WS_ConfigAutoUpgrader::Start() {
poco_notice(Logger(), "Starting..."); poco_notice(Logger(), "Starting...");
QueueManager_.start(*this); QueueManager_.start(*this);
return 0; return 0;
} }
void AP_WS_ConfigAutoUpgradeAgent::Stop() { void AP_WS_ConfigAutoUpgrader::Stop() {
poco_notice(Logger(), "Stopping..."); poco_notice(Logger(), "Stopping...");
Running_ = false; Running_ = false;
Queue_.wakeUpAll(); Queue_.wakeUpAll();
@@ -24,7 +24,7 @@ namespace OpenWifi {
poco_notice(Logger(), "Stopped..."); poco_notice(Logger(), "Stopped...");
} }
void AP_WS_ConfigAutoUpgradeAgent::run() { void AP_WS_ConfigAutoUpgrader::run() {
Utils::SetThreadName("auto:cfgmgr"); Utils::SetThreadName("auto:cfgmgr");
Running_ = true; Running_ = true;

View File

@@ -28,14 +28,14 @@ namespace OpenWifi {
std::uint64_t pending_config_=0; std::uint64_t pending_config_=0;
}; };
class AP_WS_ConfigAutoUpgradeAgent : public SubSystemServer, Poco::Runnable { class AP_WS_ConfigAutoUpgrader : public SubSystemServer, Poco::Runnable {
public: public:
int Start() final; int Start() final;
void Stop() final; void Stop() final;
void run() final; void run() final;
static auto instance() { static auto instance() {
static auto instance = new AP_WS_ConfigAutoUpgradeAgent; static auto instance = new AP_WS_ConfigAutoUpgrader;
return instance; return instance;
} }
@@ -126,12 +126,12 @@ namespace OpenWifi {
mutable std::mutex CacheMutex_; mutable std::mutex CacheMutex_;
std::map<std::uint64_t, ConfigurationCacheEntry> Cache_; std::map<std::uint64_t, ConfigurationCacheEntry> Cache_;
AP_WS_ConfigAutoUpgradeAgent() noexcept AP_WS_ConfigAutoUpgrader() noexcept
: SubSystemServer("AutoConfigUpgrade", "AUTO-CFG-MGR", "auto.config.updater") { : SubSystemServer("AutoConfigUpgrade", "AUTO-CFG-MGR", "auto.config.updater") {
} }
}; };
inline auto AP_WS_ConfigAutoUpgradeAgent() { return AP_WS_ConfigAutoUpgradeAgent::instance(); } inline auto AP_WS_ConfigAutoUpgrader() { return AP_WS_ConfigAutoUpgrader::instance(); }
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -2,49 +2,56 @@
// Created by stephane bourque on 2022-02-03. // Created by stephane bourque on 2022-02-03.
// //
#include "AP_WS_Connection.h"
#include <Poco/Base64Decoder.h> #include "Poco/Base64Decoder.h"
#include <Poco/Net/Context.h> #include "Poco/Net/Context.h"
#include <Poco/Net/HTTPServerRequestImpl.h> #include "Poco/Net/HTTPServerRequestImpl.h"
#include <Poco/Net/HTTPServerResponseImpl.h> #include "Poco/Net/HTTPServerResponseImpl.h"
#include <Poco/Net/NetException.h> #include "Poco/Net/NetException.h"
#include <Poco/Net/SSLException.h> #include "Poco/Net/SSLException.h"
#include <Poco/Net/SecureStreamSocketImpl.h> #include "Poco/Net/SecureStreamSocketImpl.h"
#include <Poco/Net/WebSocketImpl.h> #include "Poco/Net/WebSocketImpl.h"
#include "Poco/zlib.h"
#include <framework/KafkaManager.h> #include "AP_WS_Server.h"
#include <framework/MicroServiceFuncs.h> #include "CentralConfig.h"
#include <framework/utils.h> #include "CommandManager.h"
#include <framework/ow_constants.h> #include "ConfigurationCache.h"
#include "StorageService.h"
#include "TelemetryStream.h"
#include <fmt/format.h> #include "GWKafkaEvents.h"
#include "UI_GW_WebSocketNotifications.h"
#include "framework/KafkaManager.h"
#include "framework/MicroServiceFuncs.h"
#include "framework/utils.h"
#include <AP_WS_Connection.h> #include "fmt/format.h"
#include <AP_WS_Server.h>
#include <CentralConfig.h> #include "framework/ow_constants.h"
#include <CommandManager.h>
#include <StorageService.h> #include "RADIUSSessionTracker.h"
#include <RADIUSSessionTracker.h> #include "RADIUS_proxy_server.h"
#include <RADIUS_proxy_server.h>
#include <GWKafkaEvents.h>
#include <UI_GW_WebSocketNotifications.h>
namespace OpenWifi { namespace OpenWifi {
#define DBL \
{ \
std::cout << __LINE__ << " ID: " << ConnectionId_ << " Ser: " << SerialNumber_ \
<< std::endl; \
}
void AP_WS_Connection::LogException(const Poco::Exception &E) { void AP_WS_Connection::LogException(const Poco::Exception &E) {
poco_information(Logger_, fmt::format("EXCEPTION({}): {}", CId_, E.displayText())); poco_information(Logger_, fmt::format("EXCEPTION({}): {}", CId_, E.displayText()));
} }
AP_WS_Connection::AP_WS_Connection(Poco::Net::HTTPServerRequest &request, AP_WS_Connection::AP_WS_Connection(Poco::Net::HTTPServerRequest &request,
Poco::Net::HTTPServerResponse &response, Poco::Net::HTTPServerResponse &response,
uint64_t session_id, Poco::Logger &L, uint64_t connection_id, Poco::Logger &L,
std::pair<std::shared_ptr<Poco::Net::SocketReactor>, std::shared_ptr<LockedDbSession>> R) Poco::Net::SocketReactor &R)
: Logger_(L) { : Logger_(L), Reactor_(R) {
State_.sessionId = connection_id;
Reactor_ = R.first;
DbSession_ = R.second;
State_.sessionId = session_id;
WS_ = std::make_unique<Poco::Net::WebSocket>(request, response); WS_ = std::make_unique<Poco::Net::WebSocket>(request, response);
auto TS = Poco::Timespan(360, 0); auto TS = Poco::Timespan(360, 0);
@@ -54,86 +61,29 @@ namespace OpenWifi {
WS_->setNoDelay(false); WS_->setNoDelay(false);
WS_->setKeepAlive(true); WS_->setKeepAlive(true);
WS_->setBlocking(false); WS_->setBlocking(false);
uuid_ = MicroServiceRandom(std::numeric_limits<std::uint64_t>::max()-1);
AP_WS_Server()->IncrementConnectionCount(); Reactor_.addEventHandler(*WS_,
}
void AP_WS_Connection::Start() {
Registered_ = true;
LastContact_ = Utils::Now();
Reactor_->addEventHandler(*WS_,
Poco::NObserver<AP_WS_Connection, Poco::Net::ReadableNotification>( Poco::NObserver<AP_WS_Connection, Poco::Net::ReadableNotification>(
*this, &AP_WS_Connection::OnSocketReadable)); *this, &AP_WS_Connection::OnSocketReadable));
Reactor_->addEventHandler(*WS_, Reactor_.addEventHandler(*WS_,
Poco::NObserver<AP_WS_Connection, Poco::Net::ShutdownNotification>( Poco::NObserver<AP_WS_Connection, Poco::Net::ShutdownNotification>(
*this, &AP_WS_Connection::OnSocketShutdown)); *this, &AP_WS_Connection::OnSocketShutdown));
Reactor_->addEventHandler(*WS_, Reactor_.addEventHandler(*WS_,
Poco::NObserver<AP_WS_Connection, Poco::Net::ErrorNotification>( Poco::NObserver<AP_WS_Connection, Poco::Net::ErrorNotification>(
*this, &AP_WS_Connection::OnSocketError)); *this, &AP_WS_Connection::OnSocketError));
Registered_ = true;
} Valid_ = true;
uuid_ = MicroServiceRandom(std::numeric_limits<std::uint64_t>::max()-1);
AP_WS_Connection::~AP_WS_Connection() {
std::lock_guard G(ConnectionMutex_);
AP_WS_Server()->DecrementConnectionCount();
EndConnection();
poco_debug(Logger_, fmt::format("TERMINATION({}): Session={}, Connection removed.", SerialNumber_,
State_.sessionId));
}
static void NotifyKafkaDisconnect(const std::string &SerialNumber, std::uint64_t uuid) {
try {
Poco::JSON::Object Disconnect;
Poco::JSON::Object Details;
Details.set(uCentralProtocol::SERIALNUMBER, SerialNumber);
Details.set(uCentralProtocol::TIMESTAMP, Utils::Now());
Details.set(uCentralProtocol::UUID,uuid);
Disconnect.set(uCentralProtocol::DISCONNECTION, Details);
KafkaManager()->PostMessage(KafkaTopics::CONNECTION, SerialNumber, Disconnect);
} catch (...) {
}
}
void AP_WS_Connection::EndConnection() {
bool expectedValue=false;
if (Dead_.compare_exchange_strong(expectedValue,true,std::memory_order_release,std::memory_order_relaxed)) {
if(!SerialNumber_.empty() && State_.LastContact!=0) {
StorageService()->SetDeviceLastRecordedContact(SerialNumber_, State_.LastContact);
}
if (Registered_) {
Registered_ = false;
Reactor_->removeEventHandler(
*WS_, Poco::NObserver<AP_WS_Connection, Poco::Net::ReadableNotification>(
*this, &AP_WS_Connection::OnSocketReadable));
Reactor_->removeEventHandler(
*WS_, Poco::NObserver<AP_WS_Connection, Poco::Net::ShutdownNotification>(
*this, &AP_WS_Connection::OnSocketShutdown));
Reactor_->removeEventHandler(
*WS_, Poco::NObserver<AP_WS_Connection, Poco::Net::ErrorNotification>(
*this, &AP_WS_Connection::OnSocketError));
Registered_=false;
}
WS_->close();
if(!SerialNumber_.empty()) {
DeviceDisconnectionCleanup(SerialNumber_, uuid_);
}
AP_WS_Server()->AddCleanupSession(State_.sessionId, SerialNumberInt_);
}
} }
bool AP_WS_Connection::ValidatedDevice() { bool AP_WS_Connection::ValidatedDevice() {
if(Dead_)
return false;
if (DeviceValidated_) if (DeviceValidated_)
return true; return true;
if (!Valid_)
return false;
std::lock_guard Lock(ConnectionMutex_);
try { try {
auto SockImpl = dynamic_cast<Poco::Net::WebSocketImpl *>(WS_->impl()); auto SockImpl = dynamic_cast<Poco::Net::WebSocketImpl *>(WS_->impl());
auto SS = auto SS =
@@ -148,6 +98,7 @@ namespace OpenWifi {
poco_warning(Logger_, fmt::format("TLS-CONNECTION({}): Session={} Connection is " poco_warning(Logger_, fmt::format("TLS-CONNECTION({}): Session={} Connection is "
"NOT secure. Device is not allowed.", "NOT secure. Device is not allowed.",
CId_, State_.sessionId)); CId_, State_.sessionId));
EndConnection();
return false; return false;
} }
@@ -160,6 +111,7 @@ namespace OpenWifi {
Logger_, Logger_,
fmt::format("TLS-CONNECTION({}): Session={} No certificates available..", CId_, fmt::format("TLS-CONNECTION({}): Session={} No certificates available..", CId_,
State_.sessionId)); State_.sessionId));
EndConnection();
return false; return false;
} }
@@ -170,19 +122,11 @@ namespace OpenWifi {
fmt::format("TLS-CONNECTION({}): Session={} Device certificate is not " fmt::format("TLS-CONNECTION({}): Session={} Device certificate is not "
"valid. Device is not allowed.", "valid. Device is not allowed.",
CId_, State_.sessionId)); CId_, State_.sessionId));
EndConnection();
return false; return false;
} }
CN_ = Poco::trim(Poco::toLower(PeerCert.commonName())); CN_ = Poco::trim(Poco::toLower(PeerCert.commonName()));
if(!Utils::ValidSerialNumber(CN_)) {
poco_trace(Logger_,
fmt::format("TLS-CONNECTION({}): Session={} Invalid serial number: CN={}", CId_,
State_.sessionId, CN_));
return false;
}
SerialNumber_ = CN_;
SerialNumberInt_ = Utils::SerialNumberToInt(SerialNumber_);
State_.VerifiedCertificate = GWObjects::VALID_CERTIFICATE; State_.VerifiedCertificate = GWObjects::VALID_CERTIFICATE;
poco_trace(Logger_, poco_trace(Logger_,
fmt::format("TLS-CONNECTION({}): Session={} Valid certificate: CN={}", CId_, fmt::format("TLS-CONNECTION({}): Session={} Valid certificate: CN={}", CId_,
@@ -192,28 +136,31 @@ namespace OpenWifi {
poco_warning(Logger_, fmt::format("TLS-CONNECTION({}): Session={} Sim Device {} is " poco_warning(Logger_, fmt::format("TLS-CONNECTION({}): Session={} Sim Device {} is "
"not allowed. Disconnecting.", "not allowed. Disconnecting.",
CId_, State_.sessionId, CN_)); CId_, State_.sessionId, CN_));
EndConnection();
return false; return false;
} }
if(AP_WS_Server::IsSim(SerialNumber_)) { if(AP_WS_Server::IsSim(CN_)) {
State_.VerifiedCertificate = GWObjects::SIMULATED; State_.VerifiedCertificate = GWObjects::SIMULATED;
Simulated_ = true; Simulated_ = true;
} }
std::string reason, author; std::string reason, author;
std::uint64_t created; std::uint64_t created;
if (!CN_.empty() && StorageService()->IsBlackListed(SerialNumberInt_, reason, author, created)) { if (!CN_.empty() && StorageService()->IsBlackListed(CN_, reason, author, created)) {
DeviceBlacklistedKafkaEvent KE(Utils::SerialNumberToInt(CN_), Utils::Now(), reason, author, created, CId_); DeviceBlacklistedKafkaEvent KE(Utils::SerialNumberToInt(CN_), Utils::Now(), reason, author, created, CId_);
poco_warning( poco_warning(
Logger_, Logger_,
fmt::format( fmt::format(
"TLS-CONNECTION({}): Session={} Device {} is black listed. Disconnecting.", "TLS-CONNECTION({}): Session={} Device {} is black listed. Disconnecting.",
CId_, State_.sessionId, CN_)); CId_, State_.sessionId, CN_));
EndConnection();
return false; return false;
} }
State_.certificateExpiryDate = PeerCert.expiresOn().timestamp().epochTime(); State_.certificateExpiryDate = PeerCert.expiresOn().timestamp().epochTime();
State_.certificateIssuerName = PeerCert.issuerName(); SerialNumber_ = CN_;
SerialNumberInt_ = Utils::SerialNumberToInt(SerialNumber_);
poco_trace(Logger_, poco_trace(Logger_,
fmt::format("TLS-CONNECTION({}): Session={} CN={} Completed. (t={})", CId_, fmt::format("TLS-CONNECTION({}): Session={} CN={} Completed. (t={})", CId_,
@@ -277,15 +224,150 @@ namespace OpenWifi {
return false; return false;
} }
void AP_WS_Connection::DeviceDisconnectionCleanup(const std::string &SerialNumber, std::uint64_t uuid) { static void NotifyKafkaDisconnect(const std::string &SerialNumber, std::uint64_t uuid) {
try {
Poco::JSON::Object Disconnect;
Poco::JSON::Object Details;
Details.set(uCentralProtocol::SERIALNUMBER, SerialNumber);
Details.set(uCentralProtocol::TIMESTAMP, Utils::Now());
Details.set(uCentralProtocol::UUID,uuid);
Disconnect.set(uCentralProtocol::DISCONNECTION, Details);
KafkaManager()->PostMessage(KafkaTopics::CONNECTION, SerialNumber, Disconnect);
} catch (...) {
}
}
AP_WS_Connection::~AP_WS_Connection() {
Valid_ = false;
EndConnection();
}
void DeviceDisconnectionCleanup(const std::string &SerialNumber, std::uint64_t uuid) {
if (KafkaManager()->Enabled()) { if (KafkaManager()->Enabled()) {
NotifyKafkaDisconnect(SerialNumber, uuid); NotifyKafkaDisconnect(SerialNumber, uuid);
} }
RADIUSSessionTracker()->DeviceDisconnect(SerialNumber); RADIUSSessionTracker()->DeviceDisconnect(SerialNumber);
}
void AP_WS_Connection::EndConnection(bool DeleteSession) {
Valid_ = false;
if (!Dead_.test_and_set()) {
if(!SerialNumber_.empty() && State_.LastContact!=0) {
StorageService()->SetDeviceLastRecordedContact(SerialNumber_, State_.LastContact);
}
if (Registered_) {
Registered_ = false;
Reactor_.removeEventHandler(
*WS_, Poco::NObserver<AP_WS_Connection, Poco::Net::ReadableNotification>(
*this, &AP_WS_Connection::OnSocketReadable));
Reactor_.removeEventHandler(
*WS_, Poco::NObserver<AP_WS_Connection, Poco::Net::ShutdownNotification>(
*this, &AP_WS_Connection::OnSocketShutdown));
Reactor_.removeEventHandler(
*WS_, Poco::NObserver<AP_WS_Connection, Poco::Net::ErrorNotification>(
*this, &AP_WS_Connection::OnSocketError));
}
WS_->close();
if(!SerialNumber_.empty()) {
std::thread Cleanup(DeviceDisconnectionCleanup,SerialNumber_, uuid_);
Cleanup.detach();
}
bool SessionDeleted = false;
if(DeleteSession)
SessionDeleted = AP_WS_Server()->EndSession(State_.sessionId, SerialNumberInt_);
if (SessionDeleted || !DeleteSession) {
GWWebSocketNotifications::SingleDevice_t N; GWWebSocketNotifications::SingleDevice_t N;
N.content.serialNumber = SerialNumber; N.content.serialNumber = SerialNumber_;
GWWebSocketNotifications::DeviceDisconnected(N); GWWebSocketNotifications::DeviceDisconnected(N);
} }
}
}
bool AP_WS_Connection::LookForUpgrade(const uint64_t UUID, uint64_t &UpgradedUUID) {
// A UUID of zero means ignore updates for that connection.
if (UUID == 0)
return false;
uint64_t GoodConfig = ConfigurationCache().CurrentConfig(SerialNumberInt_);
if (GoodConfig && (GoodConfig == UUID || GoodConfig == State_.PendingUUID)) {
UpgradedUUID = UUID;
return false;
}
GWObjects::Device D;
if (StorageService()->GetDevice(SerialNumber_, D)) {
if(D.pendingUUID!=0 && UUID==D.pendingUUID) {
// so we sent an upgrade to a device, and now it is completing now...
UpgradedUUID = D.pendingUUID;
StorageService()->CompleteDeviceConfigurationChange(SerialNumber_);
return true;
}
// This is the case where the cache is empty after a restart. So GoodConfig will 0. If
// the device already has the right UUID, we just return.
if (D.UUID == UUID) {
UpgradedUUID = UUID;
ConfigurationCache().Add(SerialNumberInt_, UUID);
return false;
}
Config::Config Cfg(D.Configuration);
if (UUID > D.UUID) {
// so we have a problem, the device has a newer config than we have. So we need to
// make sure our config is newer.
D.UUID = UUID + 2;
UpgradedUUID = D.UUID;
}
Cfg.SetUUID(D.UUID);
D.Configuration = Cfg.get();
State_.PendingUUID = UpgradedUUID = D.UUID;
GWObjects::CommandDetails Cmd;
Cmd.SerialNumber = SerialNumber_;
Cmd.UUID = MicroServiceCreateUUID();
Cmd.SubmittedBy = uCentralProtocol::SUBMITTED_BY_SYSTEM;
Cmd.Status = uCentralProtocol::PENDING;
Cmd.Command = uCentralProtocol::CONFIGURE;
Poco::JSON::Parser P;
auto ParsedConfig = P.parse(D.Configuration).extract<Poco::JSON::Object::Ptr>();
Poco::JSON::Object Params;
Params.set(uCentralProtocol::SERIAL, SerialNumber_);
Params.set(uCentralProtocol::UUID, D.UUID);
Params.set(uCentralProtocol::WHEN, 0);
Params.set(uCentralProtocol::CONFIG, ParsedConfig);
std::ostringstream O;
Poco::JSON::Stringifier::stringify(Params, O);
Cmd.Details = O.str();
poco_information(Logger_,
fmt::format("CFG-UPGRADE({}): Current ID: {}, newer configuration {}.",
CId_, UUID, D.UUID));
bool Sent;
StorageService()->AddCommand(SerialNumber_, Cmd,
Storage::CommandExecutionType::COMMAND_EXECUTED);
CommandManager()->PostCommand(
CommandManager()->Next_RPC_ID(), APCommands::to_apcommand(Cmd.Command.c_str()),
SerialNumber_, Cmd.Command, Params, Cmd.UUID, Sent, false, false);
GWWebSocketNotifications::SingleDeviceConfigurationChange_t Notification;
Notification.content.serialNumber = D.SerialNumber;
Notification.content.oldUUID = UUID;
Notification.content.newUUID = UpgradedUUID;
GWWebSocketNotifications::DeviceConfigurationChange(Notification);
return true;
}
return false;
}
void AP_WS_Connection::ProcessJSONRPCResult(Poco::JSON::Object::Ptr Doc) { void AP_WS_Connection::ProcessJSONRPCResult(Poco::JSON::Object::Ptr Doc) {
poco_trace(Logger_, fmt::format("RECEIVED-RPC({}): {}.", CId_, poco_trace(Logger_, fmt::format("RECEIVED-RPC({}): {}.", CId_,
@@ -365,7 +447,7 @@ namespace OpenWifi {
std::string reason, author; std::string reason, author;
std::uint64_t created; std::uint64_t created;
if (StorageService()->IsBlackListed(SerialNumberInt_, reason, author, created)) { if (StorageService()->IsBlackListed(Serial, reason, author, created)) {
DeviceBlacklistedKafkaEvent KE(Utils::SerialNumberToInt(CN_), Utils::Now(), reason, author, created, CId_); DeviceBlacklistedKafkaEvent KE(Utils::SerialNumberToInt(CN_), Utils::Now(), reason, author, created, CId_);
Poco::Exception E( Poco::Exception E(
fmt::format("BLACKLIST({}): device is blacklisted and not allowed to connect.", fmt::format("BLACKLIST({}): device is blacklisted and not allowed to connect.",
@@ -496,17 +578,17 @@ namespace OpenWifi {
} }
bool AP_WS_Connection::SetWebSocketTelemetryReporting( bool AP_WS_Connection::SetWebSocketTelemetryReporting(
std::uint64_t RPCID, std::uint64_t Interval, std::uint64_t LifeTime, uint64_t RPCID, uint64_t Interval, uint64_t LifeTime,
const std::vector<std::string> &TelemetryTypes) { const std::vector<std::string> &TelemetryTypes) {
std::unique_lock Lock(TelemetryMutex_); std::unique_lock Lock(TelemetryMutex_);
TelemetryWebSocketRefCount_++; TelemetryWebSocketRefCount_++;
TelemetryInterval_ = TelemetryInterval_ TelemetryInterval_ = TelemetryInterval_
? (Interval < (std::uint64_t)TelemetryInterval_ ? Interval : (std::uint64_t )TelemetryInterval_) ? (Interval < TelemetryInterval_ ? Interval : TelemetryInterval_)
: Interval; : Interval;
auto TelemetryWebSocketTimer = LifeTime + Utils::Now(); auto TelemetryWebSocketTimer = LifeTime + Utils::Now();
TelemetryWebSocketTimer_ = TelemetryWebSocketTimer > (std::uint64_t)TelemetryWebSocketTimer_ TelemetryWebSocketTimer_ = TelemetryWebSocketTimer > TelemetryWebSocketTimer_
? (std::uint64_t)TelemetryWebSocketTimer ? TelemetryWebSocketTimer
: (std::uint64_t)TelemetryWebSocketTimer_; : TelemetryWebSocketTimer_;
UpdateCounts(); UpdateCounts();
if (!TelemetryReporting_) { if (!TelemetryReporting_) {
TelemetryReporting_ = true; TelemetryReporting_ = true;
@@ -522,11 +604,11 @@ namespace OpenWifi {
std::unique_lock Lock(TelemetryMutex_); std::unique_lock Lock(TelemetryMutex_);
TelemetryKafkaRefCount_++; TelemetryKafkaRefCount_++;
TelemetryInterval_ = TelemetryInterval_ TelemetryInterval_ = TelemetryInterval_
? (Interval < (std::uint64_t)TelemetryInterval_ ? (std::uint64_t)Interval : (std::uint64_t)TelemetryInterval_) ? (Interval < TelemetryInterval_ ? Interval : TelemetryInterval_)
: Interval; : Interval;
auto TelemetryKafkaTimer = LifeTime + Utils::Now(); auto TelemetryKafkaTimer = LifeTime + Utils::Now();
TelemetryKafkaTimer_ = TelemetryKafkaTimer_ =
TelemetryKafkaTimer > (std::uint64_t)TelemetryKafkaTimer_ ? (std::uint64_t)TelemetryKafkaTimer : (std::uint64_t)TelemetryKafkaTimer_; TelemetryKafkaTimer > TelemetryKafkaTimer_ ? TelemetryKafkaTimer : TelemetryKafkaTimer_;
UpdateCounts(); UpdateCounts();
if (!TelemetryReporting_) { if (!TelemetryReporting_) {
TelemetryReporting_ = true; TelemetryReporting_ = true;
@@ -562,48 +644,47 @@ namespace OpenWifi {
void AP_WS_Connection::OnSocketShutdown( void AP_WS_Connection::OnSocketShutdown(
[[maybe_unused]] const Poco::AutoPtr<Poco::Net::ShutdownNotification> &pNf) { [[maybe_unused]] const Poco::AutoPtr<Poco::Net::ShutdownNotification> &pNf) {
poco_trace(Logger_, fmt::format("SOCKET-SHUTDOWN({}): Closing.", CId_)); poco_trace(Logger_, fmt::format("SOCKET-SHUTDOWN({}): Closing.", CId_));
// std::lock_guard G(ConnectionMutex_);
return EndConnection(); return EndConnection();
} }
void AP_WS_Connection::OnSocketError( void AP_WS_Connection::OnSocketError(
[[maybe_unused]] const Poco::AutoPtr<Poco::Net::ErrorNotification> &pNf) { [[maybe_unused]] const Poco::AutoPtr<Poco::Net::ErrorNotification> &pNf) {
poco_trace(Logger_, fmt::format("SOCKET-ERROR({}): Closing.", CId_)); poco_trace(Logger_, fmt::format("SOCKET-ERROR({}): Closing.", CId_));
// std::lock_guard G(ConnectionMutex_);
return EndConnection(); return EndConnection();
} }
void AP_WS_Connection::OnSocketReadable( void AP_WS_Connection::OnSocketReadable(
[[maybe_unused]] const Poco::AutoPtr<Poco::Net::ReadableNotification> &pNf) { [[maybe_unused]] const Poco::AutoPtr<Poco::Net::ReadableNotification> &pNf) {
if (Dead_) // we are dead, so we do not process anything. if (!Valid_)
return; return;
std::lock_guard G(ConnectionMutex_); if (!AP_WS_Server()->Running())
return EndConnection();
if (!ValidatedDevice())
return;
State_.LastContact = LastContact_ = Utils::Now();
if (AP_WS_Server()->Running() && (DeviceValidated_ || ValidatedDevice())) {
try { try {
return ProcessIncomingFrame(); return ProcessIncomingFrame();
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger_.log(E); Logger_.log(E);
return EndConnection();
} catch (const std::exception &E) { } catch (const std::exception &E) {
std::string W = E.what(); std::string W = E.what();
poco_information( poco_information(
Logger_, fmt::format("std::exception caught: {}. Connection terminated with {}", Logger_,
W, CId_)); fmt::format("std::exception caught: {}. Connection terminated with {}", W, CId_));
return EndConnection();
} catch (...) { } catch (...) {
poco_information( poco_information(Logger_,
Logger_, fmt::format("Unknown exception for {}. Connection terminated.", CId_)); fmt::format("Unknown exception for {}. Connection terminated.", CId_));
return EndConnection();
} }
} }
EndConnection();
}
void AP_WS_Connection::ProcessIncomingFrame() { void AP_WS_Connection::ProcessIncomingFrame() {
Poco::Buffer<char> IncomingFrame(0); Poco::Buffer<char> IncomingFrame(0);
bool KillConnection=false;
try { try {
int Op, flags; int Op, flags;
auto IncomingSize = WS_->receiveFrame(IncomingFrame, flags); auto IncomingSize = WS_->receiveFrame(IncomingFrame, flags);
@@ -645,10 +726,12 @@ namespace OpenWifi {
poco_trace(Logger_,fmt::format("Sending PING for {}", SerialNumber_)); poco_trace(Logger_,fmt::format("Sending PING for {}", SerialNumber_));
KafkaManager()->PostMessage(KafkaTopics::CONNECTION, SerialNumber_,PingObject); KafkaManager()->PostMessage(KafkaTopics::CONNECTION, SerialNumber_,PingObject);
} }
return;
} break; } break;
case Poco::Net::WebSocket::FRAME_OP_PONG: { case Poco::Net::WebSocket::FRAME_OP_PONG: {
poco_trace(Logger_, fmt::format("PONG({}): received and ignored.", CId_)); poco_trace(Logger_, fmt::format("PONG({}): received and ignored.", CId_));
return;
} break; } break;
case Poco::Net::WebSocket::FRAME_OP_TEXT: { case Poco::Net::WebSocket::FRAME_OP_TEXT: {
@@ -680,26 +763,26 @@ namespace OpenWifi {
} else { } else {
std::ostringstream iS; std::ostringstream iS;
IncomingJSON->stringify(iS); IncomingJSON->stringify(iS);
std::cout << iS.str() << std::endl;
poco_warning( poco_warning(
Logger_, Logger_,
fmt::format("FRAME({}): illegal transaction header, missing 'jsonrpc': {}", fmt::format("FRAME({}): illegal transaction header, missing 'jsonrpc'",
CId_, iS.str())); CId_));
Errors_++; Errors_++;
} }
return;
} break; } break;
case Poco::Net::WebSocket::FRAME_OP_CLOSE: { case Poco::Net::WebSocket::FRAME_OP_CLOSE: {
poco_information(Logger_, poco_information(Logger_,
fmt::format("CLOSE({}): Device is closing its connection.", CId_)); fmt::format("CLOSE({}): Device is closing its connection.", CId_));
KillConnection=true; return EndConnection();
} break; } break;
default: { default: {
poco_warning(Logger_, fmt::format("UNKNOWN({}): unknown WS Frame operation: {}", poco_warning(Logger_, fmt::format("UNKNOWN({}): unknown WS Frame operation: {}",
CId_, std::to_string(Op))); CId_, std::to_string(Op)));
Errors_++; } break;
return;
}
} }
} catch (const Poco::Net::ConnectionResetException &E) { } catch (const Poco::Net::ConnectionResetException &E) {
poco_warning(Logger_, poco_warning(Logger_,
@@ -707,21 +790,21 @@ namespace OpenWifi {
CId_, E.displayText(), CId_, E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const Poco::JSON::JSONException &E) { } catch (const Poco::JSON::JSONException &E) {
poco_warning(Logger_, poco_warning(Logger_,
fmt::format("JSONException({}): Text:{} Payload:{} Session:{}", CId_, fmt::format("JSONException({}): Text:{} Payload:{} Session:{}", CId_,
E.displayText(), E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const Poco::Net::WebSocketException &E) { } catch (const Poco::Net::WebSocketException &E) {
poco_warning(Logger_, poco_warning(Logger_,
fmt::format("WebSocketException({}): Text:{} Payload:{} Session:{}", CId_, fmt::format("WebSocketException({}): Text:{} Payload:{} Session:{}", CId_,
E.displayText(), E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const Poco::Net::SSLConnectionUnexpectedlyClosedException &E) { } catch (const Poco::Net::SSLConnectionUnexpectedlyClosedException &E) {
poco_warning( poco_warning(
Logger_, Logger_,
@@ -730,54 +813,54 @@ namespace OpenWifi {
CId_, E.displayText(), CId_, E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const Poco::Net::SSLException &E) { } catch (const Poco::Net::SSLException &E) {
poco_warning(Logger_, poco_warning(Logger_,
fmt::format("SSLException({}): Text:{} Payload:{} Session:{}", CId_, fmt::format("SSLException({}): Text:{} Payload:{} Session:{}", CId_,
E.displayText(), E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const Poco::Net::NetException &E) { } catch (const Poco::Net::NetException &E) {
poco_warning(Logger_, poco_warning(Logger_,
fmt::format("NetException({}): Text:{} Payload:{} Session:{}", CId_, fmt::format("NetException({}): Text:{} Payload:{} Session:{}", CId_,
E.displayText(), E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const Poco::IOException &E) { } catch (const Poco::IOException &E) {
poco_warning(Logger_, poco_warning(Logger_,
fmt::format("IOException({}): Text:{} Payload:{} Session:{}", CId_, fmt::format("IOException({}): Text:{} Payload:{} Session:{}", CId_,
E.displayText(), E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger_, poco_warning(Logger_,
fmt::format("Exception({}): Text:{} Payload:{} Session:{}", CId_, fmt::format("Exception({}): Text:{} Payload:{} Session:{}", CId_,
E.displayText(), E.displayText(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (const std::exception &E) { } catch (const std::exception &E) {
poco_warning(Logger_, poco_warning(Logger_,
fmt::format("std::exception({}): Text:{} Payload:{} Session:{}", CId_, fmt::format("std::exception({}): Text:{} Payload:{} Session:{}", CId_,
E.what(), E.what(),
IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(), IncomingFrame.begin() == nullptr ? "" : IncomingFrame.begin(),
State_.sessionId)); State_.sessionId));
KillConnection=true; return EndConnection();
} catch (...) { } catch (...) {
poco_error(Logger_, fmt::format("UnknownException({}): Device must be disconnected. " poco_error(Logger_, fmt::format("UnknownException({}): Device must be disconnected. "
"Unknown exception. Session:{}", "Unknown exception. Session:{}",
CId_, State_.sessionId)); CId_, State_.sessionId));
KillConnection=true; return EndConnection();
} }
if (!KillConnection && Errors_ < 10) if (Errors_ < 10)
return; return;
poco_warning(Logger_, fmt::format("DISCONNECTING({}): ConnectionException: {} Errors: {}", CId_, KillConnection, Errors_ )); poco_warning(Logger_, fmt::format("DISCONNECTING({}): Too many errors", CId_));
EndConnection(); return EndConnection();
} }
bool AP_WS_Connection::Send(const std::string &Payload) { bool AP_WS_Connection::Send(const std::string &Payload) {
@@ -889,36 +972,4 @@ namespace OpenWifi {
} }
} }
} }
void AP_WS_Connection::SetLastStats(const std::string &LastStats) {
RawLastStats_ = LastStats;
try {
Poco::JSON::Parser P;
auto Stats = P.parse(LastStats).extract<Poco::JSON::Object::Ptr>();
State_.hasGPS = Stats->isObject("gps");
auto Unit = Stats->getObject("unit");
auto Memory = Unit->getObject("memory");
std::uint64_t TotalMemory = Memory->get("total");
std::uint64_t FreeMemory = Memory->get("free");
if (TotalMemory > 0) {
State_.memoryUsed =
(100.0 * ((double)TotalMemory - (double)FreeMemory)) / (double)TotalMemory;
}
if (Unit->isArray("load")) {
Poco::JSON::Array::Ptr Load = Unit->getArray("load");
if (Load->size() > 1) {
State_.load = Load->get(1);
}
}
if (Unit->isArray("temperature")) {
Poco::JSON::Array::Ptr Temperature = Unit->getArray("temperature");
if (Temperature->size() > 1) {
State_.temperature = Temperature->get(0);
}
}
} catch (const Poco::Exception &E) {
poco_error(Logger_, "Failed to parse last stats: " + E.displayText());
}
}
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -14,10 +14,8 @@
#include "Poco/Net/SocketReactor.h" #include "Poco/Net/SocketReactor.h"
#include "Poco/Net/StreamSocket.h" #include "Poco/Net/StreamSocket.h"
#include "Poco/Net/WebSocket.h" #include "Poco/Net/WebSocket.h"
#include <Poco/Data/Session.h>
#include "RESTObjects/RESTAPI_GWobjects.h" #include "RESTObjects/RESTAPI_GWobjects.h"
#include <AP_WS_Reactor_Pool.h>
namespace OpenWifi { namespace OpenWifi {
@@ -27,17 +25,16 @@ namespace OpenWifi {
public: public:
explicit AP_WS_Connection(Poco::Net::HTTPServerRequest &request, explicit AP_WS_Connection(Poco::Net::HTTPServerRequest &request,
Poco::Net::HTTPServerResponse &response, uint64_t connection_id, Poco::Net::HTTPServerResponse &response, uint64_t connection_id,
Poco::Logger &L, std::pair<std::shared_ptr<Poco::Net::SocketReactor>, std::shared_ptr<LockedDbSession>> R); Poco::Logger &L, Poco::Net::SocketReactor &R);
~AP_WS_Connection(); ~AP_WS_Connection();
void EndConnection(); void EndConnection(bool DeleteSession=true);
void ProcessJSONRPCEvent(Poco::JSON::Object::Ptr &Doc); void ProcessJSONRPCEvent(Poco::JSON::Object::Ptr &Doc);
void ProcessJSONRPCResult(Poco::JSON::Object::Ptr Doc); void ProcessJSONRPCResult(Poco::JSON::Object::Ptr Doc);
void ProcessIncomingFrame(); void ProcessIncomingFrame();
void ProcessIncomingRadiusData(const Poco::JSON::Object::Ptr &Doc); void ProcessIncomingRadiusData(const Poco::JSON::Object::Ptr &Doc);
[[nodiscard]] bool Send(const std::string &Payload); [[nodiscard]] bool Send(const std::string &Payload);
[[nodiscard]] inline bool MustBeSecureRTTY() const { return RTTYMustBeSecure_; }
bool SendRadiusAuthenticationData(const unsigned char *buffer, std::size_t size); bool SendRadiusAuthenticationData(const unsigned char *buffer, std::size_t size);
bool SendRadiusAccountingData(const unsigned char *buffer, std::size_t size); bool SendRadiusAccountingData(const unsigned char *buffer, std::size_t size);
@@ -46,7 +43,10 @@ namespace OpenWifi {
void OnSocketReadable(const Poco::AutoPtr<Poco::Net::ReadableNotification> &pNf); void OnSocketReadable(const Poco::AutoPtr<Poco::Net::ReadableNotification> &pNf);
void OnSocketShutdown(const Poco::AutoPtr<Poco::Net::ShutdownNotification> &pNf); void OnSocketShutdown(const Poco::AutoPtr<Poco::Net::ShutdownNotification> &pNf);
void OnSocketError(const Poco::AutoPtr<Poco::Net::ErrorNotification> &pNf); void OnSocketError(const Poco::AutoPtr<Poco::Net::ErrorNotification> &pNf);
bool LookForUpgrade(Poco::Data::Session &Session, uint64_t UUID, uint64_t &UpgradedUUID); bool LookForUpgrade(uint64_t UUID, uint64_t &UpgradedUUID);
static bool ExtractBase64CompressedData(const std::string &CompressedData,
std::string &UnCompressedData,
uint64_t compress_sz);
void LogException(const Poco::Exception &E); void LogException(const Poco::Exception &E);
inline Poco::Logger &Logger() { return Logger_; } inline Poco::Logger &Logger() { return Logger_; }
bool SetWebSocketTelemetryReporting(uint64_t RPCID, uint64_t interval, bool SetWebSocketTelemetryReporting(uint64_t RPCID, uint64_t interval,
@@ -59,33 +59,81 @@ namespace OpenWifi {
bool StopKafkaTelemetry(uint64_t RPCID); bool StopKafkaTelemetry(uint64_t RPCID);
inline void GetLastStats(std::string &LastStats) { inline void GetLastStats(std::string &LastStats) {
if(!Dead_) {
std::lock_guard G(ConnectionMutex_); std::lock_guard G(ConnectionMutex_);
LastStats = RawLastStats_; LastStats = RawLastStats_;
} }
inline void SetLastStats(const std::string &LastStats) {
std::lock_guard G(ConnectionMutex_);
RawLastStats_ = LastStats;
try {
Poco::JSON::Parser P;
auto Stats = P.parse(LastStats).extract<Poco::JSON::Object::Ptr>();
hasGPS = Stats->isObject("gps");
auto Unit = Stats->getObject("unit");
auto Memory = Unit->getObject("memory");
std::uint64_t TotalMemory = Memory->get("total");
std::uint64_t FreeMemory = Memory->get("free");
if(TotalMemory>0) {
memory_used_ =
(100.0 * ((double)TotalMemory - (double)FreeMemory)) / (double)TotalMemory;
}
if(Unit->isArray("load")) {
Poco::JSON::Array::Ptr Load = Unit->getArray("load");
if(Load->size()>1) {
cpu_load_ = Load->get(1);
}
}
if(Unit->isArray("temperature")) {
Poco::JSON::Array::Ptr Temperature = Unit->getArray("temperature");
if(Temperature->size()>1) {
temperature_ = Temperature->get(0);
}
}
} catch (...) {
}
}
inline void SetLastHealthCheck(const GWObjects::HealthCheck &H) {
std::lock_guard G(ConnectionMutex_);
RawLastHealthcheck_ = H;
} }
inline void GetLastHealthCheck(GWObjects::HealthCheck &H) { inline void GetLastHealthCheck(GWObjects::HealthCheck &H) {
if(!Dead_) {
std::lock_guard G(ConnectionMutex_); std::lock_guard G(ConnectionMutex_);
H = RawLastHealthcheck_; H = RawLastHealthcheck_;
} }
}
inline void GetState(GWObjects::ConnectionState &State) { inline void GetState(GWObjects::ConnectionState &State) const {
if(!Dead_) {
std::lock_guard G(ConnectionMutex_); std::lock_guard G(ConnectionMutex_);
State = State_; State = State_;
} }
}
inline GWObjects::DeviceRestrictions GetRestrictions() { inline bool HasGPS() { return hasGPS; }
inline void GetRestrictions(GWObjects::DeviceRestrictions &R) const {
std::lock_guard G(ConnectionMutex_); std::lock_guard G(ConnectionMutex_);
return Restrictions_; R = Restrictions_;
} }
[[nodiscard]] inline bool HasGPS() const { return hasGPS_; } void Process_connect(Poco::JSON::Object::Ptr ParamsObj, const std::string &Serial);
[[nodiscard]] bool ValidatedDevice(); void Process_state(Poco::JSON::Object::Ptr ParamsObj);
void Process_healthcheck(Poco::JSON::Object::Ptr ParamsObj);
void Process_log(Poco::JSON::Object::Ptr ParamsObj);
void Process_crashlog(Poco::JSON::Object::Ptr ParamsObj);
void Process_ping(Poco::JSON::Object::Ptr ParamsObj);
void Process_cfgpending(Poco::JSON::Object::Ptr ParamsObj);
void Process_recovery(Poco::JSON::Object::Ptr ParamsObj);
void Process_deviceupdate(Poco::JSON::Object::Ptr ParamsObj, std::string &Serial);
void Process_telemetry(Poco::JSON::Object::Ptr ParamsObj);
void Process_venuebroadcast(Poco::JSON::Object::Ptr ParamsObj);
void Process_event(Poco::JSON::Object::Ptr ParamsObj);
void Process_wifiscan(Poco::JSON::Object::Ptr ParamsObj);
void Process_alarm(Poco::JSON::Object::Ptr ParamsObj);
void Process_rebootLog(Poco::JSON::Object::Ptr ParamsObj);
bool ValidatedDevice();
inline bool GetTelemetryParameters(bool &Reporting, uint64_t &Interval, inline bool GetTelemetryParameters(bool &Reporting, uint64_t &Interval,
uint64_t &WebSocketTimer, uint64_t &KafkaTimer, uint64_t &WebSocketTimer, uint64_t &KafkaTimer,
@@ -105,14 +153,18 @@ namespace OpenWifi {
friend class AP_WS_Server; friend class AP_WS_Server;
void Start(); inline GWObjects::DeviceRestrictions Restrictions() const {
std::lock_guard G(ConnectionMutex_);
return Restrictions_;
}
inline bool MustBeSecureRtty() const { return RttyMustBeSecure_; }
private: private:
mutable std::recursive_mutex ConnectionMutex_; mutable std::mutex ConnectionMutex_;
std::mutex TelemetryMutex_; std::mutex TelemetryMutex_;
Poco::Logger &Logger_; Poco::Logger &Logger_;
std::shared_ptr<Poco::Net::SocketReactor> Reactor_; Poco::Net::SocketReactor &Reactor_;
std::shared_ptr<LockedDbSession> DbSession_;
std::unique_ptr<Poco::Net::WebSocket> WS_; std::unique_ptr<Poco::Net::WebSocket> WS_;
std::string SerialNumber_; std::string SerialNumber_;
uint64_t SerialNumberInt_ = 0; uint64_t SerialNumberInt_ = 0;
@@ -123,56 +175,34 @@ namespace OpenWifi {
uint64_t Errors_ = 0; uint64_t Errors_ = 0;
Poco::Net::IPAddress PeerAddress_; Poco::Net::IPAddress PeerAddress_;
volatile bool TelemetryReporting_ = false; volatile bool TelemetryReporting_ = false;
std::atomic_uint64_t TelemetryWebSocketRefCount_ = 0; volatile uint64_t TelemetryWebSocketRefCount_ = 0;
std::atomic_uint64_t TelemetryKafkaRefCount_ = 0; volatile uint64_t TelemetryKafkaRefCount_ = 0;
std::atomic_uint64_t TelemetryWebSocketTimer_ = 0; volatile uint64_t TelemetryWebSocketTimer_ = 0;
std::atomic_uint64_t TelemetryKafkaTimer_ = 0; volatile uint64_t TelemetryKafkaTimer_ = 0;
std::atomic_uint64_t TelemetryInterval_ = 0; volatile uint64_t TelemetryInterval_ = 0;
std::atomic_uint64_t TelemetryWebSocketPackets_ = 0; volatile uint64_t TelemetryWebSocketPackets_ = 0;
std::atomic_uint64_t TelemetryKafkaPackets_ = 0; volatile uint64_t TelemetryKafkaPackets_ = 0;
GWObjects::ConnectionState State_; GWObjects::ConnectionState State_;
Utils::CompressedString RawLastStats_; std::string RawLastStats_;
GWObjects::HealthCheck RawLastHealthcheck_; GWObjects::HealthCheck RawLastHealthcheck_;
std::chrono::time_point<std::chrono::high_resolution_clock> ConnectionStart_ = std::chrono::time_point<std::chrono::high_resolution_clock> ConnectionStart_ =
std::chrono::high_resolution_clock::now(); std::chrono::high_resolution_clock::now();
std::chrono::duration<double, std::milli> ConnectionCompletionTime_{0.0}; std::chrono::duration<double, std::milli> ConnectionCompletionTime_{0.0};
std::atomic<bool> Dead_ = false; std::atomic_flag Dead_ = false;
std::atomic_bool DeviceValidated_ = false; std::atomic_bool DeviceValidated_ = false;
std::atomic_bool Valid_ = false;
OpenWifi::GWObjects::DeviceRestrictions Restrictions_; OpenWifi::GWObjects::DeviceRestrictions Restrictions_;
bool RTTYMustBeSecure_ = false; bool RttyMustBeSecure_ = false;
bool hasGPS_=false;
std::double_t memory_used_=0.0, cpu_load_ = 0.0, temperature_ = 0.0;
std::uint64_t uuid_=0;
bool Simulated_=false;
std::atomic_uint64_t LastContact_=0;
static inline std::atomic_uint64_t ConcurrentStartingDevices_ = 0; static inline std::atomic_uint64_t ConcurrentStartingDevices_ = 0;
bool StartTelemetry(uint64_t RPCID, const std::vector<std::string> &TelemetryTypes); bool StartTelemetry(uint64_t RPCID, const std::vector<std::string> &TelemetryTypes);
bool StopTelemetry(uint64_t RPCID); bool StopTelemetry(uint64_t RPCID);
void UpdateCounts(); void UpdateCounts();
static void DeviceDisconnectionCleanup(const std::string &SerialNumber, std::uint64_t uuid); bool hasGPS=false;
void SetLastStats(const std::string &LastStats); std::double_t memory_used_=0.0, cpu_load_ = 0.0, temperature_ = 0.0;
void Process_connect(Poco::JSON::Object::Ptr ParamsObj, const std::string &Serial); std::uint64_t uuid_=0;
void Process_state(Poco::JSON::Object::Ptr ParamsObj); bool Simulated_=false;
void Process_healthcheck(Poco::JSON::Object::Ptr ParamsObj);
void Process_log(Poco::JSON::Object::Ptr ParamsObj);
void Process_crashlog(Poco::JSON::Object::Ptr ParamsObj);
void Process_ping(Poco::JSON::Object::Ptr ParamsObj);
void Process_cfgpending(Poco::JSON::Object::Ptr ParamsObj);
void Process_recovery(Poco::JSON::Object::Ptr ParamsObj);
void Process_deviceupdate(Poco::JSON::Object::Ptr ParamsObj, std::string &Serial);
void Process_telemetry(Poco::JSON::Object::Ptr ParamsObj);
void Process_venuebroadcast(Poco::JSON::Object::Ptr ParamsObj);
void Process_event(Poco::JSON::Object::Ptr ParamsObj);
void Process_wifiscan(Poco::JSON::Object::Ptr ParamsObj);
void Process_alarm(Poco::JSON::Object::Ptr ParamsObj);
void Process_rebootLog(Poco::JSON::Object::Ptr ParamsObj);
inline void SetLastHealthCheck(const GWObjects::HealthCheck &H) {
RawLastHealthcheck_ = H;
}
}; };
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -1,111 +0,0 @@
#include <AP_WS_Connection.h>
#include "ConfigurationCache.h"
#include "UI_GW_WebSocketNotifications.h"
#include "CommandManager.h"
namespace OpenWifi {
bool AP_WS_Connection::LookForUpgrade(Poco::Data::Session &Session, const uint64_t UUID, uint64_t &UpgradedUUID) {
// A UUID of zero means ignore updates for that connection.
if (UUID == 0)
return false;
uint64_t GoodConfig = GetCurrentConfigurationID(SerialNumberInt_);
if (GoodConfig && (GoodConfig == UUID || GoodConfig == State_.PendingUUID)) {
UpgradedUUID = UUID;
State_.PendingUUID = 0;
return false;
}
GWObjects::Device D;
if (!StorageService()->GetDevice(Session,SerialNumber_, D)) {
return false;
}
if(State_.PendingUUID!=0 && UUID==State_.PendingUUID) {
// so we sent an upgrade to a device, and now it is completing now...
UpgradedUUID = UUID;
StorageService()->CompleteDeviceConfigurationChange(Session, SerialNumber_);
State_.PendingUUID = 0;
return true;
}
// dont upgrade a switch if it does not have a real config. Config will always be more than 20 characters
if (D.DeviceType==Platforms::SWITCH && D.Configuration.size()<20) {
return false;
}
Config::Config Cfg(D.Configuration);
// if this is a broken device (UUID==0) just fix it
auto StoredConfigurationUUID = Cfg.UUID();
if(D.UUID==0) {
D.UUID = StoredConfigurationUUID;
}
if (D.UUID == UUID) {
D.UUID = UpgradedUUID = UUID;
State_.PendingUUID = D.pendingUUID = 0;
D.pendingConfiguration.clear();
D.pendingConfigurationCmd.clear();
StorageService()->UpdateDevice(Session, D);
SetCurrentConfigurationID(SerialNumberInt_, UUID);
// std::cout << __LINE__ << ": " << SerialNumber_ << " GoodConfig: " << GoodConfig << " UUID:" << UUID << " Pending:" << State_.PendingUUID << std::endl;
return false;
}
if (UUID > D.UUID) {
// so we have a problem, the device has a newer config than we have. So we need to
// make sure our config is newer.
D.UUID = UUID + 2;
UpgradedUUID = D.UUID;
// std::cout << __LINE__ << ": " << SerialNumber_ << " GoodConfig: " << GoodConfig << " UUID:" << UUID << " Pending:" << State_.PendingUUID << std::endl;
}
Cfg.SetUUID(D.UUID);
D.Configuration = Cfg.get();
D.pendingUUID = State_.PendingUUID = UpgradedUUID = D.UUID;
StorageService()->UpdateDevice(Session, D);
GWObjects::CommandDetails Cmd;
Cmd.SerialNumber = SerialNumber_;
Cmd.UUID = MicroServiceCreateUUID();
Cmd.SubmittedBy = uCentralProtocol::SUBMITTED_BY_SYSTEM;
Cmd.Status = uCentralProtocol::PENDING;
Cmd.Command = uCentralProtocol::CONFIGURE;
Poco::JSON::Parser P;
auto ParsedConfig = P.parse(D.Configuration).extract<Poco::JSON::Object::Ptr>();
Poco::JSON::Object Params;
Params.set(uCentralProtocol::SERIAL, SerialNumber_);
Params.set(uCentralProtocol::UUID, D.UUID);
Params.set(uCentralProtocol::WHEN, 0);
Params.set(uCentralProtocol::CONFIG, ParsedConfig);
std::ostringstream O;
Poco::JSON::Stringifier::stringify(Params, O);
Cmd.Details = O.str();
poco_information(Logger_,
fmt::format("CFG-UPGRADE({}): Current ID: {}, newer configuration {}.",
CId_, UUID, D.UUID));
bool Sent;
StorageService()->AddCommand(SerialNumber_, Cmd,
Storage::CommandExecutionType::COMMAND_EXECUTED);
CommandManager()->PostCommand(
CommandManager()->Next_RPC_ID(), APCommands::to_apcommand(Cmd.Command.c_str()),
SerialNumber_, Cmd.Command, Params, Cmd.UUID, Sent, false, false);
GWWebSocketNotifications::SingleDeviceConfigurationChange_t Notification;
Notification.content.serialNumber = D.SerialNumber;
Notification.content.oldUUID = UUID;
Notification.content.newUUID = UpgradedUUID;
GWWebSocketNotifications::DeviceConfigurationChange(Notification);
// std::cout << __LINE__ << ": " << SerialNumber_ << " GoodConfig: " << GoodConfig << " UUID:" << UUID <<
// " Pending:" << State_.PendingUUID << " Upgraded:" << UpgradedUUID << std::endl;
return true;
}
}

View File

@@ -71,8 +71,9 @@ namespace OpenWifi {
CommandManager()->ClearQueue(SerialNumberInt_); CommandManager()->ClearQueue(SerialNumberInt_);
AP_WS_Server()->StartSession(State_.sessionId, SerialNumberInt_); AP_WS_Server()->SetSessionDetails(State_.sessionId, SerialNumberInt_);
std::lock_guard Lock(ConnectionMutex_);
Config::Capabilities Caps(Capabilities); Config::Capabilities Caps(Capabilities);
Compatible_ = Caps.Compatible(); Compatible_ = Caps.Compatible();
@@ -83,8 +84,6 @@ namespace OpenWifi {
State_.Address = Utils::FormatIPv6(WS_->peerAddress().toString()); State_.Address = Utils::FormatIPv6(WS_->peerAddress().toString());
CId_ = SerialNumber_ + "@" + CId_; CId_ = SerialNumber_ + "@" + CId_;
auto Platform = Poco::toLower(Caps.Platform());
if(ParamsObj->has("reason")) { if(ParamsObj->has("reason")) {
State_.connectReason = ParamsObj->get("reason").toString(); State_.connectReason = ParamsObj->get("reason").toString();
} }
@@ -101,24 +100,36 @@ namespace OpenWifi {
Restrictions_.from_json(RestrictionObject); Restrictions_.from_json(RestrictionObject);
} }
if (Capabilities->has("developer") && !Capabilities->isNull("developer")) { if (Capabilities->has("developer")) {
Restrictions_.developer = Capabilities->getValue<bool>("developer"); Restrictions_.developer = Capabilities->getValue<bool>("developer");
} }
if(Capabilities->has("secure-rtty")) { if(Capabilities->has("secure-rtty")) {
RTTYMustBeSecure_ = Capabilities->getValue<bool>("secure-rtty"); RttyMustBeSecure_ = Capabilities->getValue<bool>("secure-rtty");
} }
State_.locale = FindCountryFromIP()->Get(IP); State_.locale = FindCountryFromIP()->Get(IP);
GWObjects::Device DeviceInfo; GWObjects::Device DeviceInfo;
std::lock_guard DbSessionLock(DbSession_->Mutex()); auto DeviceExists = StorageService()->GetDevice(SerialNumber_, DeviceInfo);
auto DeviceExists = StorageService()->GetDevice(DbSession_->Session(), SerialNumber_, DeviceInfo);
if (Daemon()->AutoProvisioning() && !DeviceExists) { if (Daemon()->AutoProvisioning() && !DeviceExists) {
// check the firmware version. if this is too old, we cannot let that device connect yet, we must // check the firmware version. if this is too old, we cannot let that device connect yet, we must
// force a firmware upgrade // force a firmware upgrade
GWObjects::DefaultFirmware MinimumFirmware; GWObjects::DefaultFirmware MinimumFirmware;
if(FirmwareRevisionCache()->DeviceMustUpgrade(Compatible_, Firmware, MinimumFirmware)) { if(FirmwareRevisionCache()->DeviceMustUpgrade(Compatible_, Firmware, MinimumFirmware)) {
/*
{ "jsonrpc" : "2.0" ,
"method" : "upgrade" ,
"params" : {
"serial" : <serial number> ,
"when" : Optional - <UTC time when to upgrade the firmware, 0 mean immediate, this is a suggestion>,
"uri" : <URI to download the firmware>,
"FWsignature" : <string representation of the signature for the FW> (optional)
},
"id" : <some number>
}
*/
Poco::JSON::Object UpgradeCommand, Params; Poco::JSON::Object UpgradeCommand, Params;
UpgradeCommand.set(uCentralProtocol::JSONRPC,uCentralProtocol::JSONRPC_VERSION); UpgradeCommand.set(uCentralProtocol::JSONRPC,uCentralProtocol::JSONRPC_VERSION);
UpgradeCommand.set(uCentralProtocol::METHOD,uCentralProtocol::UPGRADE); UpgradeCommand.set(uCentralProtocol::METHOD,uCentralProtocol::UPGRADE);
@@ -146,7 +157,7 @@ namespace OpenWifi {
} }
return; return;
} else { } else {
StorageService()->CreateDefaultDevice( DbSession_->Session(), StorageService()->CreateDefaultDevice(
SerialNumber_, Caps, Firmware, PeerAddress_, SerialNumber_, Caps, Firmware, PeerAddress_,
State_.VerifiedCertificate == GWObjects::SIMULATED); State_.VerifiedCertificate == GWObjects::SIMULATED);
} }
@@ -155,7 +166,7 @@ namespace OpenWifi {
poco_warning(Logger(),fmt::format("Device {} is a {} from {} and cannot be provisioned.",SerialNumber_,Compatible_, CId_)); poco_warning(Logger(),fmt::format("Device {} is a {} from {} and cannot be provisioned.",SerialNumber_,Compatible_, CId_));
return EndConnection(); return EndConnection();
} else if (DeviceExists) { } else if (DeviceExists) {
StorageService()->UpdateDeviceCapabilities(DbSession_->Session(), SerialNumber_, Caps); StorageService()->UpdateDeviceCapabilities(SerialNumber_, Caps);
int Updated{0}; int Updated{0};
if (!Firmware.empty()) { if (!Firmware.empty()) {
if (Firmware != DeviceInfo.Firmware) { if (Firmware != DeviceInfo.Firmware) {
@@ -206,13 +217,8 @@ namespace OpenWifi {
++Updated; ++Updated;
} }
if (Compatible_ != DeviceInfo.Compatible) { if (Compatible_ != DeviceInfo.DeviceType) {
DeviceInfo.Compatible = Compatible_; DeviceInfo.DeviceType = Compatible_;
++Updated;
}
if (Platform != DeviceInfo.DeviceType) {
DeviceInfo.DeviceType = Platform;
++Updated; ++Updated;
} }
@@ -232,13 +238,12 @@ namespace OpenWifi {
} }
if (Updated) { if (Updated) {
StorageService()->UpdateDevice(DbSession_->Session(), DeviceInfo); StorageService()->UpdateDevice(DeviceInfo);
}
} }
if(!Simulated_) { if(!Simulated_) {
uint64_t UpgradedUUID = 0; uint64_t UpgradedUUID = 0;
if (LookForUpgrade(DbSession_->Session(), UUID, UpgradedUUID)) { LookForUpgrade(UUID, UpgradedUUID);
State_.UUID = UpgradedUUID; State_.UUID = UpgradedUUID;
} }
} }

View File

@@ -29,7 +29,7 @@ namespace OpenWifi {
.Recorded = Utils::Now(), .Recorded = Utils::Now(),
.LogType = 1, .LogType = 1,
.UUID = ParamsObj->get(uCentralProtocol::UUID)}; .UUID = ParamsObj->get(uCentralProtocol::UUID)};
StorageService()->AddLog(*DbSession_, DeviceLog); StorageService()->AddLog(DeviceLog);
DeviceLogKafkaEvent E(DeviceLog); DeviceLogKafkaEvent E(DeviceLog);
} else { } else {
poco_warning(Logger_, fmt::format("LOG({}): Missing parameters.", CId_)); poco_warning(Logger_, fmt::format("LOG({}): Missing parameters.", CId_));

View File

@@ -21,7 +21,7 @@ namespace OpenWifi {
if (ParamsObj->has("currentPassword")) { if (ParamsObj->has("currentPassword")) {
auto Password = ParamsObj->get("currentPassword").toString(); auto Password = ParamsObj->get("currentPassword").toString();
StorageService()->SetDevicePassword(*DbSession_,Serial, Password); StorageService()->SetDevicePassword(Serial, Password);
poco_trace( poco_trace(
Logger_, Logger_,
fmt::format("DEVICE-UPDATE({}): Device is updating its login password.", Serial)); fmt::format("DEVICE-UPDATE({}): Device is updating its login password.", Serial));

View File

@@ -3,7 +3,6 @@
// //
#include "AP_WS_Connection.h" #include "AP_WS_Connection.h"
#include "AP_WS_Server.h"
#include "StorageService.h" #include "StorageService.h"
#include "fmt/format.h" #include "fmt/format.h"
@@ -26,7 +25,6 @@ namespace OpenWifi {
uint64_t UUID = ParamsObj->get(uCentralProtocol::UUID); uint64_t UUID = ParamsObj->get(uCentralProtocol::UUID);
auto Sanity = ParamsObj->get(uCentralProtocol::SANITY); auto Sanity = ParamsObj->get(uCentralProtocol::SANITY);
State_.sanity = Sanity;
auto CheckData = ParamsObj->get(uCentralProtocol::DATA).toString(); auto CheckData = ParamsObj->get(uCentralProtocol::DATA).toString();
if (CheckData.empty()) if (CheckData.empty())
CheckData = uCentralProtocol::EMPTY_JSON_DOC; CheckData = uCentralProtocol::EMPTY_JSON_DOC;
@@ -50,14 +48,14 @@ namespace OpenWifi {
Check.Data = CheckData; Check.Data = CheckData;
Check.Sanity = Sanity; Check.Sanity = Sanity;
StorageService()->AddHealthCheckData(*DbSession_, Check); StorageService()->AddHealthCheckData(Check);
if (!request_uuid.empty()) { if (!request_uuid.empty()) {
StorageService()->SetCommandResult(request_uuid, CheckData); StorageService()->SetCommandResult(request_uuid, CheckData);
} }
SetLastHealthCheck(Check); SetLastHealthCheck(Check);
if (KafkaManager()->Enabled() && !AP_WS_Server()->KafkaDisableHealthChecks()) { if (KafkaManager()->Enabled()) {
KafkaManager()->PostMessage(KafkaTopics::HEALTHCHECK, SerialNumber_, *ParamsObj); KafkaManager()->PostMessage(KafkaTopics::HEALTHCHECK, SerialNumber_, *ParamsObj);
} }
} else { } else {

View File

@@ -36,7 +36,7 @@ namespace OpenWifi {
.Recorded = (uint64_t)time(nullptr), .Recorded = (uint64_t)time(nullptr),
.LogType = 0, .LogType = 0,
.UUID = State_.UUID}; .UUID = State_.UUID};
StorageService()->AddLog(*DbSession_, DeviceLog); StorageService()->AddLog(DeviceLog);
DeviceLogKafkaEvent E(DeviceLog); DeviceLogKafkaEvent E(DeviceLog);
} else { } else {
poco_warning(Logger_, fmt::format("LOG({}): Missing parameters.", CId_)); poco_warning(Logger_, fmt::format("LOG({}): Missing parameters.", CId_));

View File

@@ -35,7 +35,7 @@ namespace OpenWifi {
.Recorded = ParamsObj->get(uCentralProtocol::DATE), .Recorded = ParamsObj->get(uCentralProtocol::DATE),
.LogType = 2, .LogType = 2,
.UUID = ParamsObj->get(uCentralProtocol::UUID)}; .UUID = ParamsObj->get(uCentralProtocol::UUID)};
StorageService()->AddLog(*DbSession_, DeviceLog); StorageService()->AddLog(DeviceLog);
DeviceLogKafkaEvent E(DeviceLog); DeviceLogKafkaEvent E(DeviceLog);
} else { } else {
poco_warning(Logger_, fmt::format("REBOOT-LOG({}): Missing parameters.", CId_)); poco_warning(Logger_, fmt::format("REBOOT-LOG({}): Missing parameters.", CId_));

View File

@@ -35,7 +35,7 @@ namespace OpenWifi {
.LogType = 1, .LogType = 1,
.UUID = 0}; .UUID = 0};
StorageService()->AddLog(*DbSession_, DeviceLog); StorageService()->AddLog(DeviceLog);
if (ParamsObj->get(uCentralProtocol::REBOOT).toString() == "true") { if (ParamsObj->get(uCentralProtocol::REBOOT).toString() == "true") {
GWObjects::CommandDetails Cmd; GWObjects::CommandDetails Cmd;

View File

@@ -3,7 +3,6 @@
// //
#include "AP_WS_Connection.h" #include "AP_WS_Connection.h"
#include "AP_WS_Server.h"
#include "StateUtils.h" #include "StateUtils.h"
#include "StorageService.h" #include "StorageService.h"
@@ -40,27 +39,25 @@ namespace OpenWifi {
UUID, request_uuid)); UUID, request_uuid));
} }
std::lock_guard Guard(DbSession_->Mutex());
if(!Simulated_) { if(!Simulated_) {
uint64_t UpgradedUUID; uint64_t UpgradedUUID;
LookForUpgrade(DbSession_->Session(), UUID, UpgradedUUID); LookForUpgrade(UUID, UpgradedUUID);
State_.UUID = UpgradedUUID; State_.UUID = UpgradedUUID;
} }
SetLastStats(StateStr); SetLastStats(StateStr);
GWObjects::Statistics Stats{ GWObjects::Statistics Stats{
.SerialNumber = SerialNumber_, .UUID = UUID, .Data = StateStr}; .SerialNumber = SerialNumber_, .UUID = UUID, .Data = StateStr};
Stats.Recorded = Utils::Now(); Stats.Recorded = Utils::Now();
StorageService()->AddStatisticsData(DbSession_->Session(),Stats); StorageService()->AddStatisticsData(Stats);
if (!request_uuid.empty()) { if (!request_uuid.empty()) {
StorageService()->SetCommandResult(request_uuid, StateStr); StorageService()->SetCommandResult(request_uuid, StateStr);
} }
StateUtils::ComputeAssociations(StateObj, State_.Associations_2G, StateUtils::ComputeAssociations(StateObj, State_.Associations_2G,
State_.Associations_5G, State_.Associations_6G, State_.uptime); State_.Associations_5G, State_.Associations_6G);
if (KafkaManager()->Enabled() && !AP_WS_Server()->KafkaDisableState()) { if (KafkaManager()->Enabled()) {
KafkaManager()->PostMessage(KafkaTopics::STATE, SerialNumber_, *ParamsObj); KafkaManager()->PostMessage(KafkaTopics::STATE, SerialNumber_, *ParamsObj);
} }

View File

@@ -35,7 +35,8 @@ namespace OpenWifi {
} }
if (TelemetryWebSocketRefCount_) { if (TelemetryWebSocketRefCount_) {
if (now < TelemetryWebSocketTimer_) { if (now < TelemetryWebSocketTimer_) {
// std::cout << SerialNumber_ << ": Updating WebSocket telemetry" <<
// std::endl;
TelemetryWebSocketPackets_++; TelemetryWebSocketPackets_++;
State_.websocketPackets = TelemetryWebSocketPackets_; State_.websocketPackets = TelemetryWebSocketPackets_;
TelemetryStream()->NotifyEndPoint(SerialNumberInt_, KafkaPayload); TelemetryStream()->NotifyEndPoint(SerialNumberInt_, KafkaPayload);
@@ -45,6 +46,7 @@ namespace OpenWifi {
} }
if (TelemetryKafkaRefCount_) { if (TelemetryKafkaRefCount_) {
if (KafkaManager()->Enabled() && now < TelemetryKafkaTimer_) { if (KafkaManager()->Enabled() && now < TelemetryKafkaTimer_) {
// std::cout << SerialNumber_ << ": Updating Kafka telemetry" << std::endl;
TelemetryKafkaPackets_++; TelemetryKafkaPackets_++;
State_.kafkaPackets = TelemetryKafkaPackets_; State_.kafkaPackets = TelemetryKafkaPackets_;
KafkaManager()->PostMessage(KafkaTopics::DEVICE_TELEMETRY, SerialNumber_, KafkaManager()->PostMessage(KafkaTopics::DEVICE_TELEMETRY, SerialNumber_,

62
src/AP_WS_ReactorPool.h Normal file
View File

@@ -0,0 +1,62 @@
//
// Created by stephane bourque on 2022-02-03.
//
#pragma once
#include <mutex>
#include <string>
#include "Poco/Environment.h"
#include "Poco/Net/SocketAcceptor.h"
#include "framework/utils.h"
namespace OpenWifi {
class AP_WS_ReactorThreadPool {
public:
explicit AP_WS_ReactorThreadPool() {
NumberOfThreads_ = Poco::Environment::processorCount() * 4;
if (NumberOfThreads_ == 0)
NumberOfThreads_ = 4;
}
~AP_WS_ReactorThreadPool() { Stop(); }
void Start() {
for (uint64_t i = 0; i < NumberOfThreads_; ++i) {
auto NewReactor = std::make_unique<Poco::Net::SocketReactor>();
auto NewThread = std::make_unique<Poco::Thread>();
NewThread->start(*NewReactor);
std::string ThreadName{"ap:react:" + std::to_string(i)};
Utils::SetThreadName(*NewThread, ThreadName.c_str());
Reactors_.emplace_back(std::move(NewReactor));
Threads_.emplace_back(std::move(NewThread));
}
}
void Stop() {
for (auto &i : Reactors_)
i->stop();
for (auto &i : Threads_) {
i->join();
}
Reactors_.clear();
Threads_.clear();
}
Poco::Net::SocketReactor &NextReactor() {
std::lock_guard Lock(Mutex_);
NextReactor_++;
NextReactor_ %= NumberOfThreads_;
return *Reactors_[NextReactor_];
}
private:
std::mutex Mutex_;
uint64_t NumberOfThreads_;
uint64_t NextReactor_ = 0;
std::vector<std::unique_ptr<Poco::Net::SocketReactor>> Reactors_;
std::vector<std::unique_ptr<Poco::Thread>> Threads_;
};
} // namespace OpenWifi

View File

@@ -1,77 +0,0 @@
//
// Created by stephane bourque on 2022-02-03.
//
#pragma once
#include <mutex>
#include <string>
#include <framework/utils.h>
#include <Poco/Environment.h>
#include <Poco/Net/SocketAcceptor.h>
#include <Poco/Data/SessionPool.h>
#include <StorageService.h>
namespace OpenWifi {
class AP_WS_ReactorThreadPool {
public:
explicit AP_WS_ReactorThreadPool(Poco::Logger &Logger) : Logger_(Logger) {
NumberOfThreads_ = Poco::Environment::processorCount() * 4;
if (NumberOfThreads_ == 0)
NumberOfThreads_ = 8;
NumberOfThreads_ = std::min(NumberOfThreads_, (std::uint64_t) 128);
}
~AP_WS_ReactorThreadPool() { Stop(); }
void Start() {
Reactors_.reserve(NumberOfThreads_);
DbSessions_.reserve(NumberOfThreads_);
Threads_.reserve(NumberOfThreads_);
Logger_.information(fmt::format("WebSocket Processor: starting {} threads.", NumberOfThreads_));
for (uint64_t i = 0; i < NumberOfThreads_; ++i) {
auto NewReactor = std::make_shared<Poco::Net::SocketReactor>();
auto NewThread = std::make_unique<Poco::Thread>();
NewThread->start(*NewReactor);
std::string ThreadName{"ap:react:" + std::to_string(i)};
Utils::SetThreadName(*NewThread, ThreadName.c_str());
Reactors_.emplace_back(std::move(NewReactor));
Threads_.emplace_back(std::move(NewThread));
DbSessions_.emplace_back(std::make_shared<LockedDbSession>());
}
Logger_.information(fmt::format("WebSocket Processor: {} threads started.", NumberOfThreads_));
}
void Stop() {
for (auto &i : Reactors_)
i->stop();
for (auto &i : Threads_) {
i->join();
}
Reactors_.clear();
Threads_.clear();
DbSessions_.clear();
}
auto NextReactor() {
std::lock_guard Lock(Mutex_);
NextReactor_++;
NextReactor_ %= NumberOfThreads_;
return std::make_pair(Reactors_[NextReactor_], DbSessions_[NextReactor_]);
}
private:
std::mutex Mutex_;
uint64_t NumberOfThreads_;
uint64_t NextReactor_ = 0;
std::vector<std::shared_ptr<Poco::Net::SocketReactor>> Reactors_;
std::vector<std::unique_ptr<Poco::Thread>> Threads_;
std::vector<std::shared_ptr<LockedDbSession>> DbSessions_;
Poco::Logger &Logger_;
};
} // namespace OpenWifi

View File

@@ -6,83 +6,45 @@
// Arilia Wireless Inc. // Arilia Wireless Inc.
// //
#include <Poco/Net/Context.h> #include "Poco/Net/Context.h"
#include <Poco/Net/HTTPHeaderStream.h> #include "Poco/Net/HTTPHeaderStream.h"
#include <Poco/Net/HTTPServerRequest.h> #include "Poco/Net/HTTPServerRequest.h"
#include <AP_WS_Connection.h> #include "AP_WS_Connection.h"
#include <AP_WS_Server.h> #include "AP_WS_Server.h"
#include <ConfigurationCache.h> #include "ConfigurationCache.h"
#include <TelemetryStream.h> #include "TelemetryStream.h"
#include <fmt/format.h> #include "UI_GW_WebSocketNotifications.h"
#include "fmt/format.h"
#include <framework/MicroServiceFuncs.h> #include "framework/MicroServiceFuncs.h"
#include <framework/utils.h> #include "framework/utils.h"
#include <framework/KafkaManager.h> #include <framework/KafkaManager.h>
#include <UI_GW_WebSocketNotifications.h>
namespace OpenWifi { namespace OpenWifi {
class AP_WS_RequestHandler : public Poco::Net::HTTPRequestHandler { void AP_WS_RequestHandler::handleRequest(Poco::Net::HTTPServerRequest &request,
public: Poco::Net::HTTPServerResponse &response) {
explicit AP_WS_RequestHandler(Poco::Logger &L, std::uint64_t session_id) : Logger_(L),
session_id_(session_id) {
};
void handleRequest( Poco::Net::HTTPServerRequest &request,
Poco::Net::HTTPServerResponse &response) override {
try { try {
auto NewConnection = std::make_shared<AP_WS_Connection>(request, response, session_id_, Logger_, AP_WS_Server()->AddConnection(
AP_WS_Server()->NextReactor()); id_, std::make_shared<AP_WS_Connection>(request, response, id_, Logger_,
AP_WS_Server()->AddConnection(NewConnection); AP_WS_Server()->NextReactor()));
NewConnection->Start();
} catch (...) { } catch (...) {
poco_warning(Logger_, "Exception during WS creation"); poco_warning(Logger_, "Exception during WS creation");
} }
}; };
private:
Poco::Logger &Logger_;
std::uint64_t session_id_;
};
class AP_WS_RequestHandlerFactory : public Poco::Net::HTTPRequestHandlerFactory {
public:
inline explicit AP_WS_RequestHandlerFactory(Poco::Logger &L) : Logger_(L) {}
inline Poco::Net::HTTPRequestHandler *
createRequestHandler(const Poco::Net::HTTPServerRequest &request) override {
if (request.find("Upgrade") != request.end() &&
Poco::icompare(request["Upgrade"], "websocket") == 0) {
Utils::SetThreadName("ws:conn-init");
session_id_++;
return new AP_WS_RequestHandler(Logger_, session_id_);
} else {
return nullptr;
}
}
private:
Poco::Logger &Logger_;
inline static std::atomic_uint64_t session_id_ = 0;
};
bool AP_WS_Server::ValidateCertificate(const std::string &ConnectionId, bool AP_WS_Server::ValidateCertificate(const std::string &ConnectionId,
const Poco::Crypto::X509Certificate &Certificate) { const Poco::Crypto::X509Certificate &Certificate) {
if (IsCertOk()) { if (IsCertOk()) {
// validate certificate agains trusted chain if (!Certificate.issuedBy(*IssuerCert_)) {
for (const auto &cert : ClientCasCerts_) {
if (Certificate.issuedBy(cert)) {
return true;
}
}
poco_warning( poco_warning(
Logger(), Logger(),
fmt::format( fmt::format("CERTIFICATE({}): issuer mismatch. Local='{}' Incoming='{}'",
"CERTIFICATE({}): issuer mismatch. Certificate not issued by any trusted CA", ConnectionId, IssuerCert_->issuerName(), Certificate.issuerName()));
ConnectionId) return false;
); }
return true;
} }
return false; return false;
} }
@@ -95,7 +57,7 @@ namespace OpenWifi {
SessionTimeOut_ = MicroServiceConfigGetInt("openwifi.session.timeout", 10*60); SessionTimeOut_ = MicroServiceConfigGetInt("openwifi.session.timeout", 10*60);
Reactor_pool_ = std::make_unique<AP_WS_ReactorThreadPool>(Logger()); Reactor_pool_ = std::make_unique<AP_WS_ReactorThreadPool>();
Reactor_pool_->Start(); Reactor_pool_->Start();
for (const auto &Svr : ConfigServersList_) { for (const auto &Svr : ConfigServersList_) {
@@ -137,13 +99,6 @@ namespace OpenWifi {
Context->addChainCertificate(Issuing); Context->addChainCertificate(Issuing);
Context->addCertificateAuthority(Issuing); Context->addCertificateAuthority(Issuing);
// add certificates from clientcas to trust chain
ClientCasCerts_ = Poco::Net::X509Certificate::readPEM(Svr.ClientCas());
for (const auto &cert : ClientCasCerts_) {
Context->addChainCertificate(cert);
Context->addCertificateAuthority(cert);
}
Poco::Crypto::RSAKey Key("", Svr.KeyFile(), Svr.KeyFilePassword()); Poco::Crypto::RSAKey Key("", Svr.KeyFile(), Svr.KeyFilePassword());
Context->usePrivateKey(Key); Context->usePrivateKey(Key);
@@ -180,9 +135,6 @@ namespace OpenWifi {
WebServerHttpParams); WebServerHttpParams);
WebServers_.push_back(std::move(NewWebServer)); WebServers_.push_back(std::move(NewWebServer));
} }
KafkaDisableState_ = MicroServiceConfigGetBool("openwifi.kafka.disablestate", false);
KafkaDisableHealthChecks_ = MicroServiceConfigGetBool("openwifi.kafka.disablehealthchecks", false);
} }
for (auto &server : WebServers_) { for (auto &server : WebServers_) {
@@ -204,205 +156,95 @@ namespace OpenWifi {
UseDefaultConfig_ = true; UseDefaultConfig_ = true;
} }
SimulatorId_ = Poco::toLower(MicroServiceConfigGetString("simulatorid", "")); SimulatorId_ = MicroServiceConfigGetString("simulatorid", "");
SimulatorEnabled_ = !SimulatorId_.empty(); SimulatorEnabled_ = !SimulatorId_.empty();
Utils::SetThreadName(ReactorThread_, "dev:react:head"); Utils::SetThreadName(ReactorThread_, "dev:react:head");
GarbageCollectorCallback_ = std::make_unique<Poco::TimerCallback<AP_WS_Server>>(
*this, &AP_WS_Server::onGarbageCollecting);
Timer_.setStartInterval(10 * 1000);
Timer_.setPeriodicInterval(10 * 1000); // every minute
Timer_.start(*GarbageCollectorCallback_, MicroServiceTimerPool());
Running_ = true; Running_ = true;
GarbageCollector_.setName("ws:garbage");
GarbageCollector_.start(*this);
std::thread CleanupThread([this](){ CleanupSessions(); });
CleanupThread.detach();
return 0; return 0;
} }
bool AP_WS_Server::Disconnect(uint64_t SerialNumber) { void AP_WS_Server::onGarbageCollecting([[maybe_unused]] Poco::Timer &timer) {
std::shared_ptr<AP_WS_Connection> Connection; static uint64_t last_log = Utils::Now(), last_zombie_run = 0;
{ auto now = Utils::Now();
auto hashIndex = MACHash::Hash(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]);
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == SerialNumbers_[hashIndex].end() || DeviceHint->second == nullptr) {
return false;
}
Connection = DeviceHint->second;
SerialNumbers_[hashIndex].erase(DeviceHint);
}
{ {
auto H = SessionHash::Hash(Connection->State_.sessionId);
std::lock_guard SessionLock(SessionMutex_[H]);
Sessions_[H].erase(Connection->State_.sessionId);
}
return true;
}
void AP_WS_Server::CleanupSessions() {
while(Running_) {
std::this_thread::sleep_for(std::chrono::seconds(10));
while(Running_ && !CleanupSessions_.empty()) {
std::pair<uint64_t, uint64_t> Session;
{ {
std::lock_guard G(CleanupMutex_); std::lock_guard SessionLock(SessionMutex_);
Session = CleanupSessions_.front(); if (!Garbage_.empty()) {
CleanupSessions_.pop_front(); Garbage_.clear();
}
poco_trace(this->Logger(),fmt::format("Cleaning up session: {} for device: {}", Session.first, Utils::IntToSerialNumber(Session.second)));
EndSession(Session.first, Session.second);
}
} }
} }
void AP_WS_Server::run() { uint64_t total_connected_time = 0;
uint64_t last_log = Utils::Now(),
last_zombie_run = 0,
last_garbage_run = 0;
Poco::Logger &LocalLogger = Poco::Logger::create( if(now-last_zombie_run > 20) {
"WS-Session-Janitor", Poco::Logger::root().getChannel(), Poco::Logger::root().getLevel()); poco_information(Logger(), fmt::format("Garbage collecting..."));
std::vector<std::uint64_t> SessionsToRemove;
while(Running_) { NumberOfConnectedDevices_ = 0;
if(!Poco::Thread::trySleep(30000)) {
break;
}
LocalLogger.information(fmt::format("Garbage collecting starting run." ));
uint64_t total_connected_time = 0, now = Utils::Now();
if(now-last_zombie_run > 60) {
try {
poco_information(LocalLogger,
fmt::format("Garbage collecting zombies... (step 1)"));
NumberOfConnectingDevices_ = 0; NumberOfConnectingDevices_ = 0;
AverageDeviceConnectionTime_ = 0; AverageDeviceConnectionTime_ = 0;
int waits = 0;
for (int hashIndex = 0; hashIndex < MACHash::HashMax(); hashIndex++) {
last_zombie_run = now; last_zombie_run = now;
waits = 0; for(int hashIndex=0;hashIndex<256;hashIndex++) {
while (true) { std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
if (SerialNumbersMutex_[hashIndex].try_lock()) {
waits = 0;
auto hint = SerialNumbers_[hashIndex].begin(); auto hint = SerialNumbers_[hashIndex].begin();
while (hint != end(SerialNumbers_[hashIndex])) { while (hint != end(SerialNumbers_[hashIndex])) {
if (hint->second == nullptr) { if (hint->second.second == nullptr) {
poco_information(
LocalLogger,
fmt::format("Dead device found in hash index {}", hashIndex));
hint = SerialNumbers_[hashIndex].erase(hint); hint = SerialNumbers_[hashIndex].erase(hint);
} else { } else if ((now - hint->second.second->State_.LastContact) >
auto Device = hint->second; SessionTimeOut_) {
auto RightNow = Utils::Now(); hint->second.second->EndConnection(false);
if (Device->Dead_) {
AddCleanupSession(Device->State_.sessionId, Device->SerialNumberInt_);
++hint;
// hint = SerialNumbers_[hashIndex].erase(hint);
} else if (RightNow > Device->LastContact_ &&
(RightNow - Device->LastContact_) > SessionTimeOut_) {
poco_information( poco_information(
LocalLogger, Logger(),
fmt::format( fmt::format(
"{}: Session seems idle. Controller disconnecting device.", "{}: Session seems idle. Controller disconnecting device.",
Device->SerialNumber_)); hint->second.second->SerialNumber_));
// hint = SerialNumbers_[hashIndex].erase(hint); SessionsToRemove.emplace_back(hint->second.first);
AddCleanupSession(Device->State_.sessionId, Device->SerialNumberInt_); Garbage_.push_back(hint->second.second);
++hint; hint = SerialNumbers_[hashIndex].erase(hint);
} else if (hint->second.second->State_.Connected) {
NumberOfConnectedDevices_++;
total_connected_time += (now - hint->second.second->State_.started);
hint++;
} else { } else {
if (Device->State_.Connected) { NumberOfConnectingDevices_++;
total_connected_time += hint++;
(RightNow - Device->State_.started);
}
++hint;
}
}
}
SerialNumbersMutex_[hashIndex].unlock();
break;
} else if (waits < 5) {
waits++;
Poco::Thread::trySleep(10);
} else {
break;
} }
} }
} }
poco_information(LocalLogger, fmt::format("Garbage collecting zombies... (step 2)")); if(SessionsToRemove.empty()) {
LeftOverSessions_ = 0; poco_information(Logger(), fmt::format("Removing {} sessions.", SessionsToRemove.size()));
for (int i = 0; i < SessionHash::HashMax(); i++) { std::lock_guard Lock(SessionMutex_);
waits = 0; for (const auto &Session : SessionsToRemove) {
while (true) { Sessions_.erase(Session);
if (SessionMutex_[i].try_lock()) {
waits = 0;
auto hint = Sessions_[i].begin();
auto RightNow = Utils::Now();
while (hint != end(Sessions_[i])) {
if (hint->second == nullptr) {
hint = Sessions_[i].erase(hint);
} else if (hint->second->Dead_) {
// hint = Sessions_[i].erase(hint);
AddCleanupSession(hint->second->State_.sessionId, hint->second->SerialNumberInt_);
++hint;
} else if (RightNow > hint->second->LastContact_ &&
(RightNow - hint->second->LastContact_) >
SessionTimeOut_) {
poco_information(
LocalLogger,
fmt::format("{}: Session seems idle. Controller disconnecting device.",
hint->second->SerialNumber_));
AddCleanupSession(hint->second->State_.sessionId, hint->second->SerialNumberInt_);
++hint;
// hint = Sessions_[i].erase(hint);
} else {
++LeftOverSessions_;
++hint;
}
}
SessionMutex_[i].unlock();
break;
} else if (waits < 5) {
Poco::Thread::trySleep(10);
waits++;
} else {
break;
}
} }
} }
AverageDeviceConnectionTime_ = NumberOfConnectedDevices_ > 0 AverageDeviceConnectionTime_ =
? total_connected_time / NumberOfConnectedDevices_ NumberOfConnectedDevices_ > 0 ? total_connected_time / NumberOfConnectedDevices_
: 0; : 0;
poco_information(LocalLogger, fmt::format("Garbage collecting zombies done..."));
} catch (const Poco::Exception &E) { poco_information(Logger(), fmt::format("Garbage collecting done..."));
poco_error(LocalLogger, fmt::format("Poco::Exception: Garbage collecting zombies failed: {}", E.displayText())); } else {
} catch (const std::exception &E) { std::lock_guard SessionLock(SessionMutex_);
poco_error(LocalLogger, fmt::format("std::exception: Garbage collecting zombies failed: {}", E.what())); NumberOfConnectedDevices_ = Sessions_.size();
} catch (...) { AverageDeviceConnectionTime_ += 10;
poco_error(LocalLogger, fmt::format("exception:Garbage collecting zombies failed: {}", "unknown"));
} }
} if ((now - last_log) > 120) {
if(NumberOfConnectedDevices_) {
if (last_garbage_run > 0) {
AverageDeviceConnectionTime_ += (now - last_garbage_run);
}
}
try {
if ((now - last_log) > 60) {
last_log = now; last_log = now;
poco_information( poco_information(Logger(),
LocalLogger, fmt::format("Active AP connections: {} Connecting: {} Average connection time: {} seconds",
fmt::format("Active AP connections: {} Connecting: {} Average connection time: {} seconds. Left Over Sessions: {}",
NumberOfConnectedDevices_, NumberOfConnectingDevices_, NumberOfConnectedDevices_, NumberOfConnectingDevices_,
AverageDeviceConnectionTime_, LeftOverSessions_)); AverageDeviceConnectionTime_));
}
} }
GWWebSocketNotifications::NumberOfConnection_t Notification; GWWebSocketNotifications::NumberOfConnection_t Notification;
@@ -421,203 +263,136 @@ namespace OpenWifi {
FullEvent.set("payload", KafkaNotification); FullEvent.set("payload", KafkaNotification);
KafkaManager()->PostMessage(KafkaTopics::DEVICE_EVENT_QUEUE, "system", FullEvent); KafkaManager()->PostMessage(KafkaTopics::DEVICE_EVENT_QUEUE, "system", FullEvent);
LocalLogger.information(fmt::format("Garbage collection finished run."));
last_garbage_run = now;
} catch (const Poco::Exception &E) {
LocalLogger.error(fmt::format("Poco::Exception: Garbage collecting failed: {}", E.displayText()));
} catch (const std::exception &E) {
LocalLogger.error(fmt::format("std::exception: Garbage collecting failed: {}", E.what()));
} catch (...) {
LocalLogger.error(fmt::format("exception:Garbage collecting failed: {}", "unknown"));
}
}
LocalLogger.information(fmt::format("Garbage collector done for the day." ));
} }
void AP_WS_Server::Stop() { void AP_WS_Server::Stop() {
poco_information(Logger(), "Stopping..."); poco_information(Logger(), "Stopping...");
Running_ = false; Running_ = false;
GarbageCollector_.wakeUp(); Timer_.stop();
GarbageCollector_.join();
for (auto &server : WebServers_) { for (auto &server : WebServers_) {
server->stopAll(); server->stopAll();
} }
Reactor_pool_->Stop(); Reactor_pool_->Stop();
Reactor_.stop(); Reactor_.stop();
ReactorThread_.join(); ReactorThread_.join();
poco_information(Logger(), "Stopped..."); poco_information(Logger(), "Stopped...");
} }
bool AP_WS_Server::GetHealthDevices(std::uint64_t lowLimit, std::uint64_t highLimit, std::vector<std::string> & SerialNumbers) {
SerialNumbers.clear();
for(int i=0;i<SessionHash::HashMax();i++) {
std::lock_guard Lock(SessionMutex_[i]);
for (const auto &connection : Sessions_[i]) {
if (connection.second->RawLastHealthcheck_.Sanity >= lowLimit &&
connection.second->RawLastHealthcheck_.Sanity <= highLimit) {
SerialNumbers.push_back(connection.second->SerialNumber_);
}
}
}
return true;
}
bool AP_WS_Server::GetStatistics(uint64_t SerialNumber, std::string &Statistics) const { bool AP_WS_Server::GetStatistics(uint64_t SerialNumber, std::string &Statistics) const {
std::shared_ptr<AP_WS_Connection> Connection;
{ auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
auto hashIndex = MACHash::Hash(SerialNumber); std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber); if (Device == SerialNumbers_[hashIndex].end() || Device->second.second == nullptr) {
if (DeviceHint == SerialNumbers_[hashIndex].end() || DeviceHint->second == nullptr) {
return false; return false;
} }
Connection = DeviceHint->second; Device->second.second->GetLastStats(Statistics);
}
Connection->GetLastStats(Statistics);
return true; return true;
} }
bool AP_WS_Server::GetState(uint64_t SerialNumber, GWObjects::ConnectionState &State) const { bool AP_WS_Server::GetState(uint64_t SerialNumber, GWObjects::ConnectionState &State) const {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); if (Device == SerialNumbers_[hashIndex].end() || Device->second.second == nullptr) {
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == SerialNumbers_[hashIndex].end() ||
DeviceHint->second == nullptr) {
return false; return false;
} }
Connection = DeviceHint->second; Device->second.second->GetState(State);
}
Connection->GetState(State);
return true; return true;
} }
bool AP_WS_Server::GetHealthcheck(uint64_t SerialNumber, bool AP_WS_Server::GetHealthcheck(uint64_t SerialNumber,
GWObjects::HealthCheck &CheckData) const { GWObjects::HealthCheck &CheckData) const {
std::shared_ptr<AP_WS_Connection> Connection;
{ auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
auto hashIndex = MACHash::Hash(SerialNumber); std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]);
auto Device = SerialNumbers_[hashIndex].find(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
if (Device == SerialNumbers_[hashIndex].end() || Device->second == nullptr) { if (Device == SerialNumbers_[hashIndex].end() || Device->second.second == nullptr) {
return false; return false;
} }
Connection = Device->second; Device->second.second->GetLastHealthCheck(CheckData);
}
Connection->GetLastHealthCheck(CheckData);
return true; return true;
} }
void AP_WS_Server::StartSession(uint64_t session_id, uint64_t SerialNumber) { void AP_WS_Server::SetSessionDetails(uint64_t connection_id, uint64_t SerialNumber) {
auto sessionHash = SessionHash::Hash(session_id); std::lock_guard SessionLock(SessionMutex_);
std::shared_ptr<AP_WS_Connection> Connection; auto Conn = Sessions_.find(connection_id);
{ if (Conn == end(Sessions_))
std::lock_guard SessionLock(SessionMutex_[sessionHash]); return;
auto SessionHint = Sessions_[sessionHash].find(session_id);
if (SessionHint == end(Sessions_[sessionHash])) { auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto CurrentSerialNumber = SerialNumbers_[hashIndex].find(SerialNumber);
if ((CurrentSerialNumber == SerialNumbers_[hashIndex].end()) ||
(CurrentSerialNumber->second.first < connection_id)) {
SerialNumbers_[hashIndex][SerialNumber] = std::make_pair(connection_id, Conn->second);
return; return;
} }
Connection = SessionHint->second;
Sessions_[sessionHash].erase(SessionHint);
}
auto deviceHash = MACHash::Hash(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[deviceHash]);
SerialNumbers_[deviceHash][SerialNumber] = Connection;
} }
bool AP_WS_Server::EndSession(uint64_t session_id, uint64_t SerialNumber) { bool AP_WS_Server::EndSession(uint64_t session_id, uint64_t SerialNumber) {
{ std::lock_guard SessionLock(SessionMutex_);
poco_trace(Logger(), fmt::format("Ending session 1: {} for device: {}", session_id, Utils::IntToSerialNumber(SerialNumber))); auto Session = Sessions_.find(session_id);
auto sessionHash = SessionHash::Hash(session_id); if (Session == end(Sessions_))
std::lock_guard SessionLock(SessionMutex_[sessionHash]); return false;
Sessions_[sessionHash].erase(session_id);
poco_trace(Logger(), fmt::format("Ended session 1: {} for device: {}", session_id, Utils::IntToSerialNumber(SerialNumber)));
}
{ Garbage_.push_back(Session->second);
auto hashIndex = MACHash::Hash(SerialNumber);
poco_trace(Logger(), fmt::format("Ending session 2.0: {} for device: {} hi:{}", session_id, Utils::IntToSerialNumber(SerialNumber), hashIndex)); auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
poco_trace(Logger(), fmt::format("Ending session 2.1: {} for device: {} hi:{}", session_id, Utils::IntToSerialNumber(SerialNumber), hashIndex)); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber); if (Device == end(SerialNumbers_[hashIndex])) {
poco_trace(Logger(), fmt::format("Ending session 2.2: {} for device: {} hi:{}", session_id, Utils::IntToSerialNumber(SerialNumber), hashIndex)); Sessions_.erase(Session);
if (DeviceHint == SerialNumbers_[hashIndex].end()
|| DeviceHint->second == nullptr
|| DeviceHint->second->State_.sessionId != session_id) {
poco_trace(Logger(), fmt::format("Did not end session 2: {} for device: {}", session_id, Utils::IntToSerialNumber(SerialNumber)));
return false; return false;
} }
SerialNumbers_[hashIndex].erase(DeviceHint);
poco_trace(Logger(), fmt::format("Ended session 2: {} for device: {}", session_id, Utils::IntToSerialNumber(SerialNumber))); if (Device->second.first == session_id) {
} Sessions_.erase(Session);
SerialNumbers_[hashIndex].erase(Device);
return true; return true;
} }
Sessions_.erase(Session);
return false;
}
bool AP_WS_Server::Connected(uint64_t SerialNumber, bool AP_WS_Server::Connected(uint64_t SerialNumber,
GWObjects::DeviceRestrictions &Restrictions) const { GWObjects::DeviceRestrictions &Restrictions) const {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) {
return false; return false;
} }
Connection = DeviceHint->second; Device->second.second->GetRestrictions(Restrictions);
return Device->second.second->State_.Connected;
} }
if(Connection->Dead_) {
return false;
}
Restrictions = Connection->GetRestrictions();
return Connection->State_.Connected;
}
bool AP_WS_Server::Connected(uint64_t SerialNumber) const { bool AP_WS_Server::Connected(uint64_t SerialNumber) const {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) {
return false; return false;
} }
Connection = DeviceHint->second; return Device->second.second->State_.Connected;
}
if(Connection->Dead_) {
return false;
}
return Connection->State_.Connected;
} }
bool AP_WS_Server::SendFrame(uint64_t SerialNumber, const std::string &Payload) const { bool AP_WS_Server::SendFrame(uint64_t SerialNumber, const std::string &Payload) const {
auto hashIndex = MACHash::Hash(SerialNumber); auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
std::shared_ptr<AP_WS_Connection> Connection; auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
{ if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]);
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) {
return false;
}
Connection = DeviceHint->second;
}
if(Connection->Dead_) {
return false; return false;
} }
try { try {
return Connection->Send(Payload); return Device->second.second->Send(Payload);
} catch (...) { } catch (...) {
poco_debug(Logger(), fmt::format(": SendFrame: Could not send data to device '{}'", poco_debug(Logger(), fmt::format(": SendFrame: Could not send data to device '{}'",
Utils::IntToSerialNumber(SerialNumber))); Utils::IntToSerialNumber(SerialNumber)));
@@ -626,64 +401,48 @@ namespace OpenWifi {
} }
void AP_WS_Server::StopWebSocketTelemetry(uint64_t RPCID, uint64_t SerialNumber) { void AP_WS_Server::StopWebSocketTelemetry(uint64_t RPCID, uint64_t SerialNumber) {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]);
auto Device = SerialNumbers_[hashIndex].find(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
if (Device == end(SerialNumbers_[hashIndex]) || Device->second == nullptr) { if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
return; return;
} }
Connection = Device->second; Device->second.second->StopWebSocketTelemetry(RPCID);
}
Connection->StopWebSocketTelemetry(RPCID);
} }
void void
AP_WS_Server::SetWebSocketTelemetryReporting(uint64_t RPCID, uint64_t SerialNumber, AP_WS_Server::SetWebSocketTelemetryReporting(uint64_t RPCID, uint64_t SerialNumber,
uint64_t Interval, uint64_t Lifetime, uint64_t Interval, uint64_t Lifetime,
const std::vector<std::string> &TelemetryTypes) { const std::vector<std::string> &TelemetryTypes) {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) {
return; return;
} }
Connection = DeviceHint->second; Device->second.second->SetWebSocketTelemetryReporting(RPCID, Interval, Lifetime, TelemetryTypes);
}
Connection->SetWebSocketTelemetryReporting(RPCID, Interval, Lifetime, TelemetryTypes);
} }
void AP_WS_Server::SetKafkaTelemetryReporting(uint64_t RPCID, uint64_t SerialNumber, void AP_WS_Server::SetKafkaTelemetryReporting(uint64_t RPCID, uint64_t SerialNumber,
uint64_t Interval, uint64_t Lifetime, uint64_t Interval, uint64_t Lifetime,
const std::vector<std::string> &TelemetryTypes) { const std::vector<std::string> &TelemetryTypes) {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) {
return; return;
} }
Connection = DeviceHint->second; Device->second.second->SetKafkaTelemetryReporting(RPCID, Interval, Lifetime, TelemetryTypes);
}
Connection->SetKafkaTelemetryReporting(RPCID, Interval, Lifetime, TelemetryTypes);
} }
void AP_WS_Server::StopKafkaTelemetry(uint64_t RPCID, uint64_t SerialNumber) { void AP_WS_Server::StopKafkaTelemetry(uint64_t RPCID, uint64_t SerialNumber) {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
std::lock_guard DevicesLock(SerialNumbersMutex_[hashIndex]); if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) {
return; return;
} }
Connection = DeviceHint->second; Device->second.second->StopKafkaTelemetry(RPCID);
}
Connection->StopKafkaTelemetry(RPCID);
} }
void AP_WS_Server::GetTelemetryParameters( void AP_WS_Server::GetTelemetryParameters(
@@ -692,18 +451,14 @@ namespace OpenWifi {
uint64_t &TelemetryWebSocketCount, uint64_t &TelemetryKafkaCount, uint64_t &TelemetryWebSocketCount, uint64_t &TelemetryKafkaCount,
uint64_t &TelemetryWebSocketPackets, uint64_t &TelemetryKafkaPackets) { uint64_t &TelemetryWebSocketPackets, uint64_t &TelemetryKafkaPackets) {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(SerialNumber);
{ std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(SerialNumber); auto Device = SerialNumbers_[hashIndex].find(SerialNumber);
std::lock_guard DevicesLock(SerialNumbersMutex_[hashIndex]); if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
auto DeviceHint = SerialNumbers_[hashIndex].find(SerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) {
return; return;
} }
Connection = DeviceHint->second;
}
Connection->GetTelemetryParameters(TelemetryRunning, TelemetryInterval, Device->second.second->GetTelemetryParameters(TelemetryRunning, TelemetryInterval,
TelemetryWebSocketTimer, TelemetryKafkaTimer, TelemetryWebSocketTimer, TelemetryKafkaTimer,
TelemetryWebSocketCount, TelemetryKafkaCount, TelemetryWebSocketCount, TelemetryKafkaCount,
TelemetryWebSocketPackets, TelemetryKafkaPackets); TelemetryWebSocketPackets, TelemetryKafkaPackets);
@@ -712,24 +467,16 @@ namespace OpenWifi {
bool AP_WS_Server::SendRadiusAccountingData(const std::string &SerialNumber, bool AP_WS_Server::SendRadiusAccountingData(const std::string &SerialNumber,
const unsigned char *buffer, std::size_t size) { const unsigned char *buffer, std::size_t size) {
std::shared_ptr<AP_WS_Connection> Connection;
{
auto IntSerialNumber = Utils::SerialNumberToInt(SerialNumber); auto IntSerialNumber = Utils::SerialNumberToInt(SerialNumber);
auto hashIndex = MACHash::Hash(IntSerialNumber); auto hashIndex = Utils::CalculateMacAddressHash(IntSerialNumber);
std::lock_guard DevicesLock(SerialNumbersMutex_[hashIndex]); std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto DeviceHint = SerialNumbers_[hashIndex].find(IntSerialNumber); auto Device = SerialNumbers_[hashIndex].find(IntSerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) { if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
return false;
}
Connection = DeviceHint->second;
}
if(Connection->Dead_) {
return false; return false;
} }
try { try {
return Connection->SendRadiusAccountingData(buffer, size); return Device->second.second->SendRadiusAccountingData(buffer, size);
} catch (...) { } catch (...) {
poco_debug( poco_debug(
Logger(), Logger(),
@@ -741,24 +488,16 @@ namespace OpenWifi {
bool AP_WS_Server::SendRadiusAuthenticationData(const std::string &SerialNumber, bool AP_WS_Server::SendRadiusAuthenticationData(const std::string &SerialNumber,
const unsigned char *buffer, std::size_t size) { const unsigned char *buffer, std::size_t size) {
std::shared_ptr<AP_WS_Connection> Connection;
{
auto IntSerialNumber = Utils::SerialNumberToInt(SerialNumber); auto IntSerialNumber = Utils::SerialNumberToInt(SerialNumber);
auto hashIndex = MACHash::Hash(IntSerialNumber); auto hashIndex = Utils::CalculateMacAddressHash(IntSerialNumber);
std::lock_guard DevicesLock(SerialNumbersMutex_[hashIndex]); std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto DeviceHint = SerialNumbers_[hashIndex].find(IntSerialNumber); auto Device = SerialNumbers_[hashIndex].find(IntSerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) { if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
return false;
}
Connection = DeviceHint->second;
}
if(Connection->Dead_) {
return false; return false;
} }
try { try {
return Connection->SendRadiusAuthenticationData(buffer, size); return Device->second.second->SendRadiusAuthenticationData(buffer, size);
} catch (...) { } catch (...) {
poco_debug( poco_debug(
Logger(), Logger(),
@@ -770,23 +509,16 @@ namespace OpenWifi {
bool AP_WS_Server::SendRadiusCoAData(const std::string &SerialNumber, bool AP_WS_Server::SendRadiusCoAData(const std::string &SerialNumber,
const unsigned char *buffer, std::size_t size) { const unsigned char *buffer, std::size_t size) {
std::shared_ptr<AP_WS_Connection> Connection;
{
auto IntSerialNumber = Utils::SerialNumberToInt(SerialNumber); auto IntSerialNumber = Utils::SerialNumberToInt(SerialNumber);
auto hashIndex = MACHash::Hash(IntSerialNumber); auto hashIndex = Utils::CalculateMacAddressHash(IntSerialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); std::lock_guard Lock(SerialNumbersMutex_[hashIndex]);
auto DeviceHint = SerialNumbers_[hashIndex].find(IntSerialNumber); auto Device = SerialNumbers_[hashIndex].find(IntSerialNumber);
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr) { if (Device == end(SerialNumbers_[hashIndex]) || Device->second.second == nullptr) {
return false; return false;
} }
Connection = DeviceHint->second;
}
if(Connection->Dead_) {
return false;
}
try { try {
return Connection->SendRadiusCoAData(buffer, size); return Device->second.second->SendRadiusCoAData(buffer, size);
} catch (...) { } catch (...) {
poco_debug(Logger(), poco_debug(Logger(),
fmt::format(": SendRadiusCoAData: Could not send data to device '{}'", fmt::format(": SendRadiusCoAData: Could not send data to device '{}'",

View File

@@ -24,51 +24,46 @@
#include "Poco/Timer.h" #include "Poco/Timer.h"
#include "AP_WS_Connection.h" #include "AP_WS_Connection.h"
#include "AP_WS_Reactor_Pool.h" #include "AP_WS_ReactorPool.h"
#include "framework/SubSystemServer.h" #include "framework/SubSystemServer.h"
#include "framework/utils.h" #include "framework/utils.h"
namespace OpenWifi { namespace OpenWifi {
constexpr uint MACHashMax = 256; class AP_WS_RequestHandler : public Poco::Net::HTTPRequestHandler {
constexpr uint MACHashMask = MACHashMax-1;
class MACHash {
public: public:
[[nodiscard]] static inline uint16_t Hash(std::uint64_t value) { explicit AP_WS_RequestHandler(Poco::Logger &L, uint64_t id) : Logger_(L), id_(id){};
uint8_t hash = 0, i=6;
while(i) {
hash ^= (value & MACHashMask) + 1;
value >>= 8;
--i;
}
return hash;
}
[[nodiscard]] static inline uint16_t Hash(const std::string & value) { void handleRequest(Poco::Net::HTTPServerRequest &request,
return Hash(Utils::MACToInt(value)); Poco::Net::HTTPServerResponse &response) override;
}
[[nodiscard]] static inline uint16_t HashMax() { private:
return MACHashMax; Poco::Logger &Logger_;
} uint64_t id_ = 0;
}; };
constexpr uint SessionHashMax = 256; class AP_WS_RequestHandlerFactory : public Poco::Net::HTTPRequestHandlerFactory {
constexpr uint SessionHashMask = SessionHashMax-1;
class SessionHash {
public: public:
[[nodiscard]] static inline uint16_t Hash(std::uint64_t value) { inline explicit AP_WS_RequestHandlerFactory(Poco::Logger &L) : Logger_(L) {}
return (value & SessionHashMask);
inline Poco::Net::HTTPRequestHandler *
createRequestHandler(const Poco::Net::HTTPServerRequest &request) override {
if (request.find("Upgrade") != request.end() &&
Poco::icompare(request["Upgrade"], "websocket") == 0) {
Utils::SetThreadName("ws:conn-init");
return new AP_WS_RequestHandler(Logger_, id_++);
} else {
return nullptr;
}
} }
[[nodiscard]] static inline uint16_t HashMax() { private:
return SessionHashMax; Poco::Logger &Logger_;
} inline static uint64_t id_ = 1;
}; };
class AP_WS_Server : public SubSystemServer {
class AP_WS_Server : public SubSystemServer, public Poco::Runnable {
public: public:
static auto instance() { static auto instance() {
static auto instance_ = new AP_WS_Server; static auto instance_ = new AP_WS_Server;
@@ -82,50 +77,48 @@ namespace OpenWifi {
const Poco::Crypto::X509Certificate &Certificate); const Poco::Crypto::X509Certificate &Certificate);
inline bool IsSimSerialNumber(const std::string &SerialNumber) const { inline bool IsSimSerialNumber(const std::string &SerialNumber) const {
return IsSim(SerialNumber) && return IsSim(Poco::toLower(SerialNumber)) &&
SerialNumber == SimulatorId_; Poco::toLower(SerialNumber) == Poco::toLower(SimulatorId_);
} }
inline static bool IsSim(const std::string &SerialNumber) { inline static bool IsSim(const std::string &SerialNumber) {
return SerialNumber.substr(0, 6) == "53494d"; return SerialNumber.substr(0, 6) == "53494d";
} }
void run() override; // Garbage collector thread. inline bool IsSimEnabled() const { return SimulatorEnabled_; }
[[nodiscard]] inline bool IsSimEnabled() const { return SimulatorEnabled_; }
[[nodiscard]] inline bool AllowSerialNumberMismatch() const { return AllowSerialNumberMismatch_; } inline bool AllowSerialNumberMismatch() const { return AllowSerialNumberMismatch_; }
[[nodiscard]] inline uint64_t MismatchDepth() const { return MismatchDepth_; }
[[nodiscard]] inline bool UseProvisioning() const { return LookAtProvisioning_; } inline uint64_t MismatchDepth() const { return MismatchDepth_; }
[[nodiscard]] inline bool UseDefaults() const { return UseDefaultConfig_; }
[[nodiscard]] inline bool Running() const { return Running_; } inline bool UseProvisioning() const { return LookAtProvisioning_; }
[[nodiscard]] inline std::pair<std::shared_ptr<Poco::Net::SocketReactor>, std::shared_ptr<LockedDbSession>> NextReactor() { inline bool UseDefaults() const { return UseDefaultConfig_; }
[[nodiscard]] inline Poco::Net::SocketReactor &NextReactor() {
return Reactor_pool_->NextReactor(); return Reactor_pool_->NextReactor();
} }
[[nodiscard]] inline bool Running() const { return Running_; }
inline void AddConnection(std::shared_ptr<AP_WS_Connection> Connection) { inline void AddConnection(uint64_t session_id,
std::uint64_t sessionHash = SessionHash::Hash(Connection->State_.sessionId); std::shared_ptr<AP_WS_Connection> Connection) {
std::lock_guard SessionLock(SessionMutex_[sessionHash]); std::lock_guard Lock(SessionMutex_);
if(Sessions_[sessionHash].find(Connection->State_.sessionId)==end(Sessions_[sessionHash])) { Sessions_[session_id] = std::move(Connection);
Sessions_[sessionHash][Connection->State_.sessionId] = std::move(Connection);
}
} }
[[nodiscard]] inline bool DeviceRequiresSecureRTTY(uint64_t serialNumber) const { inline bool DeviceRequiresSecureRtty(uint64_t serialNumber) const {
std::shared_ptr<AP_WS_Connection> Connection; auto hashIndex = Utils::CalculateMacAddressHash(serialNumber);
{ std::lock_guard G(SerialNumbersMutex_[hashIndex]);
auto hashIndex = MACHash::Hash(serialNumber);
std::lock_guard DeviceLock(SerialNumbersMutex_[hashIndex]); auto Connection = SerialNumbers_[hashIndex].find(serialNumber);
auto DeviceHint = SerialNumbers_[hashIndex].find(serialNumber); if (Connection==end(SerialNumbers_[hashIndex]) || Connection->second.second==nullptr)
if (DeviceHint == end(SerialNumbers_[hashIndex]) || DeviceHint->second == nullptr)
return false; return false;
Connection = DeviceHint->second; return Connection->second.second->RttyMustBeSecure_;
}
return Connection->RTTYMustBeSecure_;
} }
inline bool GetStatistics(const std::string &SerialNumber, std::string &Statistics) const { inline bool GetStatistics(const std::string &SerialNumber, std::string &Statistics) const {
return GetStatistics(Utils::SerialNumberToInt(SerialNumber), Statistics); return GetStatistics(Utils::SerialNumberToInt(SerialNumber), Statistics);
} }
[[nodiscard]] bool GetStatistics(uint64_t SerialNumber, std::string &Statistics) const; bool GetStatistics(uint64_t SerialNumber, std::string &Statistics) const;
inline bool GetState(const std::string &SerialNumber, inline bool GetState(const std::string &SerialNumber,
GWObjects::ConnectionState &State) const { GWObjects::ConnectionState &State) const {
@@ -141,8 +134,13 @@ namespace OpenWifi {
bool Connected(uint64_t SerialNumber, GWObjects::DeviceRestrictions &Restrictions) const; bool Connected(uint64_t SerialNumber, GWObjects::DeviceRestrictions &Restrictions) const;
bool Connected(uint64_t SerialNumber) const; bool Connected(uint64_t SerialNumber) const;
bool Disconnect(uint64_t SerialNumber);
inline bool SendFrame(const std::string &SerialNumber, const std::string &Payload) const {
return SendFrame(Utils::SerialNumberToInt(SerialNumber), Payload);
}
bool SendFrame(uint64_t SerialNumber, const std::string &Payload) const; bool SendFrame(uint64_t SerialNumber, const std::string &Payload) const;
bool SendRadiusAuthenticationData(const std::string &SerialNumber, bool SendRadiusAuthenticationData(const std::string &SerialNumber,
const unsigned char *buffer, std::size_t size); const unsigned char *buffer, std::size_t size);
bool SendRadiusAccountingData(const std::string &SerialNumber, const unsigned char *buffer, bool SendRadiusAccountingData(const std::string &SerialNumber, const unsigned char *buffer,
@@ -150,8 +148,9 @@ namespace OpenWifi {
bool SendRadiusCoAData(const std::string &SerialNumber, const unsigned char *buffer, bool SendRadiusCoAData(const std::string &SerialNumber, const unsigned char *buffer,
std::size_t size); std::size_t size);
void StartSession(uint64_t session_id, uint64_t SerialNumber); void SetSessionDetails(uint64_t connection_id, uint64_t SerialNumber);
bool EndSession(uint64_t session_id, uint64_t SerialNumber); bool EndSession(uint64_t connection_id, uint64_t serial_number);
bool EndSessionUnSafe(uint64_t session_id, uint64_t serial_number);
void SetWebSocketTelemetryReporting(uint64_t RPCID, uint64_t SerialNumber, void SetWebSocketTelemetryReporting(uint64_t RPCID, uint64_t SerialNumber,
uint64_t Interval, uint64_t Lifetime, uint64_t Interval, uint64_t Lifetime,
const std::vector<std::string> &TelemetryTypes); const std::vector<std::string> &TelemetryTypes);
@@ -168,9 +167,7 @@ namespace OpenWifi {
uint64_t &TelemetryWebSocketPackets, uint64_t &TelemetryWebSocketPackets,
uint64_t &TelemetryKafkaPackets); uint64_t &TelemetryKafkaPackets);
bool GetHealthDevices(std::uint64_t lowLimit, std::uint64_t highLimit, std::vector<std::string> & SerialNumbers); void onGarbageCollecting(Poco::Timer &timer);
// bool ExtendedAttributes(const std::string &serialNumber, bool & hasGPS, std::uint64_t &Sanity,
// std::double_t &MemoryUsed, std::double_t &Load, std::double_t &Temperature);
inline void AverageDeviceStatistics(uint64_t &Connections, uint64_t &AverageConnectionTime, inline void AverageDeviceStatistics(uint64_t &Connections, uint64_t &AverageConnectionTime,
uint64_t &NumberOfConnectingDevices) const { uint64_t &NumberOfConnectingDevices) const {
@@ -179,80 +176,93 @@ namespace OpenWifi {
NumberOfConnectingDevices = NumberOfConnectingDevices_; NumberOfConnectingDevices = NumberOfConnectingDevices_;
} }
inline bool SendFrame(const std::string &SerialNumber, const std::string &Payload) const {
return SendFrame(Utils::SerialNumberToInt(SerialNumber), Payload);
}
inline void AddRX(std::uint64_t bytes) { inline void AddRX(std::uint64_t bytes) {
std::lock_guard G(StatsMutex_);
RX_ += bytes; RX_ += bytes;
} }
inline void AddTX(std::uint64_t bytes) { inline void AddTX(std::uint64_t bytes) {
std::lock_guard G(StatsMutex_);
TX_ += bytes; TX_ += bytes;
} }
inline void GetTotalDataStatistics(std::uint64_t &TX, std::uint64_t &RX) const { inline void GetTotalDataStatistics(std::uint64_t &TX, std::uint64_t &RX) const {
std::lock_guard G(StatsMutex_);
TX = TX_; TX = TX_;
RX = RX_; RX = RX_;
} }
bool KafkaDisableState() const { return KafkaDisableState_; } // TOD: move to hash based map.
bool KafkaDisableHealthChecks() const { return KafkaDisableHealthChecks_; } inline bool GetHealthDevices(std::uint64_t lowLimit, std::uint64_t highLimit, std::vector<std::string> & SerialNumbers) {
std::lock_guard Lock(SessionMutex_);
inline void IncrementConnectionCount() { for(const auto &connection:Sessions_) {
++NumberOfConnectedDevices_; if( connection.second->RawLastHealthcheck_.Sanity>=lowLimit &&
connection.second->RawLastHealthcheck_.Sanity<=highLimit) {
SerialNumbers.push_back(connection.second->SerialNumber_);
}
}
return true;
} }
inline void DecrementConnectionCount() { inline bool ExtendedAttributes(const std::string &serialNumber,
--NumberOfConnectedDevices_; bool & hasGPS,
} std::uint64_t &Sanity,
std::double_t &MemoryUsed,
std::double_t &Load,
std::double_t &Temperature
) {
inline void AddCleanupSession(uint64_t session_id, uint64_t SerialNumber) { auto serialNumberInt = Utils::SerialNumberToInt(serialNumber);
std::lock_guard G(CleanupMutex_); auto hashIndex = Utils::CalculateMacAddressHash(serialNumberInt);
CleanupSessions_.emplace_back(session_id, SerialNumber); std::lock_guard G(SerialNumbersMutex_[hashIndex]);
auto session_hint = SerialNumbers_[hashIndex].find(Utils::SerialNumberToInt(serialNumber));
if(session_hint==end(SerialNumbers_[hashIndex])) {
return false;
}
hasGPS = session_hint->second.second->hasGPS;
Sanity = session_hint->second.second->RawLastHealthcheck_.Sanity;
MemoryUsed = session_hint->second.second->memory_used_;
Load = session_hint->second.second->cpu_load_;
Temperature = session_hint->second.second->temperature_;
return true;
} }
void CleanupSessions();
private: private:
std::array<std::mutex,SessionHashMax> SessionMutex_; mutable std::mutex SessionMutex_;
std::array<std::map<std::uint64_t, std::shared_ptr<AP_WS_Connection>>,SessionHashMax> Sessions_; mutable std::mutex StatsMutex_;
using SerialNumberMap = std::map<uint64_t /* serial number */,
std::shared_ptr<AP_WS_Connection>>;
std::array<SerialNumberMap,MACHashMax> SerialNumbers_;
mutable std::array<std::mutex,MACHashMax> SerialNumbersMutex_;
std::unique_ptr<Poco::Crypto::X509Certificate> IssuerCert_; std::unique_ptr<Poco::Crypto::X509Certificate> IssuerCert_;
std::vector<Poco::Crypto::X509Certificate> ClientCasCerts_;
std::list<std::unique_ptr<Poco::Net::HTTPServer>> WebServers_; std::list<std::unique_ptr<Poco::Net::HTTPServer>> WebServers_;
Poco::ThreadPool DeviceConnectionPool_{"ws:dev-pool", 4, 256};
Poco::Net::SocketReactor Reactor_; Poco::Net::SocketReactor Reactor_;
Poco::Thread ReactorThread_; Poco::Thread ReactorThread_;
std::string SimulatorId_; std::string SimulatorId_;
Poco::ThreadPool DeviceConnectionPool_{"ws:dev-pool", 2, 64};
bool LookAtProvisioning_ = false; bool LookAtProvisioning_ = false;
bool UseDefaultConfig_ = true; bool UseDefaultConfig_ = true;
bool SimulatorEnabled_ = false; bool SimulatorEnabled_ = false;
bool AllowSerialNumberMismatch_ = true;
Poco::Thread CleanupThread_;
std::mutex CleanupMutex_;
std::deque<std::pair<uint64_t, uint64_t>> CleanupSessions_;
std::unique_ptr<AP_WS_ReactorThreadPool> Reactor_pool_; std::unique_ptr<AP_WS_ReactorThreadPool> Reactor_pool_;
std::atomic_bool Running_ = false; std::atomic_bool Running_ = false;
std::map<std::uint64_t, std::shared_ptr<AP_WS_Connection>> Sessions_;
std::uint64_t MismatchDepth_ = 2; using SerialNumberMap = std::map<uint64_t /* serial number */, std::pair<uint64_t /* session id*/,
std::shared_ptr<AP_WS_Connection>>>;
std::atomic_uint64_t NumberOfConnectedDevices_ = 0; std::array<SerialNumberMap,256> SerialNumbers_;
std::atomic_uint64_t AverageDeviceConnectionTime_ = 0; mutable std::array<std::mutex,256> SerialNumbersMutex_;
std::atomic_bool AllowSerialNumberMismatch_ = true;
std::atomic_uint64_t MismatchDepth_ = 2;
std::uint64_t NumberOfConnectedDevices_ = 0;
std::uint64_t AverageDeviceConnectionTime_ = 0;
std::uint64_t NumberOfConnectingDevices_ = 0; std::uint64_t NumberOfConnectingDevices_ = 0;
std::uint64_t SessionTimeOut_ = 10*60; std::uint64_t SessionTimeOut_ = 10*60;
std::uint64_t LeftOverSessions_ = 0;
std::atomic_uint64_t TX_=0,RX_=0; std::atomic_uint64_t TX_=0,RX_=0;
std::atomic_bool KafkaDisableState_=false, std::vector<std::shared_ptr<AP_WS_Connection>> Garbage_;
KafkaDisableHealthChecks_=false;
std::unique_ptr<Poco::TimerCallback<AP_WS_Server>> GarbageCollectorCallback_;
Poco::Timer Timer_;
Poco::Thread GarbageCollector_; Poco::Thread GarbageCollector_;
AP_WS_Server() noexcept AP_WS_Server() noexcept

View File

@@ -10,7 +10,6 @@
#include <string> #include <string>
#include "framework/MicroServiceFuncs.h" #include "framework/MicroServiceFuncs.h"
#include "framework/ow_constants.h"
#include "CentralConfig.h" #include "CentralConfig.h"
#include "nlohmann/json.hpp" #include "nlohmann/json.hpp"
@@ -35,7 +34,7 @@ namespace OpenWifi {
std::lock_guard G(Mutex_); std::lock_guard G(Mutex_);
if (!PlatformsLoaded_) if (!PlatformsLoaded_)
LoadPlatforms(); LoadPlatforms();
auto P = Poco::toLower(Caps.Platform()); auto P = Poco::toUpper(Caps.Platform());
auto Hint = Platforms_.find(Caps.Compatible()); auto Hint = Platforms_.find(Caps.Compatible());
if (Hint == Platforms_.end()) { if (Hint == Platforms_.end()) {
Platforms_.insert(std::make_pair(Caps.Compatible(), P)); Platforms_.insert(std::make_pair(Caps.Compatible(), P));
@@ -69,7 +68,7 @@ namespace OpenWifi {
auto Hint = Platforms_.find(DeviceType); auto Hint = Platforms_.find(DeviceType);
if (Hint == Platforms_.end()) if (Hint == Platforms_.end())
return Platforms::AP; return "AP";
return Hint->second; return Hint->second;
} }
@@ -111,7 +110,7 @@ namespace OpenWifi {
i >> cache; i >> cache;
for (const auto &[Type, Platform] : cache.items()) { for (const auto &[Type, Platform] : cache.items()) {
Platforms_[Type] = Poco::toLower(Platform.get<std::string>()); Platforms_[Type] = Platform;
} }
} catch (...) { } catch (...) {
} }

View File

@@ -204,17 +204,6 @@ namespace OpenWifi::Config {
return false; return false;
} }
std::uint64_t Config::UUID() {
try {
Poco::JSON::Parser Parser;
auto object = Parser.parse(Config_).extract<Poco::JSON::Object::Ptr>();
if (object->has("uuid"))
return object->get("uuid");
} catch (...) {
}
return 0;
}
bool Config::Valid() { bool Config::Valid() {
try { try {
Poco::JSON::Parser Parser; Poco::JSON::Parser Parser;
@@ -265,11 +254,7 @@ namespace OpenWifi::Config {
Model_ = Caps->get("model").toString(); Model_ = Caps->get("model").toString();
if (Caps->has("platform")) if (Caps->has("platform"))
Platform_ = Poco::toLower(Caps->get("platform").toString()); Platform_ = Caps->get("platform").toString();
if(Compatible_.empty()) {
Compatible_ = Model_;
}
std::ostringstream OS; std::ostringstream OS;
Caps->stringify(OS); Caps->stringify(OS);

View File

@@ -23,7 +23,6 @@ namespace OpenWifi::Config {
[[nodiscard]] std::string get() { return Config_; }; [[nodiscard]] std::string get() { return Config_; };
[[nodiscard]] std::string Default(); [[nodiscard]] std::string Default();
[[nodiscard]] Poco::JSON::Object::Ptr to_json(); [[nodiscard]] Poco::JSON::Object::Ptr to_json();
[[nodiscard]] std::uint64_t UUID();
private: private:
void Init(); void Init();

View File

@@ -45,9 +45,11 @@ namespace OpenWifi {
std::lock_guard Lock(LocalMutex_); std::lock_guard Lock(LocalMutex_);
auto RPC = OutStandingRequests_.find(ID); auto RPC = OutStandingRequests_.find(ID);
if (RPC == OutStandingRequests_.end()) { if (RPC == OutStandingRequests_.end()) {
// std::cout << __LINE__ << std::endl;
poco_debug(Logger(), fmt::format("({}): RPC {} cannot be found.", poco_debug(Logger(), fmt::format("({}): RPC {} cannot be found.",
SerialNumberStr, ID)); SerialNumberStr, ID));
} else if (RPC->second.SerialNumber != Resp->SerialNumber_) { } else if (RPC->second.SerialNumber != Resp->SerialNumber_) {
// std::cout << __LINE__ << std::endl;
poco_debug( poco_debug(
Logger(), Logger(),
fmt::format("({}): RPC {} serial number mismatch {}!={}.", fmt::format("({}): RPC {} serial number mismatch {}!={}.",
@@ -58,6 +60,7 @@ namespace OpenWifi {
std::chrono::duration<double, std::milli> rpc_execution_time = std::chrono::duration<double, std::milli> rpc_execution_time =
std::chrono::high_resolution_clock::now() - std::chrono::high_resolution_clock::now() -
RPC->second.submitted; RPC->second.submitted;
// std::cout << __LINE__ << std::endl;
poco_debug(Logger(), poco_debug(Logger(),
fmt::format("({}): Received RPC answer {}. Command={}", fmt::format("({}): Received RPC answer {}. Command={}",
SerialNumberStr, ID, SerialNumberStr, ID,
@@ -137,6 +140,7 @@ namespace OpenWifi {
} }
} }
} else { } else {
// std::cout << __LINE__ << std::endl;
} }
Command.State = 0; Command.State = 0;
@@ -159,6 +163,7 @@ namespace OpenWifi {
if (Command.rpc_entry) { if (Command.rpc_entry) {
TmpRpcEntry = Command.rpc_entry; TmpRpcEntry = Command.rpc_entry;
} }
// std::cout << __LINE__ << " State=" << Command.State << std::endl;
if (Command.State == 2) { if (Command.State == 2) {
// look at the payload to see if we should continue or not... // look at the payload to see if we should continue or not...
if (Payload->has("result")) { if (Payload->has("result")) {
@@ -168,10 +173,12 @@ namespace OpenWifi {
std::uint64_t Error = Status->get("error"); std::uint64_t Error = Status->get("error");
if (Error == 0) { if (Error == 0) {
// std::cout << __LINE__ << std::endl;
StorageService()->CommandCompleted(Command.UUID, Payload, StorageService()->CommandCompleted(Command.UUID, Payload,
rpc_execution_time, true); rpc_execution_time, true);
Command.State = 1; Command.State = 1;
} else { } else {
// std::cout << __LINE__ << std::endl;
StorageService()->CommandCompleted(Command.UUID, Payload, StorageService()->CommandCompleted(Command.UUID, Payload,
rpc_execution_time, true); rpc_execution_time, true);
std::string ErrorTxt = Status->get("result"); std::string ErrorTxt = Status->get("result");
@@ -179,11 +186,14 @@ namespace OpenWifi {
Command.State = 0; Command.State = 0;
} }
} else { } else {
// std::cout << __LINE__ << std::endl;
} }
} else { } else {
// std::cout << __LINE__ << std::endl;
Command.State = 0; Command.State = 0;
} }
} else if (Command.State == 1) { } else if (Command.State == 1) {
// std::cout << "Completing script 2 phase commit." << std::endl;
StorageService()->CommandCompleted(Command.UUID, Payload, rpc_execution_time, true); StorageService()->CommandCompleted(Command.UUID, Payload, rpc_execution_time, true);
if (Command.Deferred) { if (Command.Deferred) {
Reply = false; Reply = false;
@@ -192,6 +202,7 @@ namespace OpenWifi {
} }
if (Command.State == 0) { if (Command.State == 0) {
// std::cout << __LINE__ << " State=" << Command.State << std::endl;
OutStandingRequests_.erase(Command.Id); OutStandingRequests_.erase(Command.Id);
} }
if (Reply && TmpRpcEntry != nullptr) if (Reply && TmpRpcEntry != nullptr)
@@ -251,6 +262,8 @@ namespace OpenWifi {
for (auto request = OutStandingRequests_.begin(); request != OutStandingRequests_.end();) { for (auto request = OutStandingRequests_.begin(); request != OutStandingRequests_.end();) {
std::chrono::duration<double, std::milli> delta = now - request->second.submitted; std::chrono::duration<double, std::milli> delta = now - request->second.submitted;
if (delta > 10min) { if (delta > 10min) {
// std::cout << __LINE__ << " -->> " << request->second.Id <<
// std::endl;
MyLogger.debug(fmt::format("{}: Command={} for {} Timed out.", request->second.UUID, MyLogger.debug(fmt::format("{}: Command={} for {} Timed out.", request->second.UUID,
APCommands::to_string(request->second.Command), APCommands::to_string(request->second.Command),
Utils::IntToSerialNumber(request->second.SerialNumber))); Utils::IntToSerialNumber(request->second.SerialNumber)));
@@ -262,6 +275,8 @@ namespace OpenWifi {
StorageService()->SetCommandTimedOut(request->second.UUID); StorageService()->SetCommandTimedOut(request->second.UUID);
request = OutStandingRequests_.erase(request); request = OutStandingRequests_.erase(request);
} else { } else {
// std::cout << __LINE__ << " -->> " << request->second.Id <<
// std::endl;
++request; ++request;
} }
} }

View File

@@ -11,12 +11,12 @@
namespace OpenWifi { namespace OpenWifi {
class ConfigurationCache { class ConfigurationCache {
public: public:
static auto instance() { static ConfigurationCache &instance() {
static auto instance = new ConfigurationCache; static ConfigurationCache instance;
return instance; return instance;
} }
inline uint64_t GetCurrentConfig(std::uint64_t SerialNumber) { inline uint64_t CurrentConfig(uint64_t SerialNumber) {
std::lock_guard G(Mutex_); std::lock_guard G(Mutex_);
const auto Hint = Cache_.find(SerialNumber); const auto Hint = Cache_.find(SerialNumber);
if (Hint == end(Cache_)) if (Hint == end(Cache_))
@@ -24,25 +24,25 @@ namespace OpenWifi {
return Hint->second; return Hint->second;
} }
inline void SetCurrentConfig(std::uint64_t SerialNumber, uint64_t Id) { inline void Add(uint64_t SerialNumber, uint64_t Id) {
std::lock_guard G(Mutex_); std::lock_guard G(Mutex_);
Cache_[SerialNumber] = Id; Cache_[SerialNumber] = Id;
} }
private: private:
std::mutex Mutex_; std::recursive_mutex Mutex_;
std::map<uint64_t, uint64_t> Cache_; std::map<uint64_t, uint64_t> Cache_;
}; };
inline auto GetCurrentConfigurationID(std::uint64_t SerialNumber) { inline uint64_t GetCurrentConfigurationID(uint64_t SerialNumber) {
return ConfigurationCache::instance()->GetCurrentConfig(SerialNumber); return ConfigurationCache::instance().CurrentConfig(SerialNumber);
} }
inline void SetCurrentConfigurationID(const std::string &SerialNumber, std::uint64_t ID) { inline void SetCurrentConfigurationID(const std::string &SerialNumber, uint64_t ID) {
return ConfigurationCache::instance()->SetCurrentConfig(Utils::SerialNumberToInt(SerialNumber), ID); return ConfigurationCache::instance().Add(Utils::SerialNumberToInt(SerialNumber), ID);
} }
inline void SetCurrentConfigurationID(uint64_t SerialNumber, std::uint64_t ID) { inline void SetCurrentConfigurationID(uint64_t SerialNumber, uint64_t ID) {
return ConfigurationCache::instance()->SetCurrentConfig(SerialNumber, ID); return ConfigurationCache::instance().Add(SerialNumber, ID);
} }
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -49,7 +49,7 @@ namespace OpenWifi {
SignatureManager(), AP_WS_Server(), SignatureManager(), AP_WS_Server(),
RegulatoryInfo(), RegulatoryInfo(),
RADIUSSessionTracker(), RADIUSSessionTracker(),
AP_WS_ConfigAutoUpgradeAgent(), AP_WS_ConfigAutoUpgrader(),
FirmwareRevisionCache() FirmwareRevisionCache()
}); });
return &instance; return &instance;
@@ -78,7 +78,7 @@ namespace OpenWifi {
if (Id == DeviceType) if (Id == DeviceType)
return Type; return Type;
} }
return Platforms::AP; return "AP";
} }
void DaemonPostInitialization(Poco::Util::Application &self) { void DaemonPostInitialization(Poco::Util::Application &self) {

View File

@@ -21,6 +21,7 @@ namespace OpenWifi {
void DeviceDashboard::Generate(GWObjects::Dashboard &D, Poco::Logger &Logger) { void DeviceDashboard::Generate(GWObjects::Dashboard &D, Poco::Logger &Logger) {
if (GeneratingDashboard_.load()) { if (GeneratingDashboard_.load()) {
// std::cout << "Trying to generate dashboard but already being generated" << std::endl;
while (GeneratingDashboard_.load()) { while (GeneratingDashboard_.load()) {
Poco::Thread::trySleep(100); Poco::Thread::trySleep(100);
} }
@@ -30,6 +31,7 @@ namespace OpenWifi {
GeneratingDashboard_ = true; GeneratingDashboard_ = true;
ValidDashboard_ = false; ValidDashboard_ = false;
try { try {
// std::cout << "Generating dashboard." << std::endl;
poco_information(Logger, "DASHBOARD: Generating a new dashboard."); poco_information(Logger, "DASHBOARD: Generating a new dashboard.");
GWObjects::Dashboard NewData; GWObjects::Dashboard NewData;
StorageService()->AnalyzeCommands(NewData.commands); StorageService()->AnalyzeCommands(NewData.commands);

View File

@@ -8,12 +8,12 @@
#pragma once #pragma once
#include <Poco/Net/HTTPRequestHandler.h> #include "Poco/Net/HTTPRequestHandler.h"
#include <Poco/Net/HTTPRequestHandlerFactory.h> #include "Poco/Net/HTTPRequestHandlerFactory.h"
#include <Poco/Net/HTTPServer.h> #include "Poco/Net/HTTPServer.h"
#include <Poco/Net/HTTPServerRequest.h> #include "Poco/Net/HTTPServerRequest.h"
#include <framework/SubSystemServer.h> #include "framework/SubSystemServer.h"
namespace OpenWifi { namespace OpenWifi {

View File

@@ -50,22 +50,12 @@ namespace OpenWifi {
class DeviceConfigurationChangeKafkaEvent : public GWKafkaEvents { class DeviceConfigurationChangeKafkaEvent : public GWKafkaEvents {
public: public:
DeviceConfigurationChangeKafkaEvent(std::uint64_t serialNumber, DeviceConfigurationChangeKafkaEvent(std::uint64_t serialNumber,
std::uint64_t timestamp, std::uint64_t timestamp, const Poco::JSON::Object::Ptr config)
const Poco::JSON::Object::Ptr config)
: GWKafkaEvents(serialNumber, "unit.configuration_change", timestamp), config_(config) { : GWKafkaEvents(serialNumber, "unit.configuration_change", timestamp), config_(config) {
} }
~DeviceConfigurationChangeKafkaEvent() { ~DeviceConfigurationChangeKafkaEvent() {
if(config_!= nullptr) {
std::ostringstream os;
config_->stringify(os);
if(os.str().size()> KafkaManager()->KafkaManagerMaximumPayloadSize()) {
payload_->set("configuration", "{}");
payload_->set("configurationTooBig", true);
} else {
payload_->set("configuration", *config_); payload_->set("configuration", *config_);
}
}
Send(); Send();
} }

View File

@@ -1753,6 +1753,7 @@ namespace OpenWifi {
nlohmann::json new_ie; nlohmann::json new_ie;
nlohmann::json content; nlohmann::json content;
// std::cout << BufferToHex(&data[0],data.size()) << std::endl;
uint offset = 0; uint offset = 0;
auto sub_ie = data[offset++]; auto sub_ie = data[offset++];
switch (sub_ie) { switch (sub_ie) {
@@ -1787,6 +1788,7 @@ namespace OpenWifi {
try { try {
nlohmann::json D = nlohmann::json::parse(ofs.str()); nlohmann::json D = nlohmann::json::parse(ofs.str());
// std::cout << "Start of parsing wifi" << std::endl;
if (D.contains("status")) { if (D.contains("status")) {
auto Status = D["status"]; auto Status = D["status"];
if (Status.contains("scan") && Status["scan"].is_array()) { if (Status.contains("scan") && Status["scan"].is_array()) {
@@ -1801,6 +1803,8 @@ namespace OpenWifi {
if (ie.contains("type") && ie.contains("data")) { if (ie.contains("type") && ie.contains("data")) {
uint64_t ie_type = ie["type"]; uint64_t ie_type = ie["type"];
std::string ie_data = ie["data"]; std::string ie_data = ie["data"];
// std::cout << "TYPE:" << ie_type << " DATA:" << ie_data
// << std::endl;
auto data = Base64Decode2Vec(ie_data); auto data = Base64Decode2Vec(ie_data);
if (ie_type == ieee80211_eid::WLAN_EID_COUNTRY) { if (ie_type == ieee80211_eid::WLAN_EID_COUNTRY) {
new_ies.push_back(WFS_WLAN_EID_COUNTRY(data)); new_ies.push_back(WFS_WLAN_EID_COUNTRY(data));
@@ -1854,12 +1858,18 @@ namespace OpenWifi {
} else if (ie_type == ieee80211_eid::WLAN_EID_EXTENSION) { } else if (ie_type == ieee80211_eid::WLAN_EID_EXTENSION) {
new_ies.push_back(WFS_WLAN_EID_EXTENSION(data)); new_ies.push_back(WFS_WLAN_EID_EXTENSION(data));
} else { } else {
// std::cout
// << "Skipping IE: no parsing available: " << ie_type
// << std::endl;
new_ies.push_back(ie); new_ies.push_back(ie);
} }
} else { } else {
// std::cout << "Skipping IE: no data and type" <<
// std::endl;
new_ies.push_back(ie); new_ies.push_back(ie);
} }
} catch (...) { } catch (...) {
// std::cout << "Skipping IE: exception" << std::endl;
Logger.information(fmt::format("Error parsing IEs")); Logger.information(fmt::format("Error parsing IEs"));
new_ies.push_back(ie); new_ies.push_back(ie);
} }
@@ -1867,6 +1877,7 @@ namespace OpenWifi {
scan_entry["ies"] = new_ies; scan_entry["ies"] = new_ies;
ParsedScan.push_back(scan_entry); ParsedScan.push_back(scan_entry);
} else { } else {
// std::cout << "Skipping scan" << std::endl;
ParsedScan.push_back(scan_entry); ParsedScan.push_back(scan_entry);
} }
} }
@@ -1875,6 +1886,7 @@ namespace OpenWifi {
} }
} }
Result << to_string(D); Result << to_string(D);
// std::cout << "End of parsing wifi" << std::endl;
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger.log(E); Logger.log(E);

View File

@@ -177,6 +177,15 @@ namespace OpenWifi {
} else { } else {
session_hint->second->lastTransaction = Utils::Now(); session_hint->second->lastTransaction = Utils::Now();
} }
/*
if(ap_hint!=AccountingSessions_.end()) {
std::cout << "Auth table:" << std::endl;
for(const auto &session:ap_hint->second) {
std::cout << Notification.SerialNumber_ << ": Index: " << session.first << ": ID: " << session.second->accountingSessionId << " MID:" << session.second->accountingMultiSessionId << std::endl;
}
}
*/
} }
std::uint32_t GetUiInt32(const std::uint8_t *buf) { std::uint32_t GetUiInt32(const std::uint8_t *buf) {
@@ -414,15 +423,15 @@ namespace OpenWifi {
} }
void RADIUSSessionTracker::DisconnectSession(const std::string &SerialNumber) { void RADIUSSessionTracker::DisconnectSession(const std::string &SerialNumber) {
poco_information(Logger(),fmt::format("{}: Disconnecting.", SerialNumber));
std::lock_guard Guard(Mutex_); std::lock_guard Guard(Mutex_);
auto hint = AccountingSessions_.find(SerialNumber); auto hint = AccountingSessions_.find(SerialNumber);
if(hint==end(AccountingSessions_)) { if(hint==end(AccountingSessions_)) {
return; return;
} }
poco_information(Logger(),fmt::format("{}: Disconnecting.", SerialNumber));
// we need to go through all sessions and send an accounting stop // we need to go through all sessions and send an accounting stop
for(const auto &session:hint->second) { for(const auto &session:hint->second) {
poco_debug(Logger(), fmt::format("Stopping accounting for {}:{}", SerialNumber, session.first )); poco_debug(Logger(), fmt::format("Stopping accounting for {}:{}", SerialNumber, session.first ));

View File

@@ -75,10 +75,8 @@ namespace OpenWifi {
} else if ((Utils::Now() - LastKeepAlive) > Pool_.radsecKeepAlive) { } else if ((Utils::Now() - LastKeepAlive) > Pool_.radsecKeepAlive) {
RADIUS::RadiusOutputPacket P(Pool_.authConfig.servers[ServerIndex_].radsecSecret); RADIUS::RadiusOutputPacket P(Pool_.authConfig.servers[ServerIndex_].radsecSecret);
P.MakeStatusMessage(Pool_.authConfig.servers[ServerIndex_].name); P.MakeStatusMessage(Pool_.authConfig.servers[ServerIndex_].name);
if(Type_!=GWObjects::RadiusEndpointType::generic) {
poco_trace(Logger_, fmt::format("{}: Keep-Alive message.", Pool_.authConfig.servers[ServerIndex_].name)); poco_trace(Logger_, fmt::format("{}: Keep-Alive message.", Pool_.authConfig.servers[ServerIndex_].name));
Socket_->sendBytes(P.Data(), P.Len()); Socket_->sendBytes(P.Data(), P.Len());
}
LastKeepAlive = Utils::Now(); LastKeepAlive = Utils::Now();
} }
Poco::Thread::trySleep(2000); Poco::Thread::trySleep(2000);
@@ -165,16 +163,17 @@ namespace OpenWifi {
fmt::format("Unknown packet: Type: {} (type={}) Length={}", fmt::format("Unknown packet: Type: {} (type={}) Length={}",
P.PacketType(), P.PacketTypeInt(), P.BufferLen())); P.PacketType(), P.PacketTypeInt(), P.BufferLen()));
} }
return;
} else { } else {
poco_warning(Logger_, "Invalid packet received. Resetting the connection."); poco_warning(Logger_, "Invalid packet received. Resetting the connection.");
Disconnect();
} }
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger_.log(E); Logger_.log(E);
Disconnect();
} catch (...) { } catch (...) {
Disconnect();
poco_warning(Logger_, "Exception occurred. Resetting the connection."); poco_warning(Logger_, "Exception occurred. Resetting the connection.");
} }
Disconnect();
} }
inline void inline void
@@ -325,7 +324,7 @@ namespace OpenWifi {
ofs << OpenRoamingRootCert; ofs << OpenRoamingRootCert;
ofs.close(); ofs.close();
auto SecureContext = Poco::AutoPtr<Poco::Net::Context>( Poco::Net::Context::Ptr SecureContext = Poco::AutoPtr<Poco::Net::Context>(
new Poco::Net::Context(Poco::Net::Context::TLS_CLIENT_USE, "")); new Poco::Net::Context(Poco::Net::Context::TLS_CLIENT_USE, ""));
if (Pool_.acctConfig.servers[ServerIndex_].allowSelfSigned) { if (Pool_.acctConfig.servers[ServerIndex_].allowSelfSigned) {
@@ -431,7 +430,7 @@ namespace OpenWifi {
DecodeFile(CaCertFiles_[CaCertFiles_.size() - 1]->path(), cert); DecodeFile(CaCertFiles_[CaCertFiles_.size() - 1]->path(), cert);
} }
auto SecureContext = Poco::Net::Context::Ptr SecureContext =
Poco::AutoPtr<Poco::Net::Context>(new Poco::Net::Context( Poco::AutoPtr<Poco::Net::Context>(new Poco::Net::Context(
Poco::Net::Context::TLS_CLIENT_USE, KeyFile_.path(), CertFile_.path(), "")); Poco::Net::Context::TLS_CLIENT_USE, KeyFile_.path(), CertFile_.path(), ""));
if (Pool_.acctConfig.servers[ServerIndex_].allowSelfSigned) { if (Pool_.acctConfig.servers[ServerIndex_].allowSelfSigned) {
@@ -499,9 +498,7 @@ namespace OpenWifi {
} }
inline bool Connect_Generic() { inline bool Connect_Generic() {
poco_information(Logger_, fmt::format("Connecting {}", Pool_.name)); if (TryAgain_) {
if (TryAgain_ && !Connected_) {
std::lock_guard G(LocalMutex_); std::lock_guard G(LocalMutex_);
Poco::Net::SocketAddress AuthSockAddrV4( Poco::Net::SocketAddress AuthSockAddrV4(
@@ -523,6 +520,26 @@ namespace OpenWifi {
MicroServiceConfigGetInt("radius.proxy.coa.port", DEFAULT_RADIUS_CoA_PORT)); MicroServiceConfigGetInt("radius.proxy.coa.port", DEFAULT_RADIUS_CoA_PORT));
CoASocketV4_ = std::make_unique<Poco::Net::DatagramSocket>(CoASockAddrV4, true, true); CoASocketV4_ = std::make_unique<Poco::Net::DatagramSocket>(CoASockAddrV4, true, true);
/*
AuthenticationSocketV6_ =
std::make_unique<Poco::Net::DatagramSocket>(AuthSockAddrV6, true, true);
Poco::Net::SocketAddress AuthSockAddrV6(
Poco::Net::AddressFamily::IPv6,
MicroServiceConfigGetInt("radius.proxy.authentication.port",
DEFAULT_RADIUS_AUTHENTICATION_PORT));
Poco::Net::SocketAddress AcctSockAddrV6(
Poco::Net::AddressFamily::IPv6,
MicroServiceConfigGetInt("radius.proxy.accounting.port",
DEFAULT_RADIUS_ACCOUNTING_PORT));
AccountingSocketV6_ =
std::make_unique<Poco::Net::DatagramSocket>(AcctSockAddrV6, true, true);
Poco::Net::SocketAddress CoASockAddrV6(
Poco::Net::AddressFamily::IPv6,
MicroServiceConfigGetInt("radius.proxy.coa.port", DEFAULT_RADIUS_CoA_PORT));
CoASocketV6_ = std::make_unique<Poco::Net::DatagramSocket>(CoASockAddrV6, true, true);
*/
Reactor_.addEventHandler( Reactor_.addEventHandler(
*AuthenticationSocketV4_, *AuthenticationSocketV4_,
Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>( Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>(
@@ -534,27 +551,7 @@ namespace OpenWifi {
Reactor_.addEventHandler( Reactor_.addEventHandler(
*CoASocketV4_, Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>( *CoASocketV4_, Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>(
*this, &RADIUS_Destination::OnCoASocketReadable)); *this, &RADIUS_Destination::OnCoASocketReadable));
/* /*
Poco::Net::SocketAddress AuthSockAddrV6(
Poco::Net::AddressFamily::IPv6,
MicroServiceConfigGetInt("radius.proxy.authentication.port",
DEFAULT_RADIUS_AUTHENTICATION_PORT));
AuthenticationSocketV6_ =
std::make_unique<Poco::Net::DatagramSocket>(AuthSockAddrV6, true, true);
Poco::Net::SocketAddress AcctSockAddrV6(
Poco::Net::AddressFamily::IPv6,
MicroServiceConfigGetInt("radius.proxy.accounting.port",
DEFAULT_RADIUS_AUTHENTICATION_PORT));
AccountingSocketV6_ =
std::make_unique<Poco::Net::DatagramSocket>(AcctSockAddrV6, true, true);
Poco::Net::SocketAddress CoASockAddrV6(
Poco::Net::AddressFamily::IPv6,
MicroServiceConfigGetInt("radius.proxy.coa.port", DEFAULT_RADIUS_CoA_PORT));
CoASocketV6_ = std::make_unique<Poco::Net::DatagramSocket>(CoASockAddrV6, true, true);
Reactor_.addEventHandler( Reactor_.addEventHandler(
*AuthenticationSocketV6_, *AuthenticationSocketV6_,
Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>( Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>(
@@ -568,7 +565,6 @@ namespace OpenWifi {
*CoASocketV6_, Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>( *CoASocketV6_, Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>(
*this, &RADIUS_Destination::OnCoASocketReadable)); *this, &RADIUS_Destination::OnCoASocketReadable));
*/ */
Connected_ = true;
} }
return true; return true;
} }
@@ -594,8 +590,6 @@ namespace OpenWifi {
if (Connected_) { if (Connected_) {
std::lock_guard G(LocalMutex_); std::lock_guard G(LocalMutex_);
if(Type_==GWObjects::RadiusEndpointType::generic) { if(Type_==GWObjects::RadiusEndpointType::generic) {
poco_information(Logger_, fmt::format("Disconnecting {} generic server. Releasing all UDP resources.", Pool_.name));
if(AuthenticationSocketV4_) { if(AuthenticationSocketV4_) {
Reactor_.removeEventHandler( Reactor_.removeEventHandler(
*AuthenticationSocketV4_, *AuthenticationSocketV4_,
@@ -623,8 +617,7 @@ namespace OpenWifi {
CoASocketV4_.reset(); CoASocketV4_.reset();
} }
/* /* if(AuthenticationSocketV6_) {
if(AuthenticationSocketV6_) {
Reactor_.removeEventHandler( Reactor_.removeEventHandler(
*AuthenticationSocketV6_, *AuthenticationSocketV6_,
Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>( Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>(
@@ -653,6 +646,7 @@ namespace OpenWifi {
*/ */
} else { } else {
if(Socket_!=nullptr) { if(Socket_!=nullptr) {
std::lock_guard G(LocalMutex_);
Reactor_.removeEventHandler( Reactor_.removeEventHandler(
*Socket_, Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>( *Socket_, Poco::NObserver<RADIUS_Destination, Poco::Net::ReadableNotification>(
*this, &RADIUS_Destination::onData)); *this, &RADIUS_Destination::onData));
@@ -666,9 +660,9 @@ namespace OpenWifi {
Socket_->close(); Socket_->close();
} }
} }
Connected_ = false;
} }
poco_information(Logger_, fmt::format("Disconnecting {}", Pool_.name)); Connected_ = false;
poco_information(Logger_, "Disconnecting.");
} }
static void DecodeFile(const std::string &filename, const std::string &s) { static void DecodeFile(const std::string &filename, const std::string &s) {
@@ -730,8 +724,7 @@ namespace OpenWifi {
std::unique_ptr<Poco::Net::DatagramSocket> AuthenticationSocketV4_; std::unique_ptr<Poco::Net::DatagramSocket> AuthenticationSocketV4_;
std::unique_ptr<Poco::Net::DatagramSocket> CoASocketV4_; std::unique_ptr<Poco::Net::DatagramSocket> CoASocketV4_;
/* /* std::unique_ptr<Poco::Net::DatagramSocket> CoASocketV6_;
std::unique_ptr<Poco::Net::DatagramSocket> CoASocketV6_;
std::unique_ptr<Poco::Net::DatagramSocket> AccountingSocketV6_; std::unique_ptr<Poco::Net::DatagramSocket> AccountingSocketV6_;
std::unique_ptr<Poco::Net::DatagramSocket> AuthenticationSocketV6_; std::unique_ptr<Poco::Net::DatagramSocket> AuthenticationSocketV6_;
*/ */

View File

@@ -25,24 +25,10 @@ namespace OpenWifi::RESTAPI_RPC {
if (StorageService()->AddCommand(Cmd.SerialNumber, Cmd, Status)) { if (StorageService()->AddCommand(Cmd.SerialNumber, Cmd, Status)) {
Poco::JSON::Object RetObj; Poco::JSON::Object RetObj;
Cmd.to_json(RetObj); Cmd.to_json(RetObj);
if (Handler == nullptr) { if (Handler != nullptr)
// nothing to process/return return Handler->ReturnObject(RetObj);
return; return;
} }
Poco::Net::HTTPResponse::HTTPStatus cmd_status = Poco::Net::HTTPResponse::HTTP_OK;
if (Cmd.ErrorCode > 0) {
// command returned error
cmd_status = Poco::Net::HTTPResponse::HTTP_BAD_REQUEST;
if (Cmd.Command == uCentralProtocol::CONFIGURE) {
// special handling for configure command
if (!Handler->GetBoolParameter("strict", false)) {
// in non-strict mode return success for failed configure command
cmd_status = Poco::Net::HTTPResponse::HTTP_OK;
}
}
}
return Handler->ReturnObject(RetObj, cmd_status);
}
if (Handler != nullptr) if (Handler != nullptr)
return Handler->ReturnStatus(Poco::Net::HTTPResponse::HTTP_INTERNAL_SERVER_ERROR); return Handler->ReturnStatus(Poco::Net::HTTPResponse::HTTP_INTERNAL_SERVER_ERROR);
} }
@@ -54,8 +40,8 @@ namespace OpenWifi::RESTAPI_RPC {
std::chrono::milliseconds WaitTimeInMs, Poco::JSON::Object *ObjectToReturn, std::chrono::milliseconds WaitTimeInMs, Poco::JSON::Object *ObjectToReturn,
RESTAPIHandler *Handler, Poco::Logger &Logger, bool Deferred) { RESTAPIHandler *Handler, Poco::Logger &Logger, bool Deferred) {
Logger.information(fmt::format("{},{}: New {} command. User={} Serial={} Details={}. ", Cmd.UUID, Logger.information(fmt::format("{},{}: New {} command. User={} Serial={}. ", Cmd.UUID,
RPCID, Cmd.Command, Cmd.SubmittedBy, Cmd.SerialNumber, Cmd.Details)); RPCID, Cmd.Command, Cmd.SubmittedBy, Cmd.SerialNumber));
Cmd.Submitted = Utils::Now(); Cmd.Submitted = Utils::Now();
Cmd.Executed = 0; Cmd.Executed = 0;
@@ -181,20 +167,6 @@ namespace OpenWifi::RESTAPI_RPC {
Cmd.AttachType = ""; Cmd.AttachType = "";
} }
// If the command fails on the device we should show it as failed and not return 200 OK
// exception is configure command which only reported failed in strict validation mode
if (Cmd.ErrorCode &&
(Cmd.Command != uCentralProtocol::CONFIGURE ||
(Cmd.Command == uCentralProtocol::CONFIGURE && Handler->GetBoolParameter("strict", false))
))
{
Logger.information(fmt::format(
"Command failed with error on device: {} Reason: {}.",
Cmd.ErrorCode, Cmd.ErrorText));
return SetCommandStatus(Cmd, Request, Response, Handler,
Storage::CommandExecutionType::COMMAND_FAILED, Logger);
}
if (Cmd.ErrorCode == 0 && Cmd.Command == uCentralProtocol::CONFIGURE) { if (Cmd.ErrorCode == 0 && Cmd.Command == uCentralProtocol::CONFIGURE) {
// we need to post a kafka event for this. // we need to post a kafka event for this.
if (Params.has(uCentralProtocol::CONFIG) && Params.isObject(uCentralProtocol::CONFIG)) { if (Params.has(uCentralProtocol::CONFIG) && Params.isObject(uCentralProtocol::CONFIG)) {
@@ -203,7 +175,6 @@ namespace OpenWifi::RESTAPI_RPC {
DeviceConfigurationChangeKafkaEvent KEvent( DeviceConfigurationChangeKafkaEvent KEvent(
Utils::SerialNumberToInt(Cmd.SerialNumber), Utils::Now(), Utils::SerialNumberToInt(Cmd.SerialNumber), Utils::Now(),
Config); Config);
} }
} }

View File

@@ -63,7 +63,7 @@ namespace OpenWifi {
poco_debug(Logger(), fmt::format("BLACKLIST-POST: {}", D.serialNumber)); poco_debug(Logger(), fmt::format("BLACKLIST-POST: {}", D.serialNumber));
Poco::toLowerInPlace(D.serialNumber); Poco::toLowerInPlace(D.serialNumber);
if (StorageService()->IsBlackListed(Utils::MACToInt(D.serialNumber))) { if (StorageService()->IsBlackListed(D.serialNumber)) {
return BadRequest(RESTAPI::Errors::SerialNumberExists); return BadRequest(RESTAPI::Errors::SerialNumberExists);
} }

View File

@@ -56,27 +56,17 @@ namespace OpenWifi {
return BadRequest(RESTAPI::Errors::InvalidJSONDocument); return BadRequest(RESTAPI::Errors::InvalidJSONDocument);
} }
if (DefConfig.models.empty()) { if (DefConfig.Models.empty()) {
return BadRequest(RESTAPI::Errors::ModelIDListCannotBeEmpty); return BadRequest(RESTAPI::Errors::ModelIDListCannotBeEmpty);
} }
DefConfig.platform = DefConfig.platform.empty() ? Platforms::AP : DefConfig.platform; std::vector<std::string> Error;
if(DefConfig.platform != Platforms::AP && DefConfig.platform != Platforms::SWITCH) { if (!ValidateUCentralConfiguration(DefConfig.Configuration, Error,
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
if(DefConfig.configuration.empty()) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
std::string Error;
if (!ValidateUCentralConfiguration(ConfigurationValidator::GetType(DefConfig.platform),
DefConfig.configuration, Error,
GetBoolParameter("strict", false))) { GetBoolParameter("strict", false))) {
return BadRequest(RESTAPI::Errors::ConfigBlockInvalid, Error); return BadRequest(RESTAPI::Errors::ConfigBlockInvalid);
} }
DefConfig.created = DefConfig.lastModified = Utils::Now(); DefConfig.Created = DefConfig.LastModified = Utils::Now();
if (StorageService()->CreateDefaultConfiguration(Name, DefConfig)) { if (StorageService()->CreateDefaultConfiguration(Name, DefConfig)) {
return OK(); return OK();
} }
@@ -98,31 +88,19 @@ namespace OpenWifi {
return NotFound(); return NotFound();
} }
if(Existing.platform.empty()) { if (!NewConfig.Configuration.empty()) {
Existing.platform = Platforms::AP; std::vector<std::string> Error;
} if (!ValidateUCentralConfiguration(NewConfig.Configuration, Error,
if(ParsedBody_->has("platform")) {
if(NewConfig.platform.empty() || (NewConfig.platform != Platforms::AP && NewConfig.platform != Platforms::SWITCH)) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
Existing.platform = NewConfig.platform;
}
if (!NewConfig.configuration.empty()) {
std::string Error;
if (!ValidateUCentralConfiguration(ConfigurationValidator::GetType(Existing.platform),
NewConfig.configuration, Error,
GetBoolParameter("strict", false))) { GetBoolParameter("strict", false))) {
return BadRequest(RESTAPI::Errors::ConfigBlockInvalid, Error); return BadRequest(RESTAPI::Errors::ConfigBlockInvalid);
} }
Existing.configuration = NewConfig.configuration; Existing.Configuration = NewConfig.Configuration;
} }
Existing.lastModified = Utils::Now(); Existing.LastModified = Utils::Now();
AssignIfPresent(Obj, "description", Existing.description); AssignIfPresent(Obj, "description", Existing.Description);
if (Obj->has("modelIds")) if (Obj->has("modelIds"))
Existing.models = NewConfig.models; Existing.Models = NewConfig.Models;
if (StorageService()->UpdateDefaultConfiguration(Name, Existing)) { if (StorageService()->UpdateDefaultConfiguration(Name, Existing)) {
GWObjects::DefaultConfiguration ModifiedConfig; GWObjects::DefaultConfiguration ModifiedConfig;

View File

@@ -87,7 +87,7 @@ namespace OpenWifi {
poco_debug( poco_debug(
Logger_, Logger_,
fmt::format( fmt::format(
"Command RTTY TID={} can proceed. Identified as {} and RPCID as {}. thr_id={}", "Command rtty TID={} can proceed. Identified as {} and RPCID as {}. thr_id={}",
TransactionId_, UUID, RPC, Poco::Thread::current()->id())); TransactionId_, UUID, RPC, Poco::Thread::current()->id()));
return Rtty(UUID, RPC, 60000ms, Restrictions); return Rtty(UUID, RPC, 60000ms, Restrictions);
}; };
@@ -166,12 +166,7 @@ namespace OpenWifi {
{APCommands::Commands::rrm, false, true, &RESTAPI_device_commandHandler::RRM, 60000ms}, {APCommands::Commands::rrm, false, true, &RESTAPI_device_commandHandler::RRM, 60000ms},
{APCommands::Commands::certupdate, false, true, &RESTAPI_device_commandHandler::CertUpdate, 60000ms}, {APCommands::Commands::certupdate, false, true, &RESTAPI_device_commandHandler::CertUpdate, 60000ms},
{APCommands::Commands::transfer, false, true, &RESTAPI_device_commandHandler::Transfer, 60000ms}, {APCommands::Commands::transfer, false, true, &RESTAPI_device_commandHandler::Transfer, 60000ms},
{APCommands::Commands::script, false, true, &RESTAPI_device_commandHandler::Script, 60000ms}, {APCommands::Commands::script, false, true, &RESTAPI_device_commandHandler::Script, 60000ms}
{APCommands::Commands::powercycle, false, true, &RESTAPI_device_commandHandler::PowerCycle, 60000ms},
{APCommands::Commands::fixedconfig, false, true, &RESTAPI_device_commandHandler::FixedConfig, 120000ms},
{APCommands::Commands::cablediagnostics, false, true, &RESTAPI_device_commandHandler::CableDiagnostics, 120000ms},
{APCommands::Commands::reenroll, false, true, &RESTAPI_device_commandHandler::ReEnroll, 120000ms},
}; };
void RESTAPI_device_commandHandler::DoPost() { void RESTAPI_device_commandHandler::DoPost() {
@@ -658,18 +653,13 @@ namespace OpenWifi {
return BadRequest(RESTAPI::Errors::SerialNumberMismatch); return BadRequest(RESTAPI::Errors::SerialNumberMismatch);
} }
GWObjects::Device DeviceInfo;
if (!StorageService()->GetDevice(SerialNumber_, DeviceInfo)) {
return NotFound();
}
auto Configuration = auto Configuration =
GetS(RESTAPI::Protocol::CONFIGURATION, Obj, uCentralProtocol::EMPTY_JSON_DOC); GetS(RESTAPI::Protocol::CONFIGURATION, Obj, uCentralProtocol::EMPTY_JSON_DOC);
std::string Error; std::vector<std::string> Error;
if (!ValidateUCentralConfiguration(ConfigurationValidator::GetType(DeviceInfo.DeviceType), if (!ValidateUCentralConfiguration(Configuration, Error,
Configuration, Error,
GetBoolParameter("strict", false))) { GetBoolParameter("strict", false))) {
CallCanceled("CONFIGURE", CMD_UUID, CMD_RPC, RESTAPI::Errors::ConfigBlockInvalid); CallCanceled("CONFIGURE", CMD_UUID, CMD_RPC, RESTAPI::Errors::ConfigBlockInvalid);
return BadRequest(RESTAPI::Errors::ConfigBlockInvalid, Error); return BadRequest(RESTAPI::Errors::ConfigBlockInvalid);
} }
auto When = GetWhen(Obj); auto When = GetWhen(Obj);
@@ -695,31 +685,9 @@ namespace OpenWifi {
Params.stringify(ParamStream); Params.stringify(ParamStream);
Cmd.Details = ParamStream.str(); Cmd.Details = ParamStream.str();
// retrieve capabilities and encode/compress parameters, if required
Poco::JSON::Object ConfigParams = Params;
GWObjects::Capabilities Caps;
if (StorageService()->GetDeviceCapabilities(SerialNumber_, Caps)) {
Poco::JSON::Object CapsJson;
Caps.to_json(CapsJson);
auto DeviceCaps = CapsJson.getObject(uCentralProtocol::CAPABILITIES);
if (DeviceCaps->has("compress_cmd") && DeviceCaps->get("compress_cmd")) {
// compressed command capability present and it is set, compress parameters
Poco::JSON::Object CompressedParams;
std::string CompressedBase64Data;
std::uint64_t UncompressedDataLen = ParamStream.str().length();
if (Utils::CompressAndEncodeBase64(ParamStream.str(), CompressedBase64Data)) {
// set compressed, base 64 encoded data and length of uncompressed data
CompressedParams.set(uCentralProtocol::COMPRESS_64, CompressedBase64Data);
CompressedParams.set(uCentralProtocol::COMPRESS_SZ, UncompressedDataLen);
ConfigParams = CompressedParams;
}
}
}
// AP_WS_Server()->SetPendingUUID(SerialNumber_, NewUUID); // AP_WS_Server()->SetPendingUUID(SerialNumber_, NewUUID);
RESTAPI_RPC::WaitForCommand(CMD_RPC, APCommands::Commands::configure, true, RESTAPI_RPC::WaitForCommand(CMD_RPC, APCommands::Commands::configure, true,
Cmd, ConfigParams, *Request, *Response, timeout, Cmd, Params, *Request, *Response, timeout,
nullptr, this, Logger_); nullptr, this, Logger_);
if(!Cmd.Executed) { if(!Cmd.Executed) {
@@ -1201,7 +1169,7 @@ namespace OpenWifi {
if (RTTYS_server()->UseInternal()) { if (RTTYS_server()->UseInternal()) {
std::uint64_t SN = Utils::SerialNumberToInt(SerialNumber_); std::uint64_t SN = Utils::SerialNumberToInt(SerialNumber_);
bool mTLS = AP_WS_Server()->DeviceRequiresSecureRTTY(SN); bool mTLS = AP_WS_Server()->DeviceRequiresSecureRtty(SN);
auto Hash = Utils::ComputeHash(UserInfo_.webtoken.refresh_token_, Utils::Now()); auto Hash = Utils::ComputeHash(UserInfo_.webtoken.refresh_token_, Utils::Now());
Rtty.Token = Hash.substr(0, RTTY_DEVICE_TOKEN_LENGTH); Rtty.Token = Hash.substr(0, RTTY_DEVICE_TOKEN_LENGTH);
if (!RTTYS_server()->CreateEndPoint(Rtty.ConnectionId, Rtty.Token, Requester(), if (!RTTYS_server()->CreateEndPoint(Rtty.ConnectionId, Rtty.Token, Requester(),
@@ -1433,9 +1401,9 @@ namespace OpenWifi {
Cmd.WaitingForFile = 0; Cmd.WaitingForFile = 0;
Cmd.Status= "completed"; Cmd.Status= "completed";
if(CommandManager()->FireAndForget(SerialNumber_, uCentralProtocol::RRM, Params)) { if(CommandManager()->FireAndForget(SerialNumber_, uCentralProtocol::RRM, Params)) {
Cmd.Status= "completed";
StorageService()->AddCommand(SerialNumber_, Cmd, StorageService()->AddCommand(SerialNumber_, Cmd,
Storage::CommandExecutionType::COMMAND_COMPLETED); Storage::CommandExecutionType::COMMAND_COMPLETED);
Cmd.Status= "completed";
return OK(); return OK();
} }
Cmd.Status= "failed"; // should never happen Cmd.Status= "failed"; // should never happen
@@ -1533,164 +1501,4 @@ namespace OpenWifi {
} }
void RESTAPI_device_commandHandler::PowerCycle(
const std::string &CMD_UUID, uint64_t CMD_RPC,
[[maybe_unused]] std::chrono::milliseconds timeout,
[[maybe_unused]] const GWObjects::DeviceRestrictions &Restrictions) {
if(UserInfo_.userinfo.userRole != SecurityObjects::ROOT &&
UserInfo_.userinfo.userRole != SecurityObjects::ADMIN) {
CallCanceled("RRM", CMD_UUID, CMD_RPC, RESTAPI::Errors::ACCESS_DENIED);
return UnAuthorized(RESTAPI::Errors::ACCESS_DENIED);
}
poco_debug(Logger_, fmt::format("POWERCYCLE({},{}): TID={} user={} serial={}", CMD_UUID,
CMD_RPC, TransactionId_, Requester(), SerialNumber_));
if(IsDeviceSimulated(SerialNumber_)) {
CallCanceled("RRM", CMD_UUID, CMD_RPC, RESTAPI::Errors::SimulatedDeviceNotSupported);
return BadRequest(RESTAPI::Errors::SimulatedDeviceNotSupported);
}
GWObjects::PowerCycleRequest PR;
if(!PR.from_json(ParsedBody_)) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
GWObjects::CommandDetails Cmd;
Cmd.SerialNumber = SerialNumber_;
Cmd.SubmittedBy = Requester();
Cmd.UUID = CMD_UUID;
Cmd.Command = uCentralProtocol::POWERCYCLE;
std::ostringstream os;
ParsedBody_->stringify(os);
Cmd.Details = os.str();
Cmd.RunAt = PR.when;
Cmd.ErrorCode = 0;
Cmd.WaitingForFile = 0;
return RESTAPI_RPC::WaitForCommand(CMD_RPC, APCommands::Commands::powercycle, false, Cmd,
*ParsedBody_, *Request, *Response, timeout, nullptr, this,
Logger_);
}
// `fixedconfig` command is used set country propery on AP
// This handler uses `fixedconfig` command definitions
void RESTAPI_device_commandHandler::FixedConfig(
const std::string &CMD_UUID, uint64_t CMD_RPC, std::chrono::milliseconds timeout,
[[maybe_unused]] const GWObjects::DeviceRestrictions &Restrictions) {
poco_debug(Logger_, fmt::format("FIXEDCONFIG({},{}): TID={} user={} serial={}", CMD_UUID, CMD_RPC,
TransactionId_, Requester(), SerialNumber_));
// do not allow `fixedconfig` command for simulated devices
if(IsDeviceSimulated(SerialNumber_)) {
CallCanceled("FIXEDCONFIG", CMD_UUID, CMD_RPC, RESTAPI::Errors::SimulatedDeviceNotSupported);
return BadRequest(RESTAPI::Errors::SimulatedDeviceNotSupported);
}
// setup and validate fixedconfig object
GWObjects::FixedConfig fixed_config;
if(!fixed_config.from_json(ParsedBody_)) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
// setup command message
GWObjects::CommandDetails Cmd;
Cmd.SerialNumber = SerialNumber_;
Cmd.SubmittedBy = Requester();
Cmd.UUID = CMD_UUID;
Cmd.Command = uCentralProtocol::FIXEDCONFIG;
std::ostringstream os;
ParsedBody_->stringify(os);
Cmd.Details = os.str();
Cmd.RunAt = 0;
Cmd.ErrorCode = 0;
Cmd.WaitingForFile = 0;
// send fixedconfig command to device and return status
return RESTAPI_RPC::WaitForCommand(CMD_RPC, APCommands::Commands::fixedconfig, false, Cmd,
*ParsedBody_, *Request, *Response, timeout, nullptr, this,
Logger_);
}
void RESTAPI_device_commandHandler::CableDiagnostics(
const std::string &CMD_UUID, uint64_t CMD_RPC,
[[maybe_unused]] std::chrono::milliseconds timeout,
[[maybe_unused]] const GWObjects::DeviceRestrictions &Restrictions) {
if(UserInfo_.userinfo.userRole != SecurityObjects::ROOT &&
UserInfo_.userinfo.userRole != SecurityObjects::ADMIN) {
CallCanceled("CABLEDIAGNOSTICS", CMD_UUID, CMD_RPC, RESTAPI::Errors::ACCESS_DENIED);
return UnAuthorized(RESTAPI::Errors::ACCESS_DENIED);
}
poco_debug(Logger_, fmt::format("CABLEDIAGNOSTICS({},{}): TID={} user={} serial={}", CMD_UUID,
CMD_RPC, TransactionId_, Requester(), SerialNumber_));
if(IsDeviceSimulated(SerialNumber_)) {
CallCanceled("CABLEDIAGNOSTICS", CMD_UUID, CMD_RPC, RESTAPI::Errors::SimulatedDeviceNotSupported);
return BadRequest(RESTAPI::Errors::SimulatedDeviceNotSupported);
}
GWObjects::CableDiagnostics PR;
if(!PR.from_json(ParsedBody_)) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
GWObjects::CommandDetails Cmd;
Cmd.SerialNumber = SerialNumber_;
Cmd.SubmittedBy = Requester();
Cmd.UUID = CMD_UUID;
Cmd.Command = uCentralProtocol::CABLEDIAGNOSTICS;
std::ostringstream os;
ParsedBody_->stringify(os);
Cmd.Details = os.str();
Cmd.RunAt = PR.when;
Cmd.ErrorCode = 0;
Cmd.WaitingForFile = 0;
return RESTAPI_RPC::WaitForCommand(CMD_RPC, APCommands::Commands::cablediagnostics, false, Cmd,
*ParsedBody_, *Request, *Response, timeout, nullptr, this,
Logger_);
}
void RESTAPI_device_commandHandler::ReEnroll(
const std::string &CMD_UUID, uint64_t CMD_RPC,
[[maybe_unused]] std::chrono::milliseconds timeout,
[[maybe_unused]] const GWObjects::DeviceRestrictions &Restrictions) {
if(UserInfo_.userinfo.userRole != SecurityObjects::ROOT &&
UserInfo_.userinfo.userRole != SecurityObjects::ADMIN) {
CallCanceled("REENROLL", CMD_UUID, CMD_RPC, RESTAPI::Errors::ACCESS_DENIED);
return UnAuthorized(RESTAPI::Errors::ACCESS_DENIED);
}
poco_debug(Logger_, fmt::format("REENROLL({},{}): TID={} user={} serial={}", CMD_UUID,
CMD_RPC, TransactionId_, Requester(), SerialNumber_));
if(IsDeviceSimulated(SerialNumber_)) {
CallCanceled("REENROLL", CMD_UUID, CMD_RPC, RESTAPI::Errors::SimulatedDeviceNotSupported);
return BadRequest(RESTAPI::Errors::SimulatedDeviceNotSupported);
}
GWObjects::ReEnroll PR;
if(!PR.from_json(ParsedBody_)) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
GWObjects::CommandDetails Cmd;
Cmd.SerialNumber = SerialNumber_;
Cmd.SubmittedBy = Requester();
Cmd.UUID = CMD_UUID;
Cmd.Command = uCentralProtocol::REENROLL;
std::ostringstream os;
ParsedBody_->stringify(os);
Cmd.Details = os.str();
Cmd.RunAt = PR.when;
Cmd.ErrorCode = 0;
Cmd.WaitingForFile = 0;
return RESTAPI_RPC::WaitForCommand(CMD_RPC, APCommands::Commands::reenroll, false, Cmd,
*ParsedBody_, *Request, *Response, timeout, nullptr, this,
Logger_);
}
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -68,14 +68,6 @@ namespace OpenWifi {
const GWObjects::DeviceRestrictions &R); const GWObjects::DeviceRestrictions &R);
void Transfer(const std::string &UUID, uint64_t RPC, std::chrono::milliseconds timeout, void Transfer(const std::string &UUID, uint64_t RPC, std::chrono::milliseconds timeout,
const GWObjects::DeviceRestrictions &R); const GWObjects::DeviceRestrictions &R);
void PowerCycle(const std::string &UUID, uint64_t RPC, std::chrono::milliseconds timeout,
const GWObjects::DeviceRestrictions &R);
void FixedConfig(const std::string &UUID, uint64_t RPC, std::chrono::milliseconds timeout,
const GWObjects::DeviceRestrictions &R);
void CableDiagnostics(const std::string &UUID, uint64_t RPC, std::chrono::milliseconds timeout,
const GWObjects::DeviceRestrictions &R);
void ReEnroll(const std::string &UUID, uint64_t RPC, std::chrono::milliseconds timeout,
const GWObjects::DeviceRestrictions &R);
static auto PathName() { static auto PathName() {
return std::list<std::string>{"/api/v1/device/{serialNumber}/{command}"}; return std::list<std::string>{"/api/v1/device/{serialNumber}/{command}"};

View File

@@ -17,8 +17,6 @@
#include "RESTAPI_device_helper.h" #include "RESTAPI_device_helper.h"
#include "AP_WS_Server.h"
namespace OpenWifi { namespace OpenWifi {
void RESTAPI_device_handler::DoGet() { void RESTAPI_device_handler::DoGet() {
std::string SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, ""); std::string SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, "");
@@ -82,9 +80,6 @@ namespace OpenWifi {
return OK(); return OK();
} else if (StorageService()->DeleteDevice(SerialNumber)) { } else if (StorageService()->DeleteDevice(SerialNumber)) {
if(AP_WS_Server()->Connected(Utils::SerialNumberToInt(SerialNumber))) {
AP_WS_Server()->Disconnect(Utils::SerialNumberToInt(SerialNumber));
}
return OK(); return OK();
} }
@@ -106,10 +101,9 @@ namespace OpenWifi {
} }
auto Config = Obj->get("configuration").toString(); auto Config = Obj->get("configuration").toString();
Poco::JSON::Object Answer; Poco::JSON::Object Answer;
std::string Error; std::vector<std::string> Error;
auto DeviceType = Poco::toLower(GetParameter("deviceType", Platforms::AP));
auto Res = auto Res =
ValidateUCentralConfiguration(ConfigurationValidator::GetType(DeviceType),Config, Error, GetBoolParameter("strict", false)); ValidateUCentralConfiguration(Config, Error, GetBoolParameter("strict", false));
Answer.set("valid", Res); Answer.set("valid", Res);
if (!Error.empty()) if (!Error.empty())
Answer.set("error", Error); Answer.set("error", Error);
@@ -129,13 +123,12 @@ namespace OpenWifi {
return BadRequest(RESTAPI::Errors::SerialNumberMismatch); return BadRequest(RESTAPI::Errors::SerialNumberMismatch);
} }
std::string Error; std::vector<std::string> Error;
if (Device.Configuration.empty() || if (Device.Configuration.empty() ||
(!Device.Configuration.empty() && (!Device.Configuration.empty() &&
!ValidateUCentralConfiguration(ConfigurationValidator::GetType(Device.DeviceType), !ValidateUCentralConfiguration(Device.Configuration, Error,
Device.Configuration, Error,
GetBoolParameter("strict", false)))) { GetBoolParameter("strict", false)))) {
return BadRequest(RESTAPI::Errors::ConfigBlockInvalid, Error); return BadRequest(RESTAPI::Errors::ConfigBlockInvalid);
} }
for (auto &i : Device.Notes) { for (auto &i : Device.Notes) {
@@ -176,11 +169,10 @@ namespace OpenWifi {
} }
if (!NewDevice.Configuration.empty()) { if (!NewDevice.Configuration.empty()) {
std::string Error; std::vector<std::string> Error;
if (!ValidateUCentralConfiguration(ConfigurationValidator::GetType(Existing.DeviceType), if (!ValidateUCentralConfiguration(NewDevice.Configuration, Error,
NewDevice.Configuration, Error,
GetBoolParameter("strict", false))) { GetBoolParameter("strict", false))) {
return BadRequest(RESTAPI::Errors::ConfigBlockInvalid, Error); return BadRequest(RESTAPI::Errors::ConfigBlockInvalid);
} }
Config::Config NewConfig(NewDevice.Configuration); Config::Config NewConfig(NewDevice.Configuration);
uint64_t NewConfigUUID = Utils::Now(); uint64_t NewConfigUUID = Utils::Now();

View File

@@ -82,24 +82,15 @@ namespace OpenWifi {
} }
} }
auto platform = Poco::toLower(GetParameter("platform", ""));
auto serialOnly = GetBoolParameter(RESTAPI::Protocol::SERIALONLY, false); auto serialOnly = GetBoolParameter(RESTAPI::Protocol::SERIALONLY, false);
auto deviceWithStatus = GetBoolParameter(RESTAPI::Protocol::DEVICEWITHSTATUS, false); auto deviceWithStatus = GetBoolParameter(RESTAPI::Protocol::DEVICEWITHSTATUS, false);
auto completeInfo = GetBoolParameter("completeInfo", false); auto completeInfo = GetBoolParameter("completeInfo", false);
auto includeProvisioned = GetBoolParameter("includeProvisioned", true);
if(!platform.empty() && (platform!=Platforms::AP && platform!=Platforms::SWITCH && platform!="all")) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
}
if(platform=="all")
platform="";
Poco::JSON::Object RetObj; Poco::JSON::Object RetObj;
if (!QB_.Select.empty()) { if (!QB_.Select.empty()) {
Poco::JSON::Array Objects; Poco::JSON::Array Objects;
for (auto &i : SelectedRecords()) { for (auto &i : SelectedRecords()) {
auto &SerialNumber = i; auto SerialNumber = i;
if (!Utils::ValidSerialNumber(i)) if (!Utils::ValidSerialNumber(i))
continue; continue;
GWObjects::Device D; GWObjects::Device D;
@@ -125,14 +116,14 @@ namespace OpenWifi {
else else
RetObj.set(RESTAPI::Protocol::DEVICES, Objects); RetObj.set(RESTAPI::Protocol::DEVICES, Objects);
} else if (QB_.CountOnly) { } else if (QB_.CountOnly == true) {
uint64_t Count = 0; uint64_t Count = 0;
if (StorageService()->GetDeviceCount(Count, platform)) { if (StorageService()->GetDeviceCount(Count)) {
return ReturnCountOnly(Count); return ReturnCountOnly(Count);
} }
} else if (serialOnly) { } else if (serialOnly) {
std::vector<std::string> SerialNumbers; std::vector<std::string> SerialNumbers;
StorageService()->GetDeviceSerialNumbers(QB_.Offset, QB_.Limit, SerialNumbers, OrderBy, platform, includeProvisioned); StorageService()->GetDeviceSerialNumbers(QB_.Offset, QB_.Limit, SerialNumbers, OrderBy);
Poco::JSON::Array Objects; Poco::JSON::Array Objects;
for (const auto &i : SerialNumbers) { for (const auto &i : SerialNumbers) {
Objects.add(i); Objects.add(i);
@@ -150,7 +141,7 @@ namespace OpenWifi {
RetObj.set("serialNumbers", Objects); RetObj.set("serialNumbers", Objects);
} else { } else {
std::vector<GWObjects::Device> Devices; std::vector<GWObjects::Device> Devices;
StorageService()->GetDevices(QB_.Offset, QB_.Limit, Devices, OrderBy, platform, includeProvisioned); StorageService()->GetDevices(QB_.Offset, QB_.Limit, Devices, OrderBy);
Poco::JSON::Array Objects; Poco::JSON::Array Objects;
for (const auto &i : Devices) { for (const auto &i : Devices) {
Poco::JSON::Object Obj; Poco::JSON::Object Obj;
@@ -184,13 +175,11 @@ namespace OpenWifi {
} }
if(GetBoolParameter("simulatedDevices",false)) { if(GetBoolParameter("simulatedDevices",false)) {
auto F = []() ->void { if(StorageService()->DeleteSimulatedDevice("")) {
StorageService()->DeleteSimulatedDevice("");
};
std::thread T(F);
T.detach();
return OK(); return OK();
} }
return NotFound();
}
if(!QB_.Select.empty() && !Utils::ValidSerialNumbers(QB_.Select)) { if(!QB_.Select.empty() && !Utils::ValidSerialNumbers(QB_.Select)) {
return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters); return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);

View File

@@ -22,15 +22,9 @@ namespace OpenWifi {
std::string FileType; std::string FileType;
std::string FileContent; std::string FileContent;
int WaitingForFile = 0; if (!StorageService()->GetAttachedFileContent(UUID, SerialNumber, FileContent, FileType) || FileContent.empty()) {
if (!StorageService()->GetAttachedFileContent(UUID, SerialNumber, FileContent, FileType, WaitingForFile) && !WaitingForFile) {
return NotFound(); return NotFound();
} }
else if (WaitingForFile) {
// waiting for file to be uploaded, return Accepted
return Accepted();
}
if (FileType == "pcap") { if (FileType == "pcap") {
SendFileContent(FileContent, "application/vnd.tcpdump.pcap", UUID + ".pcap"); SendFileContent(FileContent, "application/vnd.tcpdump.pcap", UUID + ".pcap");
} }

View File

@@ -8,7 +8,7 @@
#pragma once #pragma once
#include <framework/RESTAPI_Handler.h> #include "framework/RESTAPI_Handler.h"
namespace OpenWifi { namespace OpenWifi {
class RESTAPI_file : public RESTAPIHandler { class RESTAPI_file : public RESTAPIHandler {

View File

@@ -7,7 +7,6 @@
#include "RESTAPI_ProvObjects.h" #include "RESTAPI_ProvObjects.h"
#include "framework/utils.h" #include "framework/utils.h"
#include <vector> #include <vector>
#include "framework/ow_constants.h"
namespace OpenWifi { namespace OpenWifi {

View File

@@ -12,7 +12,6 @@
#include "Daemon.h" #include "Daemon.h"
#ifdef TIP_GATEWAY_SERVICE #ifdef TIP_GATEWAY_SERVICE
#include "AP_WS_Server.h" #include "AP_WS_Server.h"
#include "StorageService.h"
#include "CapabilitiesCache.h" #include "CapabilitiesCache.h"
#include "RADIUSSessionTracker.h" #include "RADIUSSessionTracker.h"
#endif #endif
@@ -30,8 +29,8 @@ namespace OpenWifi::GWObjects {
void Device::to_json(Poco::JSON::Object &Obj) const { void Device::to_json(Poco::JSON::Object &Obj) const {
field_to_json(Obj, "serialNumber", SerialNumber); field_to_json(Obj, "serialNumber", SerialNumber);
#ifdef TIP_GATEWAY_SERVICE #ifdef TIP_GATEWAY_SERVICE
field_to_json(Obj, "deviceType", StorageService()->GetPlatform(SerialNumber)); field_to_json(Obj, "deviceType", CapabilitiesCache::instance()->GetPlatform(Compatible));
field_to_json(Obj, "blackListed", StorageService()->IsBlackListed(Utils::MACToInt(SerialNumber))); field_to_json(Obj, "hasRADIUSSessions", RADIUSSessionTracker()->HasSessions(SerialNumber));
#endif #endif
field_to_json(Obj, "macAddress", MACAddress); field_to_json(Obj, "macAddress", MACAddress);
field_to_json(Obj, "manufacturer", Manufacturer); field_to_json(Obj, "manufacturer", Manufacturer);
@@ -69,14 +68,9 @@ namespace OpenWifi::GWObjects {
#ifdef TIP_GATEWAY_SERVICE #ifdef TIP_GATEWAY_SERVICE
ConnectionState ConState; ConnectionState ConState;
#ifdef USE_MEDUSA_CLIENT
auto Res = GS()->GetState(SerialNumber);
if (Res.has_value()) {
Res.value().to_json(SerialNumber,Obj);
#else
if (AP_WS_Server()->GetState(SerialNumber, ConState)) { if (AP_WS_Server()->GetState(SerialNumber, ConState)) {
ConState.to_json(SerialNumber,Obj); ConState.to_json(SerialNumber,Obj);
#endif
} else { } else {
field_to_json(Obj, "ipAddress", ""); field_to_json(Obj, "ipAddress", "");
field_to_json(Obj, "txBytes", (uint64_t)0); field_to_json(Obj, "txBytes", (uint64_t)0);
@@ -172,16 +166,13 @@ namespace OpenWifi::GWObjects {
field_to_json(Obj, "recorded", Recorded); field_to_json(Obj, "recorded", Recorded);
} }
bool HealthCheck::from_json(const Poco::JSON::Object::Ptr &Obj) { void DefaultConfiguration::to_json(Poco::JSON::Object &Obj) const {
try { EmbedDocument("configuration", Obj, Configuration);
field_from_json(Obj, "UUID", UUID); field_to_json(Obj, "name", Name);
field_from_json(Obj, "sanity", Sanity); field_to_json(Obj, "modelIds", Models);
field_from_json(Obj, "recorded", Recorded); field_to_json(Obj, "description", Description);
return true; field_to_json(Obj, "created", Created);
} catch(...) { field_to_json(Obj, "lastModified", LastModified);
}
return false;
} }
void DefaultFirmware::to_json(Poco::JSON::Object &Obj) const { void DefaultFirmware::to_json(Poco::JSON::Object &Obj) const {
@@ -231,25 +222,12 @@ namespace OpenWifi::GWObjects {
field_to_json(Obj, "deferred", deferred); field_to_json(Obj, "deferred", deferred);
} }
void DefaultConfiguration::to_json(Poco::JSON::Object &Obj) const {
EmbedDocument("configuration", Obj, configuration);
field_to_json(Obj, "name", name);
field_to_json(Obj, "modelIds", models);
field_to_json(Obj, "description", description);
field_to_json(Obj, "created", created);
field_to_json(Obj, "lastModified", lastModified);
field_to_json(Obj, "platform", platform);
}
bool DefaultConfiguration::from_json(const Poco::JSON::Object::Ptr &Obj) { bool DefaultConfiguration::from_json(const Poco::JSON::Object::Ptr &Obj) {
try { try {
field_from_json(Obj, "configuration", configuration); field_from_json(Obj, "name", Name);
field_from_json(Obj, "name", name); field_from_json(Obj, "configuration", Configuration);
field_from_json(Obj, "modelIds", models); field_from_json(Obj, "modelIds", Models);
field_from_json(Obj, "description", description); field_from_json(Obj, "description", Description);
field_from_json(Obj, "created", created);
field_from_json(Obj, "lastModified", lastModified);
field_from_json(Obj, "platform", platform);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
} }
@@ -297,13 +275,14 @@ namespace OpenWifi::GWObjects {
field_to_json(Obj, "connectionCompletionTime", connectionCompletionTime); field_to_json(Obj, "connectionCompletionTime", connectionCompletionTime);
field_to_json(Obj, "totalConnectionTime", Utils::Now() - started); field_to_json(Obj, "totalConnectionTime", Utils::Now() - started);
field_to_json(Obj, "certificateExpiryDate", certificateExpiryDate); field_to_json(Obj, "certificateExpiryDate", certificateExpiryDate);
field_to_json(Obj, "certificateIssuerName", certificateIssuerName);
field_to_json(Obj, "connectReason", connectReason); field_to_json(Obj, "connectReason", connectReason);
field_to_json(Obj, "uptime", uptime);
field_to_json(Obj, "compatible", Compatible);
#ifdef TIP_GATEWAY_SERVICE #ifdef TIP_GATEWAY_SERVICE
hasRADIUSSessions = RADIUSSessionTracker()->HasSessions(SerialNumber); hasRADIUSSessions = RADIUSSessionTracker()->HasSessions(SerialNumber);
AP_WS_Server()->ExtendedAttributes(SerialNumber, hasGPS, sanity,
memoryUsed,
load,
temperature);
#endif #endif
field_to_json(Obj, "hasRADIUSSessions", hasRADIUSSessions ); field_to_json(Obj, "hasRADIUSSessions", hasRADIUSSessions );
field_to_json(Obj, "hasGPS", hasGPS); field_to_json(Obj, "hasGPS", hasGPS);
@@ -335,46 +314,6 @@ namespace OpenWifi::GWObjects {
} }
} }
bool ConnectionState::from_json(const Poco::JSON::Object::Ptr &Obj) {
try {
field_from_json(Obj, "compatible", Compatible);
field_from_json(Obj, "ipAddress", Address);
field_from_json(Obj, "txBytes", TX);
field_from_json(Obj, "rxBytes", RX);
field_from_json(Obj, "messageCount", MessageCount);
field_from_json(Obj, "UUID", UUID);
field_from_json(Obj, "connected", Connected);
field_from_json(Obj, "firmware", Firmware);
field_from_json(Obj, "lastContact", LastContact);
field_from_json(Obj, "associations_2G", Associations_2G);
field_from_json(Obj, "associations_5G", Associations_5G);
field_from_json(Obj, "associations_6G", Associations_6G);
field_from_json(Obj, "webSocketClients", webSocketClients);
field_from_json(Obj, "websocketPackets", websocketPackets);
field_from_json(Obj, "kafkaClients", kafkaClients);
field_from_json(Obj, "kafkaPackets", kafkaPackets);
field_from_json(Obj, "locale", locale);
field_from_json(Obj, "started", started);
field_from_json(Obj, "sessionId", sessionId);
field_from_json(Obj, "connectionCompletionTime", connectionCompletionTime);
field_from_json(Obj, "totalConnectionTime", totalConnectionTime);
field_from_json(Obj, "certificateExpiryDate", certificateExpiryDate);
field_from_json(Obj, "certificateIssuerName", certificateIssuerName);
field_from_json(Obj, "connectReason", connectReason);
field_from_json(Obj, "uptime", uptime);
field_from_json(Obj, "hasRADIUSSessions", hasRADIUSSessions );
field_from_json(Obj, "hasGPS", hasGPS);
field_from_json(Obj, "sanity", sanity);
field_from_json(Obj, "memoryUsed", memoryUsed);
field_from_json(Obj, "sanity", sanity);
field_from_json(Obj, "load", load);
field_from_json(Obj, "temperature", temperature);
return true;
} catch(const Poco::Exception &E) {
}
return false;
}
void DeviceConnectionStatistics::to_json(Poco::JSON::Object &Obj) const { void DeviceConnectionStatistics::to_json(Poco::JSON::Object &Obj) const {
field_to_json(Obj, "averageConnectionTime", averageConnectionTime); field_to_json(Obj, "averageConnectionTime", averageConnectionTime);
field_to_json(Obj, "connectedDevices", connectedDevices); field_to_json(Obj, "connectedDevices", connectedDevices);
@@ -772,7 +711,7 @@ namespace OpenWifi::GWObjects {
bool DeviceCertificateUpdateRequest::from_json(const Poco::JSON::Object::Ptr &Obj) { bool DeviceCertificateUpdateRequest::from_json(const Poco::JSON::Object::Ptr &Obj) {
try { try {
field_from_json(Obj, "serial", serialNumber); field_from_json(Obj, "serialNumber", serialNumber);
field_from_json(Obj, "encodedCertificate", encodedCertificate); field_from_json(Obj, "encodedCertificate", encodedCertificate);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -780,55 +719,4 @@ namespace OpenWifi::GWObjects {
return false; return false;
} }
bool PowerCyclePort::from_json(const Poco::JSON::Object::Ptr &Obj) {
try {
field_from_json(Obj, "name", name);
field_from_json(Obj, "cycle", cycle);
return true;
} catch (const Poco::Exception &E) {
}
return false;
}
bool PowerCycleRequest::from_json(const Poco::JSON::Object::Ptr &Obj) {
try {
field_from_json(Obj, "serial", serialNumber);
field_from_json(Obj, "when", when);
field_from_json(Obj, "ports", ports);
return true;
} catch (const Poco::Exception &E) {
}
return false;
}
bool FixedConfig::from_json(const Poco::JSON::Object::Ptr &Obj) {
try {
field_from_json(Obj, "serial", serialNumber);
field_from_json(Obj, "country", country);
return true;
} catch (const Poco::Exception &E) {
}
return false;
}
bool CableDiagnostics::from_json(const Poco::JSON::Object::Ptr &Obj) {
try {
field_from_json(Obj, "serial", serialNumber);
field_from_json(Obj, "when", when);
field_from_json(Obj, "ports", ports);
return true;
} catch (const Poco::Exception &E) {
}
return false;
}
bool ReEnroll::from_json(const Poco::JSON::Object::Ptr &Obj) {
try {
field_from_json(Obj, "serial", serialNumber);
field_from_json(Obj, "when", when);
return true;
} catch (const Poco::Exception &E) {
}
return false;
}
} // namespace OpenWifi::GWObjects } // namespace OpenWifi::GWObjects

View File

@@ -42,7 +42,6 @@ namespace OpenWifi::GWObjects {
uint64_t sessionId = 0; uint64_t sessionId = 0;
double connectionCompletionTime = 0.0; double connectionCompletionTime = 0.0;
std::uint64_t certificateExpiryDate = 0; std::uint64_t certificateExpiryDate = 0;
std::string certificateIssuerName;
std::uint64_t hasRADIUSSessions = 0; std::uint64_t hasRADIUSSessions = 0;
bool hasGPS = false; bool hasGPS = false;
std::uint64_t sanity=0; std::uint64_t sanity=0;
@@ -50,11 +49,8 @@ namespace OpenWifi::GWObjects {
std::double_t load=0.0; std::double_t load=0.0;
std::double_t temperature=0.0; std::double_t temperature=0.0;
std::string connectReason; std::string connectReason;
std::uint64_t uptime=0;
std::uint64_t totalConnectionTime=0;
void to_json(const std::string &SerialNumber, Poco::JSON::Object &Obj) ; void to_json(const std::string &SerialNumber, Poco::JSON::Object &Obj) ;
bool from_json(const Poco::JSON::Object::Ptr &Obj);
}; };
struct DeviceRestrictionsKeyInfo { struct DeviceRestrictionsKeyInfo {
@@ -116,7 +112,6 @@ namespace OpenWifi::GWObjects {
std::uint64_t lastRecordedContact=0; std::uint64_t lastRecordedContact=0;
std::uint64_t certificateExpiryDate = 0; std::uint64_t certificateExpiryDate = 0;
std::string connectReason; std::string connectReason;
bool blackListed=false;
void to_json(Poco::JSON::Object &Obj) const; void to_json(Poco::JSON::Object &Obj) const;
void to_json_with_status(Poco::JSON::Object &Obj) const; void to_json_with_status(Poco::JSON::Object &Obj) const;
@@ -147,9 +142,7 @@ namespace OpenWifi::GWObjects {
std::string Data; std::string Data;
uint64_t Recorded = 0; uint64_t Recorded = 0;
uint64_t Sanity = 0; uint64_t Sanity = 0;
void to_json(Poco::JSON::Object &Obj) const; void to_json(Poco::JSON::Object &Obj) const;
bool from_json(const Poco::JSON::Object::Ptr &Obj);
}; };
struct Capabilities { struct Capabilities {
@@ -181,13 +174,12 @@ namespace OpenWifi::GWObjects {
}; };
struct DefaultConfiguration { struct DefaultConfiguration {
std::string name; std::string Name;
std::string configuration; std::string Configuration;
Types::StringVec models; Types::StringVec Models;
std::string description; std::string Description;
uint64_t created; uint64_t Created;
uint64_t lastModified; uint64_t LastModified;
std::string platform;
void to_json(Poco::JSON::Object &Obj) const; void to_json(Poco::JSON::Object &Obj) const;
bool from_json(const Poco::JSON::Object::Ptr &Obj); bool from_json(const Poco::JSON::Object::Ptr &Obj);
}; };
@@ -521,37 +513,4 @@ namespace OpenWifi::GWObjects {
bool from_json(const Poco::JSON::Object::Ptr &Obj); bool from_json(const Poco::JSON::Object::Ptr &Obj);
}; };
struct PowerCyclePort {
std::string name;
std::uint64_t cycle=10000;
bool from_json(const Poco::JSON::Object::Ptr &Obj);
};
struct PowerCycleRequest {
std::string serialNumber;
std::uint64_t when;
std::vector<PowerCyclePort> ports;
bool from_json(const Poco::JSON::Object::Ptr &Obj);
};
struct FixedConfig {
std::string serialNumber;
std::string country;
bool from_json(const Poco::JSON::Object::Ptr &Obj);
};
struct CableDiagnostics {
std::string serialNumber;
std::uint64_t when;
std::vector<std::string> ports;
bool from_json(const Poco::JSON::Object::Ptr &Obj);
};
struct ReEnroll {
std::string serialNumber;
std::uint64_t when;
bool from_json(const Poco::JSON::Object::Ptr &Obj);
};
} // namespace OpenWifi::GWObjects } // namespace OpenWifi::GWObjects

View File

@@ -587,9 +587,6 @@ namespace OpenWifi::ProvObjects {
field_to_json(Obj, "locale", locale); field_to_json(Obj, "locale", locale);
field_to_json(Obj, "realMacAddress", realMacAddress); field_to_json(Obj, "realMacAddress", realMacAddress);
field_to_json(Obj, "doNotAllowOverrides", doNotAllowOverrides); field_to_json(Obj, "doNotAllowOverrides", doNotAllowOverrides);
field_to_json(Obj, "imported", imported);
field_to_json(Obj, "connected", connected);
field_to_json(Obj, "platform", platform);
} }
bool InventoryTag::from_json(const Poco::JSON::Object::Ptr &Obj) { bool InventoryTag::from_json(const Poco::JSON::Object::Ptr &Obj) {
@@ -612,9 +609,6 @@ namespace OpenWifi::ProvObjects {
field_from_json(Obj, "locale", locale); field_from_json(Obj, "locale", locale);
field_from_json(Obj, "realMacAddress", realMacAddress); field_from_json(Obj, "realMacAddress", realMacAddress);
field_from_json(Obj, "doNotAllowOverrides", doNotAllowOverrides); field_from_json(Obj, "doNotAllowOverrides", doNotAllowOverrides);
field_from_json(Obj, "imported", imported);
field_from_json(Obj, "connected", connected);
field_from_json(Obj, "platform", platform);
return true; return true;
} catch (...) { } catch (...) {
} }

View File

@@ -490,11 +490,9 @@ namespace OpenWifi::ProvObjects {
std::string locale; std::string locale;
std::string realMacAddress; std::string realMacAddress;
bool doNotAllowOverrides = false; bool doNotAllowOverrides = false;
std::uint64_t imported=0;
std::uint64_t connected=0;
std::string platform{Platforms::AP};
void to_json(Poco::JSON::Object &Obj) const; void to_json(Poco::JSON::Object &Obj) const;
bool from_json(const Poco::JSON::Object::Ptr &Obj); bool from_json(const Poco::JSON::Object::Ptr &Obj);
}; };

View File

@@ -24,7 +24,7 @@ namespace OpenWifi::StateUtils {
} }
bool ComputeAssociations(const Poco::JSON::Object::Ptr RawObject, uint64_t &Radios_2G, bool ComputeAssociations(const Poco::JSON::Object::Ptr RawObject, uint64_t &Radios_2G,
uint64_t &Radios_5G, uint64_t &Radios_6G, uint64_t &UpTime ) { uint64_t &Radios_5G, uint64_t &Radios_6G) {
Radios_2G = 0; Radios_2G = 0;
Radios_5G = 0; Radios_5G = 0;
Radios_6G = 0; Radios_6G = 0;
@@ -90,15 +90,9 @@ namespace OpenWifi::StateUtils {
} }
} }
} }
// std::cout << Radios_2G << " " << Radios_5G << " " << Radios_6G << std::endl;
return true; return true;
} }
if(RawObject->has("unit") && !RawObject->isNull("unit") && RawObject->isObject("unit")) {
auto unit = RawObject->getObject("unit");
if(unit->has("uptime")) {
UpTime = unit->get("uptime");
}
}
return false; return false;
} }
} // namespace OpenWifi::StateUtils } // namespace OpenWifi::StateUtils

View File

@@ -8,5 +8,5 @@
namespace OpenWifi::StateUtils { namespace OpenWifi::StateUtils {
bool ComputeAssociations(const Poco::JSON::Object::Ptr RawObject, uint64_t &Radios_2G, bool ComputeAssociations(const Poco::JSON::Object::Ptr RawObject, uint64_t &Radios_2G,
uint64_t &Radios_5G, uint64_t &Radio_6G, uint64_t &UpTime); uint64_t &Radios_5G, uint64_t &Radio_6G);
} }

View File

@@ -22,8 +22,6 @@ namespace OpenWifi {
ScriptDB_->Create(); ScriptDB_->Create();
ScriptDB_->Initialize(); ScriptDB_->Initialize();
FixDeviceTypeBug();
return 0; return 0;
} }

View File

@@ -16,22 +16,6 @@
namespace OpenWifi { namespace OpenWifi {
class LockedDbSession {
public:
explicit LockedDbSession();
~LockedDbSession() = default;
inline std::mutex &Mutex() { return *Mutex_; };
inline Poco::Data::Session &Session() {
if(!Session_->isConnected()) {
Session_->reconnect();
}
return *Session_;
};
private:
std::shared_ptr<Poco::Data::Session> Session_;
std::shared_ptr<std::mutex> Mutex_;
};
class Storage : public StorageClass { class Storage : public StorageClass {
public: public:
@@ -106,8 +90,7 @@ namespace OpenWifi {
// typedef std::map<std::string,std::string> DeviceCapabilitiesCache; // typedef std::map<std::string,std::string> DeviceCapabilitiesCache;
bool AddLog(LockedDbSession &Session, const GWObjects::DeviceLog &Log); bool AddLog(const GWObjects::DeviceLog &Log);
bool AddStatisticsData(Poco::Data::Session &Session, const GWObjects::Statistics &Stats);
bool AddStatisticsData(const GWObjects::Statistics &Stats); bool AddStatisticsData(const GWObjects::Statistics &Stats);
bool GetStatisticsData(std::string &SerialNumber, uint64_t FromDate, uint64_t ToDate, bool GetStatisticsData(std::string &SerialNumber, uint64_t FromDate, uint64_t ToDate,
uint64_t Offset, uint64_t HowMany, uint64_t Offset, uint64_t HowMany,
@@ -119,7 +102,6 @@ namespace OpenWifi {
std::vector<GWObjects::Statistics> &Stats); std::vector<GWObjects::Statistics> &Stats);
bool AddHealthCheckData(const GWObjects::HealthCheck &Check); bool AddHealthCheckData(const GWObjects::HealthCheck &Check);
bool AddHealthCheckData(LockedDbSession &Session, const GWObjects::HealthCheck &Check);
bool GetHealthCheckData(std::string &SerialNumber, uint64_t FromDate, uint64_t ToDate, bool GetHealthCheckData(std::string &SerialNumber, uint64_t FromDate, uint64_t ToDate,
uint64_t Offset, uint64_t HowMany, uint64_t Offset, uint64_t HowMany,
std::vector<GWObjects::HealthCheck> &Checks); std::vector<GWObjects::HealthCheck> &Checks);
@@ -133,43 +115,31 @@ namespace OpenWifi {
uint64_t &NewUUID); uint64_t &NewUUID);
bool RollbackDeviceConfigurationChange(std::string & SerialNumber); bool RollbackDeviceConfigurationChange(std::string & SerialNumber);
bool CompleteDeviceConfigurationChange(Poco::Data::Session &Session, std::string & SerialNumber);
bool CompleteDeviceConfigurationChange(std::string & SerialNumber); bool CompleteDeviceConfigurationChange(std::string & SerialNumber);
bool CreateDevice(LockedDbSession &Session, GWObjects::Device &);
bool CreateDevice(GWObjects::Device &); bool CreateDevice(GWObjects::Device &);
bool CreateDefaultDevice(Poco::Data::Session &Session,std::string &SerialNumber, bool CreateDefaultDevice(std::string &SerialNumber, const Config::Capabilities &Caps,
const Config::Capabilities &Caps,
std::string &Firmware, const Poco::Net::IPAddress &IPAddress, std::string &Firmware, const Poco::Net::IPAddress &IPAddress,
bool simulated); bool simulated);
bool CreateDevice(Poco::Data::Session &Sess, GWObjects::Device &DeviceDetails);
bool GetDevice(LockedDbSession &Session, const std::string &SerialNumber, GWObjects::Device &); bool GetDevice(std::string &SerialNumber, GWObjects::Device &);
bool GetDevice(Poco::Data::Session &Session, const std::string &SerialNumber, GWObjects::Device &DeviceDetails);
bool GetDevice(const std::string &SerialNumber, GWObjects::Device &);
bool GetDevices(uint64_t From, uint64_t HowMany, std::vector<GWObjects::Device> &Devices, bool GetDevices(uint64_t From, uint64_t HowMany, std::vector<GWObjects::Device> &Devices,
const std::string &orderBy = "", const std::string &orderBy = "");
const std::string &platform = "",
bool includeProvisioned = true);
// bool GetDevices(uint64_t From, uint64_t HowMany, const std::string & Select, // bool GetDevices(uint64_t From, uint64_t HowMany, const std::string & Select,
// std::vector<GWObjects::Device> &Devices, const std::string & orderBy=""); // std::vector<GWObjects::Device> &Devices, const std::string & orderBy="");
bool DeleteDevice(std::string &SerialNumber); bool DeleteDevice(std::string &SerialNumber);
bool DeleteDevices(std::string &SerialPattern, bool SimulatedOnly); bool DeleteDevices(std::string &SerialPattern, bool SimulatedOnly);
bool DeleteDevices(std::uint64_t OlderContact, bool SimulatedOnly); bool DeleteDevices(std::uint64_t OlderContact, bool SimulatedOnly);
std::string GetPlatform(const std::string &SerialNumber);
bool UpdateDevice(GWObjects::Device &); bool UpdateDevice(GWObjects::Device &);
bool UpdateDevice(LockedDbSession &Session, GWObjects::Device &);
bool UpdateDevice(Poco::Data::Session &Sess, GWObjects::Device &NewDeviceDetails);
bool DeviceExists(std::string &SerialNumber); bool DeviceExists(std::string &SerialNumber);
bool SetConnectInfo(std::string &SerialNumber, std::string &Firmware); bool SetConnectInfo(std::string &SerialNumber, std::string &Firmware);
bool GetDeviceCount(uint64_t &Count, const std::string &platform = ""); bool GetDeviceCount(uint64_t &Count);
bool GetDeviceSerialNumbers(uint64_t From, uint64_t HowMany, bool GetDeviceSerialNumbers(uint64_t From, uint64_t HowMany,
std::vector<std::string> &SerialNumbers, std::vector<std::string> &SerialNumbers,
const std::string &orderBy = "", const std::string &orderBy = "");
const std::string &platform = "",
bool includeProvisioned = true);
bool GetDeviceFWUpdatePolicy(std::string &SerialNumber, std::string &Policy); bool GetDeviceFWUpdatePolicy(std::string &SerialNumber, std::string &Policy);
bool SetDevicePassword(LockedDbSession &Session, std::string &SerialNumber, std::string &Password); bool SetDevicePassword(std::string &SerialNumber, std::string &Password);
bool UpdateSerialNumberCache(); bool UpdateSerialNumberCache();
static void GetDeviceDbFieldList(Types::StringVec &Fields); static void GetDeviceDbFieldList(Types::StringVec &Fields);
@@ -178,11 +148,9 @@ namespace OpenWifi {
bool UpdateDeviceCapabilities(std::string &SerialNumber, bool UpdateDeviceCapabilities(std::string &SerialNumber,
const Config::Capabilities &Capabilities); const Config::Capabilities &Capabilities);
bool UpdateDeviceCapabilities(Poco::Data::Session &Session, std::string &SerialNumber,
const Config::Capabilities &Capabilities);
bool GetDeviceCapabilities(std::string &SerialNumber, GWObjects::Capabilities &); bool GetDeviceCapabilities(std::string &SerialNumber, GWObjects::Capabilities &);
bool DeleteDeviceCapabilities(std::string &SerialNumber); bool DeleteDeviceCapabilities(std::string &SerialNumber);
bool CreateDeviceCapabilities(Poco::Data::Session &Session, std::string &SerialNumber, bool CreateDeviceCapabilities(std::string &SerialNumber,
const Config::Capabilities &Capabilities); const Config::Capabilities &Capabilities);
bool InitCapabilitiesCache(); bool InitCapabilitiesCache();
@@ -203,7 +171,6 @@ namespace OpenWifi {
bool GetDefaultConfigurations(uint64_t From, uint64_t HowMany, bool GetDefaultConfigurations(uint64_t From, uint64_t HowMany,
std::vector<GWObjects::DefaultConfiguration> &Devices); std::vector<GWObjects::DefaultConfiguration> &Devices);
bool FindDefaultConfigurationForModel(const std::string &Model, bool FindDefaultConfigurationForModel(const std::string &Model,
const std::string &Platform,
GWObjects::DefaultConfiguration &DefConfig); GWObjects::DefaultConfiguration &DefConfig);
uint64_t GetDefaultConfigurationsCount(); uint64_t GetDefaultConfigurationsCount();
bool DefaultConfigurationAlreadyExists(std::string &Name); bool DefaultConfigurationAlreadyExists(std::string &Name);
@@ -243,7 +210,7 @@ namespace OpenWifi {
const std::string &Type); const std::string &Type);
bool CancelWaitFile(std::string &UUID, std::string &ErrorText); bool CancelWaitFile(std::string &UUID, std::string &ErrorText);
bool GetAttachedFileContent(std::string &UUID, const std::string &SerialNumber, bool GetAttachedFileContent(std::string &UUID, const std::string &SerialNumber,
std::string &FileContent, std::string &Type, int& WaitingForFile); std::string &FileContent, std::string &Type);
bool RemoveAttachedFile(std::string &UUID); bool RemoveAttachedFile(std::string &UUID);
bool SetCommandResult(std::string &UUID, std::string &Result); bool SetCommandResult(std::string &UUID, std::string &Result);
bool GetNewestCommands(std::string &SerialNumber, uint64_t HowMany, bool GetNewestCommands(std::string &SerialNumber, uint64_t HowMany,
@@ -255,15 +222,15 @@ namespace OpenWifi {
void RemovedExpiredCommands(); void RemovedExpiredCommands();
void RemoveTimedOutCommands(); void RemoveTimedOutCommands();
bool RemoveOldCommands(std::string &SerialNumber, std::string &Command); bool RemoveOldCommands(std::string &SerilNumber, std::string &Command);
bool AddBlackListDevices(std::vector<GWObjects::BlackListedDevice> &Devices); bool AddBlackListDevices(std::vector<GWObjects::BlackListedDevice> &Devices);
bool AddBlackListDevice(GWObjects::BlackListedDevice &Device); bool AddBlackListDevice(GWObjects::BlackListedDevice &Device);
bool GetBlackListDevice(std::string &SerialNumber, GWObjects::BlackListedDevice &Device); bool GetBlackListDevice(std::string &SerialNumber, GWObjects::BlackListedDevice &Device);
bool DeleteBlackListDevice(std::string &SerialNumber); bool DeleteBlackListDevice(std::string &SerialNumber);
bool IsBlackListed(std::uint64_t SerialNumber, std::string &reason, bool IsBlackListed(const std::string &SerialNumber, std::string &reason,
std::string &author, std::uint64_t &created); std::string &author, std::uint64_t &created);
bool IsBlackListed(std::uint64_t SerialNumber); bool IsBlackListed(const std::string &SerialNumber);
bool InitializeBlackListCache(); bool InitializeBlackListCache();
bool GetBlackListDevices(uint64_t Offset, uint64_t HowMany, bool GetBlackListDevices(uint64_t Offset, uint64_t HowMany,
std::vector<GWObjects::BlackListedDevice> &Devices); std::vector<GWObjects::BlackListedDevice> &Devices);
@@ -278,9 +245,7 @@ namespace OpenWifi {
bool RemoveCommandListRecordsOlderThan(uint64_t Date); bool RemoveCommandListRecordsOlderThan(uint64_t Date);
bool RemoveUploadedFilesRecordsOlderThan(uint64_t Date); bool RemoveUploadedFilesRecordsOlderThan(uint64_t Date);
bool SetDeviceLastRecordedContact(LockedDbSession &Session, std::string & SerialNumber, std::uint64_t lastRecordedContact); bool SetDeviceLastRecordedContact(std::string & SeialNumber, std::uint64_t lastRecordedContact);
bool SetDeviceLastRecordedContact(std::string & SerialNumber, std::uint64_t lastRecordedContact);
bool SetDeviceLastRecordedContact(Poco::Data::Session & Session, std::string & SerialNumber, std::uint64_t lastRecordedContact);
int Create_Tables(); int Create_Tables();
int Create_Statistics(); int Create_Statistics();
@@ -297,24 +262,13 @@ namespace OpenWifi {
bool AnalyzeCommands(Types::CountedMap &R); bool AnalyzeCommands(Types::CountedMap &R);
bool AnalyzeDevices(GWObjects::Dashboard &D); bool AnalyzeDevices(GWObjects::Dashboard &D);
void FixDeviceTypeBug();
int Start() override; int Start() override;
void Stop() override; void Stop() override;
inline Poco::Data::Session StartSession() {
return Pool_->get();
}
private: private:
std::unique_ptr<OpenWifi::ScriptDB> ScriptDB_; std::unique_ptr<OpenWifi::ScriptDB> ScriptDB_;
}; };
inline auto StorageService() { return Storage::instance(); } inline auto StorageService() { return Storage::instance(); }
inline LockedDbSession::LockedDbSession() {
Session_ = std::make_shared<Poco::Data::Session>(Poco::Data::Session(StorageService()->StartSession()));
Mutex_ = std::make_shared<std::mutex>();
}
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -111,6 +111,7 @@ namespace OpenWifi {
} }
} break; } break;
case TelemetryNotification::NotificationType::unregister: { case TelemetryNotification::NotificationType::unregister: {
std::lock_guard G(Mutex_);
auto client = Clients_.find(Notification->Data_); auto client = Clients_.find(Notification->Data_);
if (client != Clients_.end()) { if (client != Clients_.end()) {

View File

@@ -23,7 +23,7 @@
#include "framework/SubSystemServer.h" #include "framework/SubSystemServer.h"
#include "AP_WS_Reactor_Pool.h" #include "AP_WS_ReactorPool.h"
#include "TelemetryClient.h" #include "TelemetryClient.h"
namespace OpenWifi { namespace OpenWifi {

File diff suppressed because it is too large Load Diff

View File

@@ -5,7 +5,7 @@
#pragma once #pragma once
#include "framework/SubSystemServer.h" #include "framework/SubSystemServer.h"
#include "framework/ow_constants.h"
#include <valijson/adapters/poco_json_adapter.hpp> #include <valijson/adapters/poco_json_adapter.hpp>
#include <valijson/constraints/constraint.hpp> #include <valijson/constraints/constraint.hpp>
#include <valijson/constraints/constraint_visitor.hpp> #include <valijson/constraints/constraint_visitor.hpp>
@@ -17,42 +17,33 @@
namespace OpenWifi { namespace OpenWifi {
class ConfigurationValidator : public SubSystemServer { class ConfigurationValidator : public SubSystemServer {
public: public:
enum class ConfigurationType { AP = 0 , SWITCH = 1};
static auto instance() { static auto instance() {
static auto instance_ = new ConfigurationValidator; static auto instance_ = new ConfigurationValidator;
return instance_; return instance_;
} }
bool Validate(ConfigurationType Type, const std::string &C, std::string &Errors, bool Strict); bool Validate(const std::string &C, std::vector<std::string> &Errors, bool Strict);
int Start() override; int Start() override;
void Stop() override; void Stop() override;
void reinitialize(Poco::Util::Application &self) override; void reinitialize(Poco::Util::Application &self) override;
inline static ConfigurationType GetType(const std::string &type) {
std::string Type = Poco::toLower(type);
if (Type == Platforms::AP)
return ConfigurationType::AP;
if (Type == Platforms::SWITCH)
return ConfigurationType::SWITCH;
return ConfigurationType::AP;
}
private: private:
bool Initialized_ = false; bool Initialized_ = false;
bool Working_ = false; bool Working_ = false;
void Init(); void Init();
std::array<valijson::Schema,2> RootSchema_; std::unique_ptr<valijson::Schema> RootSchema_;
bool SetSchema(ConfigurationType Type, const std::string &SchemaStr); std::unique_ptr<valijson::SchemaParser> SchemaParser_;
std::unique_ptr<valijson::adapters::PocoJsonAdapter> PocoJsonAdapter_;
Poco::JSON::Object::Ptr SchemaDocPtr_;
bool SetSchema(const std::string &SchemaStr);
ConfigurationValidator() ConfigurationValidator()
: SubSystemServer("ConfigValidator", "CFG-VALIDATOR", "config.validator") {} : SubSystemServer("ConfigValidator", "CFG-VALIDATOR", "config.validator") {}
}; };
inline auto ConfigurationValidator() { return ConfigurationValidator::instance(); } inline auto ConfigurationValidator() { return ConfigurationValidator::instance(); }
inline bool ValidateUCentralConfiguration(ConfigurationValidator::ConfigurationType Type, const std::string &C, std::string &Errors, inline bool ValidateUCentralConfiguration(const std::string &C, std::vector<std::string> &Error,
bool strict) { bool strict) {
return ConfigurationValidator::instance()->Validate(Type, C, Errors, strict); return ConfigurationValidator::instance()->Validate(C, Error, strict);
} }
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -9,6 +9,8 @@
namespace OpenWifi { namespace OpenWifi {
EventBusManager::EventBusManager(Poco::Logger &L) : Logger_(L) {}
void EventBusManager::run() { void EventBusManager::run() {
Running_ = true; Running_ = true;
Utils::SetThreadName("fmwk:EventMgr"); Utils::SetThreadName("fmwk:EventMgr");
@@ -16,9 +18,9 @@ namespace OpenWifi {
KafkaManager()->PostMessage(KafkaTopics::SERVICE_EVENTS, MicroServicePrivateEndPoint(), Msg, KafkaManager()->PostMessage(KafkaTopics::SERVICE_EVENTS, MicroServicePrivateEndPoint(), Msg,
false); false);
while (Running_) { while (Running_) {
if(!Poco::Thread::trySleep((unsigned long)MicroServiceDaemonBusTimer())) { Poco::Thread::trySleep((unsigned long)MicroServiceDaemonBusTimer());
if (!Running_)
break; break;
}
Msg = (MicroServiceMakeSystemEventMessage(KafkaTopics::ServiceEvents::EVENT_KEEP_ALIVE)); Msg = (MicroServiceMakeSystemEventMessage(KafkaTopics::ServiceEvents::EVENT_KEEP_ALIVE));
KafkaManager()->PostMessage(KafkaTopics::SERVICE_EVENTS, MicroServicePrivateEndPoint(), KafkaManager()->PostMessage(KafkaTopics::SERVICE_EVENTS, MicroServicePrivateEndPoint(),
Msg, false); Msg, false);
@@ -29,7 +31,7 @@ namespace OpenWifi {
}; };
void EventBusManager::Start() { void EventBusManager::Start() {
poco_information(Logger_, "Starting..."); poco_information(Logger(), "Starting...");
if (KafkaManager()->Enabled()) { if (KafkaManager()->Enabled()) {
Thread_.start(*this); Thread_.start(*this);
} }
@@ -37,11 +39,11 @@ namespace OpenWifi {
void EventBusManager::Stop() { void EventBusManager::Stop() {
if (KafkaManager()->Enabled()) { if (KafkaManager()->Enabled()) {
poco_information(Logger_, "Stopping..."); poco_information(Logger(), "Stopping...");
Running_ = false; Running_ = false;
Thread_.wakeUp(); Thread_.wakeUp();
Thread_.join(); Thread_.join();
poco_information(Logger_, "Stopped..."); poco_information(Logger(), "Stopped...");
} }
} }

View File

@@ -12,16 +12,7 @@ namespace OpenWifi {
class EventBusManager : public Poco::Runnable { class EventBusManager : public Poco::Runnable {
public: public:
EventBusManager() : explicit EventBusManager(Poco::Logger &L);
Logger_(Poco::Logger::create(
"EventBusManager", Poco::Logger::root().getChannel(), Poco::Logger::root().getLevel())) {
}
static auto instance() {
static auto instance_ = new EventBusManager;
return instance_;
}
void run() final; void run() final;
void Start(); void Start();
void Stop(); void Stop();
@@ -33,6 +24,4 @@ namespace OpenWifi {
Poco::Logger &Logger_; Poco::Logger &Logger_;
}; };
inline auto EventBusManager() { return EventBusManager::instance(); }
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -79,10 +79,8 @@ namespace OpenWifi {
Utils::SetThreadName("Kafka:Prod"); Utils::SetThreadName("Kafka:Prod");
cppkafka::Configuration Config( cppkafka::Configuration Config(
{{"client.id", MicroServiceConfigGetString("openwifi.kafka.client.id", "")}, {{"client.id", MicroServiceConfigGetString("openwifi.kafka.client.id", "")},
{"metadata.broker.list",MicroServiceConfigGetString("openwifi.kafka.brokerlist", "")} // , {"metadata.broker.list",
// {"send.buffer.bytes", KafkaManager()->KafkaManagerMaximumPayloadSize() } MicroServiceConfigGetString("openwifi.kafka.brokerlist", "")}});
}
);
AddKafkaSecurity(Config); AddKafkaSecurity(Config);
@@ -107,19 +105,6 @@ namespace OpenWifi {
NewMessage.partition(0); NewMessage.partition(0);
NewMessage.payload(Msg->Payload()); NewMessage.payload(Msg->Payload());
Producer.produce(NewMessage); Producer.produce(NewMessage);
if (Queue_.size() < 100) {
// use flush when internal queue is lightly loaded, i.e. flush after each
// message
Producer.flush();
}
else {
// use poll when internal queue is loaded to allow messages to be sent in
// batches
Producer.poll((std::chrono::milliseconds) 0);
}
}
if (Queue_.size() == 0) {
// message queue is empty, flush all previously sent messages
Producer.flush(); Producer.flush();
} }
} catch (const cppkafka::HandleException &E) { } catch (const cppkafka::HandleException &E) {
@@ -132,7 +117,6 @@ namespace OpenWifi {
} }
Note = Queue_.waitDequeueNotification(); Note = Queue_.waitDequeueNotification();
} }
Producer.flush();
poco_information(Logger_, "Stopped..."); poco_information(Logger_, "Stopped...");
} }
@@ -291,7 +275,6 @@ namespace OpenWifi {
int KafkaManager::Start() { int KafkaManager::Start() {
if (!KafkaEnabled_) if (!KafkaEnabled_)
return 0; return 0;
MaxPayloadSize_ = MicroServiceConfigGetInt("openwifi.kafka.max.payload", 250000);
ConsumerThr_.Start(); ConsumerThr_.Start();
ProducerThr_.Start(); ProducerThr_.Start();
return 0; return 0;

View File

@@ -94,14 +94,11 @@ namespace OpenWifi {
return ConsumerThr_.UnregisterTopicWatcher(Topic,Id); return ConsumerThr_.UnregisterTopicWatcher(Topic,Id);
} }
std::uint64_t KafkaManagerMaximumPayloadSize() const { return MaxPayloadSize_; }
private: private:
bool KafkaEnabled_ = false; bool KafkaEnabled_ = false;
std::string SystemInfoWrapper_; std::string SystemInfoWrapper_;
KafkaProducer ProducerThr_; KafkaProducer ProducerThr_;
KafkaConsumer ConsumerThr_; KafkaConsumer ConsumerThr_;
std::uint64_t MaxPayloadSize_ = 250000;
void PartitionAssignment(const cppkafka::TopicPartitionList &partitions); void PartitionAssignment(const cppkafka::TopicPartitionList &partitions);
void PartitionRevocation(const cppkafka::TopicPartitionList &partitions); void PartitionRevocation(const cppkafka::TopicPartitionList &partitions);

View File

@@ -1,5 +1,4 @@
// //
//
// Created by stephane bourque on 2022-10-26. // Created by stephane bourque on 2022-10-26.
// //
@@ -30,29 +29,13 @@
#include "framework/WebSocketLogger.h" #include "framework/WebSocketLogger.h"
#include "framework/utils.h" #include "framework/utils.h"
#ifdef USE_MEDUSA_CLIENT
#include <medusa/MedusaClient.h>
#endif
namespace OpenWifi { namespace OpenWifi {
static std::string MakeServiceListString(const Types::MicroServiceMetaMap &Services) { void MicroService::Exit(int Reason) { std::exit(Reason); }
std::string SvcList;
for (const auto &Svc : Services) {
if (SvcList.empty())
SvcList = Svc.second.Type;
else
SvcList += ", " + Svc.second.Type;
}
return SvcList;
}
void MicroService::BusMessageReceived([[maybe_unused]] const std::string &Key, void MicroService::BusMessageReceived([[maybe_unused]] const std::string &Key,
const std::string &Payload) { const std::string &Payload) {
std::lock_guard G(InfraMutex_); std::lock_guard G(InfraMutex_);
Poco::Logger &BusLogger = EventBusManager()->Logger();
try { try {
Poco::JSON::Parser P; Poco::JSON::Parser P;
auto Object = P.parse(Payload).extract<Poco::JSON::Object::Ptr>(); auto Object = P.parse(Payload).extract<Poco::JSON::Object::Ptr>();
@@ -72,10 +55,13 @@ namespace OpenWifi {
Object->has(KafkaTopics::ServiceEvents::Fields::KEY)) { Object->has(KafkaTopics::ServiceEvents::Fields::KEY)) {
auto PrivateEndPoint = auto PrivateEndPoint =
Object->get(KafkaTopics::ServiceEvents::Fields::PRIVATE).toString(); Object->get(KafkaTopics::ServiceEvents::Fields::PRIVATE).toString();
if (Event == KafkaTopics::ServiceEvents::EVENT_LEAVE) { if (Event == KafkaTopics::ServiceEvents::EVENT_KEEP_ALIVE &&
Services_.find(PrivateEndPoint) != Services_.end()) {
Services_[PrivateEndPoint].LastUpdate = Utils::Now();
} else if (Event == KafkaTopics::ServiceEvents::EVENT_LEAVE) {
Services_.erase(PrivateEndPoint); Services_.erase(PrivateEndPoint);
poco_information( poco_debug(
BusLogger, logger(),
fmt::format( fmt::format(
"Service {} ID={} leaving system.", "Service {} ID={} leaving system.",
Object->get(KafkaTopics::ServiceEvents::Fields::PRIVATE) Object->get(KafkaTopics::ServiceEvents::Fields::PRIVATE)
@@ -83,7 +69,14 @@ namespace OpenWifi {
ID)); ID));
} else if (Event == KafkaTopics::ServiceEvents::EVENT_JOIN || } else if (Event == KafkaTopics::ServiceEvents::EVENT_JOIN ||
Event == KafkaTopics::ServiceEvents::EVENT_KEEP_ALIVE) { Event == KafkaTopics::ServiceEvents::EVENT_KEEP_ALIVE) {
auto ServiceInfo = Types::MicroServiceMeta{ poco_debug(
logger(),
fmt::format(
"Service {} ID={} joining system.",
Object->get(KafkaTopics::ServiceEvents::Fields::PRIVATE)
.toString(),
ID));
Services_[PrivateEndPoint] = Types::MicroServiceMeta{
.Id = ID, .Id = ID,
.Type = Poco::toLower( .Type = Poco::toLower(
Object->get(KafkaTopics::ServiceEvents::Fields::TYPE) Object->get(KafkaTopics::ServiceEvents::Fields::TYPE)
@@ -101,22 +94,6 @@ namespace OpenWifi {
.toString(), .toString(),
.LastUpdate = Utils::Now()}; .LastUpdate = Utils::Now()};
auto s1 = MakeServiceListString(Services_);
auto PreviousSize = Services_.size();
Services_[PrivateEndPoint] = ServiceInfo;
auto CurrentSize = Services_.size();
if(Event == KafkaTopics::ServiceEvents::EVENT_JOIN) {
if(!s1.empty()) {
poco_information(
BusLogger,
fmt::format(
"Service {} ID={} is joining the system.",
Object
->get(
KafkaTopics::ServiceEvents::Fields::PRIVATE)
.toString(),
ID));
}
std::string SvcList; std::string SvcList;
for (const auto &Svc : Services_) { for (const auto &Svc : Services_) {
if (SvcList.empty()) if (SvcList.empty())
@@ -125,22 +102,12 @@ namespace OpenWifi {
SvcList += ", " + Svc.second.Type; SvcList += ", " + Svc.second.Type;
} }
poco_information( poco_information(
BusLogger, logger(),
fmt::format("Current list of microservices: {}", SvcList)); fmt::format("Current list of microservices: {}", SvcList));
} else if(CurrentSize!=PreviousSize) {
poco_information(
BusLogger,
fmt::format(
"Service {} ID={} is being added back in.",
Object
->get(KafkaTopics::ServiceEvents::Fields::PRIVATE)
.toString(),
ID));
}
} }
} else { } else {
poco_information( poco_error(
BusLogger, logger(),
fmt::format("KAFKA-MSG: invalid event '{}', missing a field.", fmt::format("KAFKA-MSG: invalid event '{}', missing a field.",
Event)); Event));
} }
@@ -151,39 +118,32 @@ namespace OpenWifi {
Object->get(KafkaTopics::ServiceEvents::Fields::TOKEN).toString()); Object->get(KafkaTopics::ServiceEvents::Fields::TOKEN).toString());
#endif #endif
} else { } else {
poco_information( poco_error(
BusLogger, logger(),
fmt::format("KAFKA-MSG: invalid event '{}', missing token", Event)); fmt::format("KAFKA-MSG: invalid event '{}', missing token", Event));
} }
} else { } else {
poco_information(BusLogger, poco_error(logger(),
fmt::format("Unknown Event: {} Source: {}", Event, ID)); fmt::format("Unknown Event: {} Source: {}", Event, ID));
} }
} }
} else { } else {
poco_error(logger(), "Bad bus message.");
std::ostringstream os; std::ostringstream os;
Object->stringify(std::cout); Object->stringify(std::cout);
poco_error(BusLogger, fmt::format("Bad bus message: {}", os.str()));
} }
auto ServiceHint = Services_.begin(); auto i = Services_.begin();
auto now = Utils::Now(); auto now = Utils::Now();
auto si1 = Services_.size(); for (; i != Services_.end();) {
auto ss1 = MakeServiceListString(Services_); if ((now - i->second.LastUpdate) > 60) {
while(ServiceHint!=Services_.end()) { i = Services_.erase(i);
if ((now - ServiceHint->second.LastUpdate) > 120) {
poco_information(BusLogger, fmt::format("ZombieService: Removing service {}, ", ServiceHint->second.PublicEndPoint));
ServiceHint = Services_.erase(ServiceHint);
} else } else
++ServiceHint; ++i;
}
if(Services_.size() != si1) {
auto ss2 = MakeServiceListString(Services_);
poco_information(BusLogger, fmt::format("Current list of microservices: {} -> {}", ss1, ss2));
} }
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
BusLogger.log(E); logger().log(E);
} }
} }
@@ -207,11 +167,9 @@ namespace OpenWifi {
Res.push_back(ServiceRec); Res.push_back(ServiceRec);
} }
return Res; return Res;
} }
void MicroService::LoadConfigurationFile() { void MicroService::LoadConfigurationFile() {
if(ConfigContent_.empty()) {
std::string Location = Poco::Environment::get(DAEMON_CONFIG_ENV_VAR, "."); std::string Location = Poco::Environment::get(DAEMON_CONFIG_ENV_VAR, ".");
ConfigFileName_ = ConfigFileName_ =
ConfigFileName_.empty() ? Location + "/" + DAEMON_PROPERTIES_FILENAME : ConfigFileName_; ConfigFileName_.empty() ? Location + "/" + DAEMON_PROPERTIES_FILENAME : ConfigFileName_;
@@ -225,11 +183,9 @@ namespace OpenWifi {
<< std::endl; << std::endl;
std::exit(Poco::Util::Application::EXIT_CONFIG); std::exit(Poco::Util::Application::EXIT_CONFIG);
} }
// loadConfiguration(ConfigFile.toString());
PropConfigurationFile_ = new Poco::Util::PropertyFileConfiguration(ConfigFile.toString()); PropConfigurationFile_ = new Poco::Util::PropertyFileConfiguration(ConfigFile.toString());
} else {
std::istringstream is(ConfigContent_);
PropConfigurationFile_ = new Poco::Util::PropertyFileConfiguration(is);
}
configPtr()->addWriteable(PropConfigurationFile_, PRIO_DEFAULT); configPtr()->addWriteable(PropConfigurationFile_, PRIO_DEFAULT);
} }
@@ -432,13 +388,11 @@ namespace OpenWifi {
void DaemonPostInitialization(Poco::Util::Application &self); void DaemonPostInitialization(Poco::Util::Application &self);
void MicroService::StartEverything(Poco::Util::Application &self) { void MicroService::initialize(Poco::Util::Application &self) {
// add the default services
LoadConfigurationFile(); LoadConfigurationFile();
InitializeLoggingSystem(); InitializeLoggingSystem();
static bool InitializedBaseService=false;
if(!InitializedBaseService) {
InitializedBaseService = true;
SubSystems_.push_back(KafkaManager()); SubSystems_.push_back(KafkaManager());
SubSystems_.push_back(ALBHealthCheckServer()); SubSystems_.push_back(ALBHealthCheckServer());
SubSystems_.push_back(RESTAPI_ExtServer()); SubSystems_.push_back(RESTAPI_ExtServer());
@@ -446,13 +400,11 @@ namespace OpenWifi {
#ifndef TIP_SECURITY_SERVICE #ifndef TIP_SECURITY_SERVICE
SubSystems_.push_back(AuthClient()); SubSystems_.push_back(AuthClient());
#endif #endif
Poco::Net::initializeSSL(); Poco::Net::initializeSSL();
Poco::Net::HTTPStreamFactory::registerFactory(); Poco::Net::HTTPStreamFactory::registerFactory();
Poco::Net::HTTPSStreamFactory::registerFactory(); Poco::Net::HTTPSStreamFactory::registerFactory();
Poco::Net::FTPStreamFactory::registerFactory(); Poco::Net::FTPStreamFactory::registerFactory();
Poco::Net::FTPSStreamFactory::registerFactory(); Poco::Net::FTPSStreamFactory::registerFactory();
}
Poco::File DataDir(ConfigPath("openwifi.system.data")); Poco::File DataDir(ConfigPath("openwifi.system.data"));
DataDir_ = DataDir.path(); DataDir_ = DataDir.path();
@@ -460,7 +412,7 @@ namespace OpenWifi {
try { try {
DataDir.createDirectory(); DataDir.createDirectory();
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger_.log(E); logger().log(E);
} }
} }
WWWAssetsDir_ = ConfigPath("openwifi.restapi.wwwassets", ""); WWWAssetsDir_ = ConfigPath("openwifi.restapi.wwwassets", "");
@@ -481,22 +433,6 @@ namespace OpenWifi {
KafkaManager()->RegisterTopicWatcher(KafkaTopics::SERVICE_EVENTS, F); KafkaManager()->RegisterTopicWatcher(KafkaTopics::SERVICE_EVENTS, F);
} }
void MicroService::StopEverything([[maybe_unused]] Poco::Util::Application &self) {
LoadConfigurationFile();
InitializeLoggingSystem();
Types::TopicNotifyFunction F = [this](const std::string &Key, const std::string &Payload) {
this->BusMessageReceived(Key, Payload);
};
KafkaManager()->RegisterTopicWatcher(KafkaTopics::SERVICE_EVENTS, F);
}
void MicroService::initialize([[maybe_unused]] Poco::Util::Application &self) {
#ifndef USE_MEDUSA_CLIENT
StartEverything(self);
#endif
}
void MicroService::uninitialize() { void MicroService::uninitialize() {
// add your own uninitialization code here // add your own uninitialization code here
ServerApplication::uninitialize(); ServerApplication::uninitialize();
@@ -594,12 +530,14 @@ namespace OpenWifi {
for (auto i : SubSystems_) { for (auto i : SubSystems_) {
i->Start(); i->Start();
} }
EventBusManager()->Start(); EventBusManager_ = std::make_unique<EventBusManager>(Poco::Logger::create(
"EventBusManager", Poco::Logger::root().getChannel(), Poco::Logger::root().getLevel()));
EventBusManager_->Start();
} }
void MicroService::StopSubSystemServers() { void MicroService::StopSubSystemServers() {
AddActivity("Stopping"); AddActivity("Stopping");
EventBusManager()->Stop(); EventBusManager_->Stop();
for (auto i = SubSystems_.rbegin(); i != SubSystems_.rend(); ++i) { for (auto i = SubSystems_.rbegin(); i != SubSystems_.rend(); ++i) {
(*i)->Stop(); (*i)->Stop();
} }
@@ -759,7 +697,7 @@ namespace OpenWifi {
auto APIKEY = Request.get("X-API-KEY"); auto APIKEY = Request.get("X-API-KEY");
return APIKEY == MyHash_; return APIKEY == MyHash_;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger_.log(E); logger().log(E);
} }
return false; return false;
} }
@@ -780,8 +718,6 @@ namespace OpenWifi {
MicroServiceErrorHandler ErrorHandler(*this); MicroServiceErrorHandler ErrorHandler(*this);
Poco::ErrorHandler::set(&ErrorHandler); Poco::ErrorHandler::set(&ErrorHandler);
Args_ = args;
if (!HelpRequested_) { if (!HelpRequested_) {
SavePID(); SavePID();
@@ -797,18 +733,11 @@ namespace OpenWifi {
poco_information(logger, "Starting as a daemon."); poco_information(logger, "Starting as a daemon.");
} }
#ifdef USE_MEDUSA_CLIENT
MedusaClient::instance()->SetSubSystems(SubSystems_);
MedusaClient::instance()->Start();
waitForTerminationRequest();
MedusaClient::instance()->Stop();
#else
poco_information(logger, fmt::format("System ID set to {}", ID_)); poco_information(logger, fmt::format("System ID set to {}", ID_));
StartSubSystemServers(); StartSubSystemServers();
waitForTerminationRequest(); waitForTerminationRequest();
StopSubSystemServers(); StopSubSystemServers();
logger.notice(fmt::format("Stopped {}...", DAEMON_APP_NAME)); logger.notice(fmt::format("Stopped {}...", DAEMON_APP_NAME));
#endif
} }
return Application::EXIT_OK; return Application::EXIT_OK;

View File

@@ -55,6 +55,9 @@ namespace OpenWifi {
#include "nlohmann/json.hpp" #include "nlohmann/json.hpp"
#include "ow_version.h" #include "ow_version.h"
#define _OWDEBUG_ std::cout << __FILE__ << ":" << __LINE__ << std::endl;
// #define _OWDEBUG_ Logger().debug(Poco::format("%s: %lu",__FILE__,__LINE__));
namespace OpenWifi { namespace OpenWifi {
class MicroService : public Poco::Util::ServerApplication { class MicroService : public Poco::Util::ServerApplication {
@@ -67,6 +70,7 @@ namespace OpenWifi {
SubSystems_(std::move(Subsystems)), Logger_(Poco::Logger::get("FRAMEWORK")) { SubSystems_(std::move(Subsystems)), Logger_(Poco::Logger::get("FRAMEWORK")) {
instance_ = this; instance_ = this;
RandomEngine_.seed(std::chrono::steady_clock::now().time_since_epoch().count()); RandomEngine_.seed(std::chrono::steady_clock::now().time_since_epoch().count());
// Logger_ = Poco::Logger::root().get("BASE-SVC");
} }
inline static const char *ExtraConfigurationFilename = "/configuration_override.json"; inline static const char *ExtraConfigurationFilename = "/configuration_override.json";
@@ -88,7 +92,7 @@ namespace OpenWifi {
inline uint64_t DaemonBusTimer() const { return DAEMON_BUS_TIMER; }; inline uint64_t DaemonBusTimer() const { return DAEMON_BUS_TIMER; };
[[nodiscard]] const std::string &AppName() { return DAEMON_APP_NAME; } [[nodiscard]] const std::string &AppName() { return DAEMON_APP_NAME; }
static inline uint64_t GetPID() { return Poco::Process::id(); }; static inline uint64_t GetPID() { return Poco::Process::id(); };
[[nodiscard]] inline std::string GetPublicAPIEndPoint() const { [[nodiscard]] inline const std::string GetPublicAPIEndPoint() {
return MyPublicEndPoint_ + "/api/v1"; return MyPublicEndPoint_ + "/api/v1";
}; };
[[nodiscard]] inline const std::string &GetUIURI() const { return UIURI_; }; [[nodiscard]] inline const std::string &GetUIURI() const { return UIURI_; };
@@ -103,8 +107,7 @@ namespace OpenWifi {
} }
static MicroService &instance() { return *instance_; } static MicroService &instance() { return *instance_; }
inline void Exit(int Reason) { std::exit(Reason); } inline void Exit(int Reason);
void BusMessageReceived(const std::string &Key, const std::string &Payload); void BusMessageReceived(const std::string &Key, const std::string &Payload);
Types::MicroServiceMetaVec GetServices(const std::string &Type); Types::MicroServiceMetaVec GetServices(const std::string &Type);
Types::MicroServiceMetaVec GetServices(); Types::MicroServiceMetaVec GetServices();
@@ -112,8 +115,6 @@ namespace OpenWifi {
void Reload(); void Reload();
void LoadMyConfig(); void LoadMyConfig();
void initialize(Poco::Util::Application &self) override; void initialize(Poco::Util::Application &self) override;
void StartEverything(Poco::Util::Application &self);
void StopEverything(Poco::Util::Application &self);
void uninitialize() override; void uninitialize() override;
void reinitialize(Poco::Util::Application &self) override; void reinitialize(Poco::Util::Application &self) override;
void defineOptions(Poco::Util::OptionSet &options) override; void defineOptions(Poco::Util::OptionSet &options) override;
@@ -131,7 +132,7 @@ namespace OpenWifi {
void Reload(const std::string &Sub); void Reload(const std::string &Sub);
Types::StringVec GetSubSystems() const; Types::StringVec GetSubSystems() const;
Types::StringPairVec GetLogLevels(); Types::StringPairVec GetLogLevels();
static const Types::StringVec &GetLogLevelNames(); const Types::StringVec &GetLogLevelNames();
uint64_t ConfigGetInt(const std::string &Key, uint64_t Default); uint64_t ConfigGetInt(const std::string &Key, uint64_t Default);
uint64_t ConfigGetInt(const std::string &Key); uint64_t ConfigGetInt(const std::string &Key);
uint64_t ConfigGetBool(const std::string &Key, bool Default); uint64_t ConfigGetBool(const std::string &Key, bool Default);
@@ -165,25 +166,12 @@ namespace OpenWifi {
const std::string &FormatterPattern, const std::string &FormatterPattern,
const std::string &root_env_var); const std::string &root_env_var);
inline bool AllowExternalMicroServices() const { return AllowExternalMicroServices_; } inline bool AllowExternalMicroServices() const { return AllowExternalMicroServices_; }
const ArgVec &Args() const { return Args_; }
inline void SetConfigContent(const std::string &Content) { ConfigContent_ = Content; }
inline std::optional<OpenWifi::Types::MicroServiceMeta> GetPrivateEndPointServiceKey( const std::string & ServicePrivateEndPoint ) {
std::lock_guard G(InfraMutex_);
auto K = Services_.find(ServicePrivateEndPoint);
if(K==end(Services_)) {
return std::nullopt;
}
return K->second;
}
private: private:
static MicroService *instance_; static MicroService *instance_;
bool HelpRequested_ = false; bool HelpRequested_ = false;
std::string LogDir_; std::string LogDir_;
std::string ConfigFileName_; std::string ConfigFileName_;
std::string ConfigContent_;
uint64_t ID_ = 1; uint64_t ID_ = 1;
Poco::SharedPtr<Poco::Crypto::RSAKey> AppKey_; Poco::SharedPtr<Poco::Crypto::RSAKey> AppKey_;
bool DebugMode_ = false; bool DebugMode_ = false;
@@ -213,7 +201,7 @@ namespace OpenWifi {
Poco::JWT::Signer Signer_; Poco::JWT::Signer Signer_;
Poco::Logger &Logger_; Poco::Logger &Logger_;
Poco::ThreadPool TimerPool_{"timer:pool", 2, 32}; Poco::ThreadPool TimerPool_{"timer:pool", 2, 32};
ArgVec Args_; std::unique_ptr<EventBusManager> EventBusManager_;
}; };
inline MicroService *MicroService::instance_ = nullptr; inline MicroService *MicroService::instance_ = nullptr;

View File

@@ -133,8 +133,4 @@ namespace OpenWifi {
return MicroService::instance().Hash(); return MicroService::instance().Hash();
} }
std::optional<OpenWifi::Types::MicroServiceMeta> MicroServicePrivateAccessKey(const std::string &servicePrivateEndPoint) {
return MicroService::instance().GetPrivateEndPointServiceKey(servicePrivateEndPoint);
}
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -23,8 +23,6 @@ namespace OpenWifi {
std::string MicroServiceConfigGetString(const std::string &Key, std::string MicroServiceConfigGetString(const std::string &Key,
const std::string &DefaultValue); const std::string &DefaultValue);
std::string MicroServiceAccessKey(); std::string MicroServiceAccessKey();
std::optional<OpenWifi::Types::MicroServiceMeta> MicroServicePrivateAccessKey(const std::string &servicePrivateEndPoint);
bool MicroServiceConfigGetBool(const std::string &Key, bool DefaultValue); bool MicroServiceConfigGetBool(const std::string &Key, bool DefaultValue);
std::uint64_t MicroServiceConfigGetInt(const std::string &Key, std::uint64_t DefaultValue); std::uint64_t MicroServiceConfigGetInt(const std::string &Key, std::uint64_t DefaultValue);
std::string MicroServicePrivateEndPoint(); std::string MicroServicePrivateEndPoint();

View File

@@ -431,11 +431,6 @@ namespace OpenWifi {
} }
} }
inline void Accepted() {
PrepareResponse(Poco::Net::HTTPResponse::HTTP_ACCEPTED);
Response->send();
}
inline void SendCompressedTarFile(const std::string &FileName, const std::string &Content) { inline void SendCompressedTarFile(const std::string &FileName, const std::string &Content) {
Response->setStatus(Poco::Net::HTTPResponse::HTTPStatus::HTTP_OK); Response->setStatus(Poco::Net::HTTPResponse::HTTPStatus::HTTP_OK);
SetCommonHeaders(); SetCommonHeaders();
@@ -557,8 +552,8 @@ namespace OpenWifi {
inline bool IsAuthorized(bool &Expired, bool &Contacted, bool SubOnly = false); inline bool IsAuthorized(bool &Expired, bool &Contacted, bool SubOnly = false);
inline void ReturnObject(Poco::JSON::Object &Object, Poco::Net::HTTPResponse::HTTPStatus Status = Poco::Net::HTTPResponse::HTTP_OK) { inline void ReturnObject(Poco::JSON::Object &Object) {
PrepareResponse(Status); PrepareResponse();
if (Request != nullptr) { if (Request != nullptr) {
// can we compress ??? // can we compress ???
auto AcceptedEncoding = Request->find("Accept-Encoding"); auto AcceptedEncoding = Request->find("Accept-Encoding");

View File

@@ -47,8 +47,6 @@ namespace OpenWifi {
} }
Poco::Data::SessionPool &Pool() { return *Pool_; }
private: private:
inline int Setup_SQLite(); inline int Setup_SQLite();
inline int Setup_MySQL(); inline int Setup_MySQL();

View File

@@ -68,16 +68,6 @@ namespace OpenWifi {
Context->addCertificateAuthority(Issuing); Context->addCertificateAuthority(Issuing);
} }
if (!client_cas_.empty()) {
// add certificates specified in clientcas
std::vector<Poco::Crypto::X509Certificate> Certs =
Poco::Net::X509Certificate::readPEM(client_cas_);
for (const auto &cert : Certs) {
Context->addChainCertificate(cert);
Context->addCertificateAuthority(cert);
}
}
Poco::Crypto::RSAKey Key("", key_file_, key_file_password_); Poco::Crypto::RSAKey Key("", key_file_, key_file_password_);
Context->usePrivateKey(Key); Context->usePrivateKey(Key);

View File

@@ -45,7 +45,6 @@ namespace OpenWifi {
[[nodiscard]] inline auto KeyFile() const { return key_file_; }; [[nodiscard]] inline auto KeyFile() const { return key_file_; };
[[nodiscard]] inline auto CertFile() const { return cert_file_; }; [[nodiscard]] inline auto CertFile() const { return cert_file_; };
[[nodiscard]] inline auto RootCA() const { return root_ca_; }; [[nodiscard]] inline auto RootCA() const { return root_ca_; };
[[nodiscard]] inline auto ClientCas() const { return client_cas_; };
[[nodiscard]] inline auto KeyFilePassword() const { return key_file_password_; }; [[nodiscard]] inline auto KeyFilePassword() const { return key_file_password_; };
[[nodiscard]] inline auto IssuerCertFile() const { return issuer_cert_file_; }; [[nodiscard]] inline auto IssuerCertFile() const { return issuer_cert_file_; };
[[nodiscard]] inline auto Name() const { return name_; }; [[nodiscard]] inline auto Name() const { return name_; };

View File

@@ -58,9 +58,11 @@ namespace OpenWifi {
void UI_WebSocketClientServer::run() { void UI_WebSocketClientServer::run() {
Running_ = true; Running_ = true;
while (Running_) { while (Running_) {
if(!Poco::Thread::trySleep(2000)) { Poco::Thread::trySleep(2000);
if (!Running_)
break; break;
}
std::lock_guard G(LocalMutex_); std::lock_guard G(LocalMutex_);
for (const auto i : ToBeRemoved_) { for (const auto i : ToBeRemoved_) {
// std::cout << "Erasing old WS UI connection..." << std::endl; // std::cout << "Erasing old WS UI connection..." << std::endl;

View File

@@ -7,59 +7,57 @@
#include <vector> #include <vector>
#include <string> #include <string>
#include "ow_constants.h"
namespace OpenWifi { namespace OpenWifi {
inline const std::vector<std::pair<std::string, std::string>> DefaultDeviceTypeList{ inline const std::vector<std::pair<std::string, std::string>> DefaultDeviceTypeList{
{"actiontec_web7200", Platforms::AP}, {"actiontec_web7200", "AP"},
{"cig_wf186w", Platforms::AP}, {"cig_wf186w", "AP"},
{"cig_wf188n", Platforms::AP}, {"cig_wf188n", "AP"},
{"cig_wf194c4", Platforms::AP}, {"cig_wf194c4", "AP"},
{"cig_wf196", Platforms::AP}, {"cig_wf196", "AP"},
{"cig_wf196-ca", Platforms::AP}, {"cig_wf196-ca", "AP"},
{"cig_wf196-ca-ath12", Platforms::AP}, {"cig_wf196-ca-ath12", "AP"},
{"cig_wf196-us", Platforms::AP}, {"cig_wf196-us", "AP"},
{"cig_wf610d", Platforms::AP}, {"cig_wf610d", "AP"},
{"cig_wf660a", Platforms::AP}, {"cig_wf660a", "AP"},
{"cig_wf808", Platforms::AP}, {"cig_wf808", "AP"},
{"cybertan_eww622-a1", Platforms::AP}, {"cybertan_eww622-a1", "AP"},
{"edgecore_eap101", Platforms::AP}, {"edgecore_eap101", "AP"},
{"edgecore_eap101-ath12", Platforms::AP}, {"edgecore_eap101-ath12", "AP"},
{"edgecore_eap102", Platforms::AP}, {"edgecore_eap102", "AP"},
{"edgecore_eap104", Platforms::AP}, {"edgecore_eap104", "AP"},
{"edgecore_eap104-ath12", Platforms::AP}, {"edgecore_eap104-ath12", "AP"},
{"edgecore_ecs4100-12ph", Platforms::AP}, {"edgecore_ecs4100-12ph", "AP"},
{"edgecore_ecw5211", Platforms::AP}, {"edgecore_ecw5211", "AP"},
{"edgecore_ecw5410", Platforms::AP}, {"edgecore_ecw5410", "AP"},
{"edgecore_oap100", Platforms::AP}, {"edgecore_oap100", "AP"},
{"edgecore_spw2ac1200", Platforms::SWITCH}, {"edgecore_spw2ac1200", "SWITCH"},
{"edgecore_spw2ac1200-lan-poe", Platforms::SWITCH}, {"edgecore_spw2ac1200-lan-poe", "SWITCH"},
{"edgecore_ssw2ac2600", Platforms::SWITCH}, {"edgecore_ssw2ac2600", "SWITCH"},
{"hfcl_ion4", Platforms::AP}, {"hfcl_ion4", "AP"},
{"hfcl_ion4x", Platforms::AP}, {"hfcl_ion4x", "AP"},
{"hfcl_ion4x_2", Platforms::AP}, {"hfcl_ion4x_2", "AP"},
{"hfcl_ion4xe", Platforms::AP}, {"hfcl_ion4xe", "AP"},
{"hfcl_ion4xi", Platforms::AP}, {"hfcl_ion4xi", "AP"},
{"indio_um-305ac", Platforms::AP}, {"indio_um-305ac", "AP"},
{"indio_um-305ax", Platforms::AP}, {"indio_um-305ax", "AP"},
{"indio_um-310ax-v1", Platforms::AP}, {"indio_um-310ax-v1", "AP"},
{"indio_um-325ac", Platforms::AP}, {"indio_um-325ac", "AP"},
{"indio_um-510ac-v3", Platforms::AP}, {"indio_um-510ac-v3", "AP"},
{"indio_um-510axm-v1", Platforms::AP}, {"indio_um-510axm-v1", "AP"},
{"indio_um-510axp-v1", Platforms::AP}, {"indio_um-510axp-v1", "AP"},
{"indio_um-550ac", Platforms::AP}, {"indio_um-550ac", "AP"},
{"linksys_e8450-ubi", Platforms::AP}, {"linksys_e8450-ubi", "AP"},
{"linksys_ea6350-v4", Platforms::AP}, {"linksys_ea6350-v4", "AP"},
{"linksys_ea8300", Platforms::AP}, {"linksys_ea8300", "AP"},
{"liteon_wpx8324", Platforms::AP}, {"liteon_wpx8324", "AP"},
{"meshpp_s618_cp01", Platforms::AP}, {"meshpp_s618_cp01", "AP"},
{"meshpp_s618_cp03", Platforms::AP}, {"meshpp_s618_cp03", "AP"},
{"udaya_a5-id2", Platforms::AP}, {"udaya_a5-id2", "AP"},
{"wallys_dr40x9", Platforms::AP}, {"wallys_dr40x9", "AP"},
{"wallys_dr6018", Platforms::AP}, {"wallys_dr6018", "AP"},
{"wallys_dr6018_v4", Platforms::AP}, {"wallys_dr6018_v4", "AP"},
{"x64_vm", Platforms::AP}, {"x64_vm", "AP"},
{"yuncore_ax840", Platforms::AP}, {"yuncore_ax840", "AP"},
{"yuncore_fap640", Platforms::AP}, {"yuncore_fap640", "AP"},
{"yuncore_fap650", Platforms::AP}}; {"yuncore_fap650", "AP"}};
} }

View File

@@ -576,8 +576,8 @@ namespace ORM {
bool UpdateRecord(field_name_t FieldName, const T &Value, const RecordType &R) { bool UpdateRecord(field_name_t FieldName, const T &Value, const RecordType &R) {
try { try {
assert(ValidFieldName(FieldName)); assert(ValidFieldName(FieldName));
Poco::Data::Session Session = Pool_.get(); Poco::Data::Session Session = Pool_.get();
Session.begin();
Poco::Data::Statement Update(Session); Poco::Data::Statement Update(Session);
RecordTuple RT; RecordTuple RT;
@@ -593,7 +593,6 @@ namespace ORM {
Update.execute(); Update.execute();
if (Cache_) if (Cache_)
Cache_->UpdateCache(R); Cache_->UpdateCache(R);
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger_.log(E); Logger_.log(E);
@@ -663,7 +662,6 @@ namespace ORM {
assert(ValidFieldName(FieldName)); assert(ValidFieldName(FieldName));
Poco::Data::Session Session = Pool_.get(); Poco::Data::Session Session = Pool_.get();
Session.begin();
Poco::Data::Statement Delete(Session); Poco::Data::Statement Delete(Session);
std::string St = "delete from " + TableName_ + " where " + FieldName + "=?"; std::string St = "delete from " + TableName_ + " where " + FieldName + "=?";
@@ -673,7 +671,6 @@ namespace ORM {
Delete.execute(); Delete.execute();
if (Cache_) if (Cache_)
Cache_->Delete(FieldName, Value); Cache_->Delete(FieldName, Value);
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger_.log(E); Logger_.log(E);
@@ -685,13 +682,11 @@ namespace ORM {
try { try {
assert(!WhereClause.empty()); assert(!WhereClause.empty());
Poco::Data::Session Session = Pool_.get(); Poco::Data::Session Session = Pool_.get();
Session.begin();
Poco::Data::Statement Delete(Session); Poco::Data::Statement Delete(Session);
std::string St = "delete from " + TableName_ + " where " + WhereClause; std::string St = "delete from " + TableName_ + " where " + WhereClause;
Delete << St; Delete << St;
Delete.execute(); Delete.execute();
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger_.log(E); Logger_.log(E);

View File

@@ -565,7 +565,6 @@ namespace OpenWifi::RESTAPI::Protocol {
static const char *TRANSFER = "transfer"; static const char *TRANSFER = "transfer";
static const char *CERTUPDATE = "certupdate"; static const char *CERTUPDATE = "certupdate";
static const char *POWERCYCLE = "powercycle";
static const char *RRM = "rrm"; static const char *RRM = "rrm";
static const char *REQUIREMENTS = "requirements"; static const char *REQUIREMENTS = "requirements";
@@ -580,10 +579,6 @@ namespace OpenWifi::RESTAPI::Protocol {
static const char *INTERVAL = "interval"; static const char *INTERVAL = "interval";
static const char *UI = "UI"; static const char *UI = "UI";
static const char *BANDWIDTH = "bandwidth"; static const char *BANDWIDTH = "bandwidth";
static const char *FIXEDCONFIG = "fixedconfig";
static const char *CABLEDIAGNOSTICS = "cable-diagnostics";
static const char *REENROLL = "reenroll";
} // namespace OpenWifi::RESTAPI::Protocol } // namespace OpenWifi::RESTAPI::Protocol
namespace OpenWifi::uCentralProtocol { namespace OpenWifi::uCentralProtocol {
@@ -612,7 +607,6 @@ namespace OpenWifi::uCentralProtocol {
static const char *CFGPENDING = "cfgpending"; static const char *CFGPENDING = "cfgpending";
static const char *RECOVERY = "recovery"; static const char *RECOVERY = "recovery";
static const char *COMPRESS_64 = "compress_64"; static const char *COMPRESS_64 = "compress_64";
static const char *COMPRESS_SZ = "compress_sz";
static const char *CAPABILITIES = "capabilities"; static const char *CAPABILITIES = "capabilities";
static const char *REQUEST_UUID = "request_uuid"; static const char *REQUEST_UUID = "request_uuid";
static const char *SANITY = "sanity"; static const char *SANITY = "sanity";
@@ -693,15 +687,9 @@ namespace OpenWifi::uCentralProtocol {
static const char *TRANSFER = "transfer"; static const char *TRANSFER = "transfer";
static const char *CERTUPDATE = "certupdate"; static const char *CERTUPDATE = "certupdate";
static const char *POWERCYCLE = "powercycle";
static const char *RRM = "rrm"; static const char *RRM = "rrm";
static const char *ACTIONS = "actions"; static const char *ACTIONS = "actions";
static const char *FIXEDCONFIG = "fixedconfig";
static const char *CABLEDIAGNOSTICS = "cable-diagnostics";
static const char *REENROLL = "reenroll";
} // namespace OpenWifi::uCentralProtocol } // namespace OpenWifi::uCentralProtocol
namespace OpenWifi::uCentralProtocol::Events { namespace OpenWifi::uCentralProtocol::Events {
@@ -797,10 +785,6 @@ namespace OpenWifi::APCommands {
rrm, rrm,
certupdate, certupdate,
transfer, transfer,
powercycle,
fixedconfig,
cablediagnostics,
reenroll,
unknown unknown
}; };
@@ -815,9 +799,7 @@ namespace OpenWifi::APCommands {
RESTAPI::Protocol::EVENTQUEUE, RESTAPI::Protocol::TELEMETRY, RESTAPI::Protocol::EVENTQUEUE, RESTAPI::Protocol::TELEMETRY,
RESTAPI::Protocol::PING, RESTAPI::Protocol::SCRIPT, RESTAPI::Protocol::PING, RESTAPI::Protocol::SCRIPT,
RESTAPI::Protocol::RRM, RESTAPI::Protocol::CERTUPDATE, RESTAPI::Protocol::RRM, RESTAPI::Protocol::CERTUPDATE,
RESTAPI::Protocol::TRANSFER, RESTAPI::Protocol::POWERCYCLE, RESTAPI::Protocol::TRANSFER
RESTAPI::Protocol::FIXEDCONFIG, RESTAPI::Protocol::CABLEDIAGNOSTICS,
RESTAPI::Protocol::REENROLL
}; };
inline const char *to_string(Commands Cmd) { return uCentralAPCommands[(uint8_t)Cmd]; } inline const char *to_string(Commands Cmd) { return uCentralAPCommands[(uint8_t)Cmd]; }
@@ -846,11 +828,6 @@ namespace OpenWifi::Provisioning::DeviceClass {
} // namespace OpenWifi::Provisioning::DeviceClass } // namespace OpenWifi::Provisioning::DeviceClass
namespace OpenWifi::Platforms {
static const std::string AP = "ap";
static const std::string SWITCH = "switch";
}
#if defined(__GNUC__) #if defined(__GNUC__)
#pragma GCC diagnostic pop #pragma GCC diagnostic pop
#endif #endif

View File

@@ -590,26 +590,6 @@ namespace OpenWifi::Utils {
return false; return false;
} }
//
// Compress given data using utility function and encode it in base64 format.
//
bool CompressAndEncodeBase64(const std::string& UnCompressedData, std::string& CompressedBase64Data) {
unsigned long CompressedDataSize = UnCompressedData.size();
std::vector<Bytef> CompressedData(CompressedDataSize);
auto status = compress(&CompressedData[0], &CompressedDataSize,
(Bytef*) UnCompressedData.c_str(), UnCompressedData.size());
if (status == Z_OK) {
CompressedBase64Data = OpenWifi::Utils::base64encode(&CompressedData[0], CompressedDataSize);
}
else {
// failed to compress data
return false;
}
return true;
}
bool IsAlphaNumeric(const std::string &s) { bool IsAlphaNumeric(const std::string &s) {
return std::all_of(s.begin(), s.end(), [](char c) -> bool { return isalnum(c); }); return std::all_of(s.begin(), s.end(), [](char c) -> bool { return isalnum(c); });
} }

View File

@@ -126,6 +126,20 @@ namespace OpenWifi::Utils {
[[nodiscard]] std::uint64_t ConvertDate(const std::string &d); [[nodiscard]] std::uint64_t ConvertDate(const std::string &d);
[[nodiscard]] inline uint8_t CalculateMacAddressHash(std::uint64_t value) {
uint8_t hash = 0, i=6;
while(i) {
hash ^= (value & 0xFF) + 1;
value >>= 8;
--i;
}
return hash;
}
[[nodiscard]] inline uint8_t CalculateMacAddressHash(const std::string & value) {
return CalculateMacAddressHash(MACToInt(value));
}
template <typename T> std::string int_to_hex(T i) { template <typename T> std::string int_to_hex(T i) {
std::stringstream stream; std::stringstream stream;
stream << std::setfill('0') << std::setw(12) << std::hex << i; stream << std::setfill('0') << std::setw(12) << std::hex << i;
@@ -151,8 +165,6 @@ namespace OpenWifi::Utils {
bool ExtractBase64CompressedData(const std::string &CompressedData, bool ExtractBase64CompressedData(const std::string &CompressedData,
std::string &UnCompressedData, uint64_t compress_sz); std::string &UnCompressedData, uint64_t compress_sz);
bool CompressAndEncodeBase64(const std::string& UnCompressedData, std::string& CompressedData);
inline bool match(const char* first, const char* second) inline bool match(const char* first, const char* second)
{ {
// If we reach at the end of both strings, we are done // If we reach at the end of both strings, we are done
@@ -318,90 +330,5 @@ namespace OpenWifi::Utils {
uint32_t Port; uint32_t Port;
}; };
class CompressedString {
public:
CompressedString() {
DecompressedSize_ = 0;
};
explicit CompressedString(const std::string &Data) : DecompressedSize_(Data.size()) {
CompressIt(Data);
}
CompressedString(const CompressedString &Data) {
this->DecompressedSize_ = Data.DecompressedSize_;
this->CompressedData_ = Data.CompressedData_;
}
CompressedString& operator=(const CompressedString& rhs) {
if (this != &rhs) {
this->DecompressedSize_ = rhs.DecompressedSize_;
this->CompressedData_ = rhs.CompressedData_;
}
return *this;
}
CompressedString& operator=(CompressedString&& rhs) {
if (this != &rhs) {
this->DecompressedSize_ = rhs.DecompressedSize_;
this->CompressedData_ = rhs.CompressedData_;
}
return *this;
}
~CompressedString() = default;
operator std::string() const {
return DecompressIt();
}
CompressedString &operator=(const std::string &Data) {
DecompressedSize_ = Data.size();
CompressIt(Data);
return *this;
}
auto CompressedSize() const { return CompressedData_.size(); }
auto DecompressedSize() const { return DecompressedSize_; }
private:
std::string CompressedData_;
std::size_t DecompressedSize_;
inline void CompressIt(const std::string &Data) {
z_stream strm; // = {0};
CompressedData_.resize(Data.size());
strm.next_in = (Bytef *)Data.data();
strm.avail_in = Data.size();
strm.next_out = (Bytef *)CompressedData_.data();
strm.avail_out = Data.size();
strm.zalloc = Z_NULL;
strm.zfree = Z_NULL;
strm.opaque = Z_NULL;
deflateInit2(&strm, Z_DEFAULT_COMPRESSION, Z_DEFLATED, 15 + 16, 8, Z_DEFAULT_STRATEGY);
deflate(&strm, Z_FINISH);
deflateEnd(&strm);
CompressedData_.resize(strm.total_out);
}
[[nodiscard]] std::string DecompressIt() const {
std::string Result;
if(DecompressedSize_!=0) {
Result.resize(DecompressedSize_);
z_stream strm ; //= {0};
strm.next_in = (Bytef *)CompressedData_.data();
strm.avail_in = CompressedData_.size();
strm.next_out = (Bytef *)Result.data();
strm.avail_out = Result.size();
strm.zalloc = Z_NULL;
strm.zfree = Z_NULL;
strm.opaque = Z_NULL;
inflateInit2(&strm, 15 + 32);
inflate(&strm, Z_FINISH);
inflateEnd(&strm);
}
return Result;
}
};
} // namespace OpenWifi::Utils } // namespace OpenWifi::Utils

View File

@@ -14,7 +14,6 @@
#include "nlohmann/json.hpp" #include "nlohmann/json.hpp"
#include "Poco/NObserver.h" #include "Poco/NObserver.h"
#include <Poco/Net/Context.h>
#include "Poco/Net/SocketNotification.h" #include "Poco/Net/SocketNotification.h"
#include "Poco/Net/NetException.h" #include "Poco/Net/NetException.h"
#include "Poco/Net/WebSocketImpl.h" #include "Poco/Net/WebSocketImpl.h"
@@ -72,7 +71,6 @@ namespace OpenWifi {
const auto &RootCas = const auto &RootCas =
MicroServiceConfigPath("ucentral.websocket.host.0.rootca", ""); MicroServiceConfigPath("ucentral.websocket.host.0.rootca", "");
const auto &Cas = MicroServiceConfigPath("ucentral.websocket.host.0.cas", ""); const auto &Cas = MicroServiceConfigPath("ucentral.websocket.host.0.cas", "");
const auto &ClientCasFile = MicroServiceConfigPath("ucentral.websocket.host.0.clientcas", "");
Poco::Net::Context::Params P; Poco::Net::Context::Params P;
@@ -88,7 +86,6 @@ namespace OpenWifi {
Poco::Crypto::X509Certificate Cert(CertFileName); Poco::Crypto::X509Certificate Cert(CertFileName);
Poco::Crypto::X509Certificate Root(RootCaFileName); Poco::Crypto::X509Certificate Root(RootCaFileName);
Poco::Crypto::X509Certificate Issuing(IssuerFileName); Poco::Crypto::X509Certificate Issuing(IssuerFileName);
std::vector<Poco::Crypto::X509Certificate> ClientCasCerts;
Poco::Crypto::RSAKey Key("", KeyFileName, KeyPassword); Poco::Crypto::RSAKey Key("", KeyFileName, KeyPassword);
DeviceSecureContext->useCertificate(Cert); DeviceSecureContext->useCertificate(Cert);
@@ -96,11 +93,7 @@ namespace OpenWifi {
DeviceSecureContext->addCertificateAuthority(Root); DeviceSecureContext->addCertificateAuthority(Root);
DeviceSecureContext->addChainCertificate(Issuing); DeviceSecureContext->addChainCertificate(Issuing);
DeviceSecureContext->addCertificateAuthority(Issuing); DeviceSecureContext->addCertificateAuthority(Issuing);
ClientCasCerts = Poco::Net::X509Certificate::readPEM(ClientCasFile); DeviceSecureContext->addCertificateAuthority(Root);
for (const auto &cert : ClientCasCerts) {
DeviceSecureContext->addChainCertificate(cert);
DeviceSecureContext->addCertificateAuthority(cert);
}
DeviceSecureContext->enableSessionCache(true); DeviceSecureContext->enableSessionCache(true);
DeviceSecureContext->setSessionCacheSize(0); DeviceSecureContext->setSessionCacheSize(0);
DeviceSecureContext->setSessionTimeout(120); DeviceSecureContext->setSessionTimeout(120);
@@ -153,7 +146,7 @@ namespace OpenWifi {
auto WebClientSecureContext = auto WebClientSecureContext =
new Poco::Net::Context(Poco::Net::Context::SERVER_USE, KeyFileName, new Poco::Net::Context(Poco::Net::Context::SERVER_USE, KeyFileName,
CertFileName, "", Poco::Net::Context::VERIFY_NONE); CertFileName, "", Poco::Net::Context::VERIFY_RELAXED);
Poco::Crypto::X509Certificate WebRoot(RootCaFileName); Poco::Crypto::X509Certificate WebRoot(RootCaFileName);
WebClientSecureContext->addCertificateAuthority(WebRoot); WebClientSecureContext->addCertificateAuthority(WebRoot);
WebClientSecureContext->disableStatelessSessionResumption(); WebClientSecureContext->disableStatelessSessionResumption();

View File

@@ -56,10 +56,10 @@ namespace OpenWifi {
struct DeviceDetails { struct DeviceDetails {
std::string reason; std::string reason;
std::string author; std::string author;
std::uint64_t created=Utils::Now(); std::uint64_t created;
}; };
static std::map<std::uint64_t , DeviceDetails> BlackListDevices; static std::map<std::string, DeviceDetails> BlackListDevices;
static std::recursive_mutex BlackListMutex; static std::recursive_mutex BlackListMutex;
bool Storage::InitializeBlackListCache() { bool Storage::InitializeBlackListCache() {
@@ -78,7 +78,7 @@ namespace OpenWifi {
auto Reason = RSet[1].convert<std::string>(); auto Reason = RSet[1].convert<std::string>();
auto Author = RSet[2].convert<std::string>(); auto Author = RSet[2].convert<std::string>();
auto Created = RSet[3].convert<std::uint64_t>(); auto Created = RSet[3].convert<std::uint64_t>();
BlackListDevices[Utils::MACToInt(SerialNumber)] = BlackListDevices[SerialNumber] =
DeviceDetails{.reason = Reason, .author = Author, .created = Created}; DeviceDetails{.reason = Reason, .author = Author, .created = Created};
More = RSet.moveNext(); More = RSet.moveNext();
} }
@@ -93,7 +93,6 @@ namespace OpenWifi {
bool Storage::AddBlackListDevice(GWObjects::BlackListedDevice &Device) { bool Storage::AddBlackListDevice(GWObjects::BlackListedDevice &Device) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Insert(Sess); Poco::Data::Statement Insert(Sess);
std::string St{"INSERT INTO BlackList (" + DB_BlackListDeviceSelectFields + ") " + std::string St{"INSERT INTO BlackList (" + DB_BlackListDeviceSelectFields + ") " +
@@ -103,9 +102,9 @@ namespace OpenWifi {
ConvertBlackListDeviceRecord(Device, T); ConvertBlackListDeviceRecord(Device, T);
Insert << ConvertParams(St), Poco::Data::Keywords::use(T); Insert << ConvertParams(St), Poco::Data::Keywords::use(T);
Insert.execute(); Insert.execute();
Sess.commit();
std::lock_guard G(BlackListMutex); std::lock_guard G(BlackListMutex);
BlackListDevices[Utils::MACToInt(Device.serialNumber)] = DeviceDetails{ BlackListDevices[Device.serialNumber] = DeviceDetails{
.reason = Device.reason, .author = Device.author, .created = Device.created}; .reason = Device.reason, .author = Device.author, .created = Device.created};
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -131,7 +130,6 @@ namespace OpenWifi {
bool Storage::DeleteBlackListDevice(std::string &SerialNumber) { bool Storage::DeleteBlackListDevice(std::string &SerialNumber) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St{"DELETE FROM BlackList WHERE SerialNumber=?"}; std::string St{"DELETE FROM BlackList WHERE SerialNumber=?"};
@@ -139,9 +137,9 @@ namespace OpenWifi {
Poco::toLowerInPlace(SerialNumber); Poco::toLowerInPlace(SerialNumber);
Delete << ConvertParams(St), Poco::Data::Keywords::use(SerialNumber); Delete << ConvertParams(St), Poco::Data::Keywords::use(SerialNumber);
Delete.execute(); Delete.execute();
Sess.commit();
std::lock_guard G(BlackListMutex); std::lock_guard G(BlackListMutex);
BlackListDevices.erase(Utils::MACToInt(SerialNumber)); BlackListDevices.erase(SerialNumber);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -179,7 +177,6 @@ namespace OpenWifi {
GWObjects::BlackListedDevice &Device) { GWObjects::BlackListedDevice &Device) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
std::string St{"UPDATE BlackList SET " + DB_BlackListDeviceUpdateFields + std::string St{"UPDATE BlackList SET " + DB_BlackListDeviceUpdateFields +
@@ -190,9 +187,9 @@ namespace OpenWifi {
Update << ConvertParams(St), Poco::Data::Keywords::use(T), Update << ConvertParams(St), Poco::Data::Keywords::use(T),
Poco::Data::Keywords::use(SerialNumber); Poco::Data::Keywords::use(SerialNumber);
Update.execute(); Update.execute();
Sess.commit();
std::lock_guard G(BlackListMutex); std::lock_guard G(BlackListMutex);
BlackListDevices[Utils::MACToInt(Device.serialNumber)] = DeviceDetails{ BlackListDevices[Device.serialNumber] = DeviceDetails{
.reason = Device.reason, .author = Device.author, .created = Device.created}; .reason = Device.reason, .author = Device.author, .created = Device.created};
return true; return true;
@@ -236,10 +233,10 @@ namespace OpenWifi {
return BlackListDevices.size(); return BlackListDevices.size();
} }
bool Storage::IsBlackListed(std::uint64_t SerialNumber, std::string &reason, bool Storage::IsBlackListed(const std::string &SerialNumber, std::string &reason,
std::string &author, std::uint64_t &created) { std::string &author, std::uint64_t &created) {
std::lock_guard G(BlackListMutex); std::lock_guard G(BlackListMutex);
auto DeviceHint = BlackListDevices.find(SerialNumber); auto DeviceHint = BlackListDevices.find(Poco::toLower(SerialNumber));
if (DeviceHint == end(BlackListDevices)) if (DeviceHint == end(BlackListDevices))
return false; return false;
reason = DeviceHint->second.reason; reason = DeviceHint->second.reason;
@@ -248,9 +245,9 @@ namespace OpenWifi {
return true; return true;
} }
bool Storage::IsBlackListed(std::uint64_t SerialNumber) { bool Storage::IsBlackListed(const std::string &SerialNumber) {
std::lock_guard G(BlackListMutex); std::lock_guard G(BlackListMutex);
auto DeviceHint = BlackListDevices.find(SerialNumber); auto DeviceHint = BlackListDevices.find(Poco::toLower(SerialNumber));
return DeviceHint != end(BlackListDevices); return DeviceHint != end(BlackListDevices);
} }
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -17,11 +17,11 @@
namespace OpenWifi { namespace OpenWifi {
bool Storage::CreateDeviceCapabilities(Poco::Data::Session &Session, std::string &SerialNumber, bool Storage::CreateDeviceCapabilities(std::string &SerialNumber,
const Config::Capabilities &Capabilities) { const Config::Capabilities &Capabilities) {
try { try {
Session.begin(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement UpSert(Session); Poco::Data::Statement UpSert(Sess);
std::string TCaps{Capabilities.AsString()}; std::string TCaps{Capabilities.AsString()};
uint64_t Now = Utils::Now(); uint64_t Now = Utils::Now();
@@ -33,7 +33,6 @@ namespace OpenWifi {
Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(TCaps), Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(TCaps),
Poco::Data::Keywords::use(Now); Poco::Data::Keywords::use(Now);
UpSert.execute(); UpSert.execute();
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -42,11 +41,11 @@ namespace OpenWifi {
return false; return false;
} }
bool Storage::UpdateDeviceCapabilities(Poco::Data::Session &Session, std::string &SerialNumber, bool Storage::UpdateDeviceCapabilities(std::string &SerialNumber,
const Config::Capabilities &Caps) { const Config::Capabilities &Caps) {
try { try {
Session.begin(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement UpSert(Session); Poco::Data::Statement UpSert(Sess);
uint64_t Now = Utils::Now(); uint64_t Now = Utils::Now();
if (!Caps.Compatible().empty() && !Caps.Platform().empty()) if (!Caps.Compatible().empty() && !Caps.Platform().empty())
@@ -62,7 +61,6 @@ namespace OpenWifi {
Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(TCaps), Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(TCaps),
Poco::Data::Keywords::use(Now); Poco::Data::Keywords::use(Now);
UpSert.execute(); UpSert.execute();
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -101,14 +99,13 @@ namespace OpenWifi {
bool Storage::DeleteDeviceCapabilities(std::string &SerialNumber) { bool Storage::DeleteDeviceCapabilities(std::string &SerialNumber) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St{"DELETE FROM Capabilities WHERE SerialNumber=?"}; std::string St{"DELETE FROM Capabilities WHERE SerialNumber=?"};
Delete << ConvertParams(St), Poco::Data::Keywords::use(SerialNumber); Delete << ConvertParams(St), Poco::Data::Keywords::use(SerialNumber);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),

View File

@@ -105,7 +105,6 @@ namespace OpenWifi {
bool Storage::RemoveOldCommands(std::string &SerialNumber, std::string &Command) { bool Storage::RemoveOldCommands(std::string &SerialNumber, std::string &Command) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St{ std::string St{
@@ -113,7 +112,8 @@ namespace OpenWifi {
Delete << ConvertParams(St), Poco::Data::Keywords::use(SerialNumber), Delete << ConvertParams(St), Poco::Data::Keywords::use(SerialNumber),
Poco::Data::Keywords::use(Command); Poco::Data::Keywords::use(Command);
Delete.execute(); Delete.execute();
Sess.commit(); Delete.reset(Sess);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -146,7 +146,6 @@ namespace OpenWifi {
RemoveOldCommands(SerialNumber, Command.Command); RemoveOldCommands(SerialNumber, Command.Command);
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Insert(Sess); Poco::Data::Statement Insert(Sess);
std::string St{"INSERT INTO CommandList ( " + DB_Command_SelectFields + " ) VALUES( " + std::string St{"INSERT INTO CommandList ( " + DB_Command_SelectFields + " ) VALUES( " +
@@ -157,7 +156,7 @@ namespace OpenWifi {
Insert << ConvertParams(St), Poco::Data::Keywords::use(R); Insert << ConvertParams(St), Poco::Data::Keywords::use(R);
Insert.execute(); Insert.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -216,7 +215,6 @@ namespace OpenWifi {
bool Storage::DeleteCommands(std::string &SerialNumber, uint64_t FromDate, uint64_t ToDate) { bool Storage::DeleteCommands(std::string &SerialNumber, uint64_t FromDate, uint64_t ToDate) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
bool DatesIncluded = (FromDate != 0 || ToDate != 0); bool DatesIncluded = (FromDate != 0 || ToDate != 0);
@@ -239,7 +237,8 @@ namespace OpenWifi {
Delete << IntroStatement + DateSelector; Delete << IntroStatement + DateSelector;
Delete.execute(); Delete.execute();
Sess.commit(); Delete.reset(Sess);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -275,6 +274,7 @@ namespace OpenWifi {
if (Records.size() < HowMany) if (Records.size() < HowMany)
Done = true; Done = true;
} }
Select.reset(Sess);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -286,7 +286,6 @@ namespace OpenWifi {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
std::string St{"UPDATE CommandList SET Status=?, Executed=?, Completed=?, " std::string St{"UPDATE CommandList SET Status=?, Executed=?, Completed=?, "
@@ -300,7 +299,7 @@ namespace OpenWifi {
Poco::Data::Keywords::use(Command.ErrorCode), Poco::Data::Keywords::use(UUID); Poco::Data::Keywords::use(Command.ErrorCode), Poco::Data::Keywords::use(UUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -312,7 +311,6 @@ namespace OpenWifi {
bool Storage::SetCommandExecuted(std::string &CommandUUID) { bool Storage::SetCommandExecuted(std::string &CommandUUID) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
auto Now = Utils::Now(); auto Now = Utils::Now();
@@ -323,7 +321,6 @@ namespace OpenWifi {
Update << ConvertParams(St), Poco::Data::Keywords::use(Now), Update << ConvertParams(St), Poco::Data::Keywords::use(Now),
Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(CommandUUID); Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(CommandUUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -334,7 +331,6 @@ namespace OpenWifi {
void Storage::RemovedExpiredCommands() { void Storage::RemovedExpiredCommands() {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
auto Now = Utils::Now(), Window = Now - CommandManager()->CommandTimeout(); auto Now = Utils::Now(), Window = Now - CommandManager()->CommandTimeout();
@@ -345,7 +341,8 @@ namespace OpenWifi {
Update << ConvertParams(St), Poco::Data::Keywords::use(Now), Update << ConvertParams(St), Poco::Data::Keywords::use(Now),
Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(Window); Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(Window);
Update.execute(); Update.execute();
Sess.commit(); Update.reset(Sess);
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
} }
@@ -354,7 +351,6 @@ namespace OpenWifi {
bool Storage::SetCommandLastTry(std::string &CommandUUID) { bool Storage::SetCommandLastTry(std::string &CommandUUID) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
auto Now = Utils::Now(); auto Now = Utils::Now();
@@ -363,7 +359,6 @@ namespace OpenWifi {
Update << ConvertParams(St), Poco::Data::Keywords::use(Now), Update << ConvertParams(St), Poco::Data::Keywords::use(Now),
Poco::Data::Keywords::use(CommandUUID); Poco::Data::Keywords::use(CommandUUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -374,7 +369,6 @@ namespace OpenWifi {
void Storage::RemoveTimedOutCommands() { void Storage::RemoveTimedOutCommands() {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
auto Now = Utils::Now(), Window = Now - CommandManager()->CommandTimeout(); auto Now = Utils::Now(), Window = Now - CommandManager()->CommandTimeout();
@@ -383,7 +377,7 @@ namespace OpenWifi {
Update << ConvertParams(St), Poco::Data::Keywords::use(Now), Update << ConvertParams(St), Poco::Data::Keywords::use(Now),
Poco::Data::Keywords::use(Window); Poco::Data::Keywords::use(Window);
Update.execute(); Update.execute();
Sess.commit(); Update.reset(Sess);
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
} }
@@ -392,7 +386,6 @@ namespace OpenWifi {
bool Storage::SetCommandTimedOut(std::string &CommandUUID) { bool Storage::SetCommandTimedOut(std::string &CommandUUID) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
auto Now = Utils::Now(); auto Now = Utils::Now();
@@ -402,7 +395,6 @@ namespace OpenWifi {
Update << ConvertParams(St), Poco::Data::Keywords::use(Now), Update << ConvertParams(St), Poco::Data::Keywords::use(Now),
Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(CommandUUID); Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(CommandUUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -433,7 +425,6 @@ namespace OpenWifi {
bool Storage::DeleteCommand(std::string &UUID) { bool Storage::DeleteCommand(std::string &UUID) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St{"DELETE FROM CommandList WHERE UUID=?"}; std::string St{"DELETE FROM CommandList WHERE UUID=?"};
@@ -444,7 +435,8 @@ namespace OpenWifi {
St = "DELETE FROM FileUploads WHERE UUID=?"; St = "DELETE FROM FileUploads WHERE UUID=?";
Delete << ConvertParams(St), Poco::Data::Keywords::use(UUID); Delete << ConvertParams(St), Poco::Data::Keywords::use(UUID);
Delete.execute(); Delete.execute();
Sess.commit(); Delete.reset(Sess);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -518,7 +510,6 @@ namespace OpenWifi {
auto Now = Utils::Now(); auto Now = Utils::Now();
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
std::string St{"UPDATE CommandList SET Executed=? WHERE UUID=?"}; std::string St{"UPDATE CommandList SET Executed=? WHERE UUID=?"};
@@ -527,7 +518,7 @@ namespace OpenWifi {
Poco::Data::Keywords::use(UUID); Poco::Data::Keywords::use(UUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -564,7 +555,6 @@ namespace OpenWifi {
} }
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
auto Status = to_string(Storage::CommandExecutionType::COMMAND_COMPLETED); auto Status = to_string(Storage::CommandExecutionType::COMMAND_COMPLETED);
@@ -576,7 +566,6 @@ namespace OpenWifi {
Poco::Data::Keywords::use(ResultStr), Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(ResultStr), Poco::Data::Keywords::use(Status),
Poco::Data::Keywords::use(tET), Poco::Data::Keywords::use(UUID); Poco::Data::Keywords::use(tET), Poco::Data::Keywords::use(UUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -614,7 +603,6 @@ namespace OpenWifi {
bool Storage::CancelWaitFile(std::string &UUID, std::string &ErrorText) { bool Storage::CancelWaitFile(std::string &UUID, std::string &ErrorText) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
auto Now = Utils::Now(); auto Now = Utils::Now();
uint64_t Size = 0, WaitForFile = 0; uint64_t Size = 0, WaitForFile = 0;
@@ -628,7 +616,6 @@ namespace OpenWifi {
Poco::Data::Keywords::use(ErrorText), Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(ErrorText), Poco::Data::Keywords::use(Now),
Poco::Data::Keywords::use(UUID); Poco::Data::Keywords::use(UUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -644,6 +631,19 @@ namespace OpenWifi {
uint64_t Size = FileContent.str().size(); uint64_t Size = FileContent.str().size();
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Statement(Sess);
std::string StatementStr;
// Get the existing command
StatementStr =
"UPDATE CommandList SET WaitingForFile=?, AttachDate=?, AttachSize=? WHERE UUID=?";
Statement << ConvertParams(StatementStr), Poco::Data::Keywords::use(WaitForFile),
Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(Size),
Poco::Data::Keywords::use(UUID);
Statement.execute();
if (Size < FileUploader()->MaxSize()) { if (Size < FileUploader()->MaxSize()) {
@@ -651,7 +651,7 @@ namespace OpenWifi {
TheBlob.appendRaw((const unsigned char *)FileContent.str().c_str(), TheBlob.appendRaw((const unsigned char *)FileContent.str().c_str(),
FileContent.str().size()); FileContent.str().size());
Sess.begin();
Poco::Data::Statement Insert(Sess); Poco::Data::Statement Insert(Sess);
std::string FileType{Type}; std::string FileType{Type};
@@ -662,25 +662,10 @@ namespace OpenWifi {
Poco::Data::Keywords::use(FileType), Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(FileType), Poco::Data::Keywords::use(Now),
Poco::Data::Keywords::use(TheBlob); Poco::Data::Keywords::use(TheBlob);
Insert.execute(); Insert.execute();
Sess.commit(); return true;
} else { } else {
poco_warning(Logger(), fmt::format("File {} is too large.", UUID)); poco_warning(Logger(), fmt::format("File {} is too large.", UUID));
} }
// update CommandList here to ensure that file us uploaded
Sess.begin();
Poco::Data::Statement Statement(Sess);
std::string StatementStr;
StatementStr =
"UPDATE CommandList SET WaitingForFile=?, AttachDate=?, AttachSize=? WHERE UUID=?";
Statement << ConvertParams(StatementStr), Poco::Data::Keywords::use(WaitForFile),
Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(Size),
Poco::Data::Keywords::use(UUID);
Statement.execute();
Sess.commit();
return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
} }
@@ -688,7 +673,7 @@ namespace OpenWifi {
} }
bool Storage::GetAttachedFileContent(std::string &UUID, const std::string &SerialNumber, bool Storage::GetAttachedFileContent(std::string &UUID, const std::string &SerialNumber,
std::string &FileContent, std::string &Type, int &WaitingForFile) { std::string &FileContent, std::string &Type) {
try { try {
Poco::Data::BLOB L; Poco::Data::BLOB L;
/* /*
@@ -701,10 +686,10 @@ namespace OpenWifi {
Poco::Data::Statement Select1(Sess); Poco::Data::Statement Select1(Sess);
std::string TmpSerialNumber; std::string TmpSerialNumber;
std::string st1{"SELECT SerialNumber, Command , WaitingForFile FROM CommandList WHERE UUID=?"}; std::string st1{"SELECT SerialNumber, Command FROM CommandList WHERE UUID=?"};
std::string Command; std::string Command;
Select1 << ConvertParams(st1), Poco::Data::Keywords::into(TmpSerialNumber), Select1 << ConvertParams(st1), Poco::Data::Keywords::into(TmpSerialNumber),
Poco::Data::Keywords::into(Command), Poco::Data::Keywords::into(WaitingForFile), Poco::Data::Keywords::use(UUID); Poco::Data::Keywords::into(Command), Poco::Data::Keywords::use(UUID);
Select1.execute(); Select1.execute();
if (TmpSerialNumber != SerialNumber) { if (TmpSerialNumber != SerialNumber) {
@@ -727,7 +712,6 @@ namespace OpenWifi {
bool Storage::SetCommandResult(std::string &UUID, std::string &Result) { bool Storage::SetCommandResult(std::string &UUID, std::string &Result) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
auto Now = Utils::Now(); auto Now = Utils::Now();
@@ -738,7 +722,6 @@ namespace OpenWifi {
Poco::Data::Keywords::use(Result), Poco::Data::Keywords::use(Status), Poco::Data::Keywords::use(Result), Poco::Data::Keywords::use(Status),
Poco::Data::Keywords::use(UUID); Poco::Data::Keywords::use(UUID);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -750,14 +733,13 @@ namespace OpenWifi {
bool Storage::RemoveAttachedFile(std::string &UUID) { bool Storage::RemoveAttachedFile(std::string &UUID) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St{"DELETE FROM FileUploads WHERE UUID=?"}; std::string St{"DELETE FROM FileUploads WHERE UUID=?"};
Delete << ConvertParams(St), Poco::Data::Keywords::use(UUID); Delete << ConvertParams(St), Poco::Data::Keywords::use(UUID);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -769,13 +751,11 @@ namespace OpenWifi {
bool Storage::RemoveUploadedFilesRecordsOlderThan(uint64_t Date) { bool Storage::RemoveUploadedFilesRecordsOlderThan(uint64_t Date) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St1{"delete from FileUploads where Created<?"}; std::string St1{"delete from FileUploads where Created<?"};
Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date); Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -786,13 +766,11 @@ namespace OpenWifi {
bool Storage::RemoveCommandListRecordsOlderThan(uint64_t Date) { bool Storage::RemoveCommandListRecordsOlderThan(uint64_t Date) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St1{"delete from CommandList where Submitted<?"}; std::string St1{"delete from CommandList where Submitted<?"};
Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date); Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);

View File

@@ -82,7 +82,6 @@ namespace OpenWifi {
if (!TmpName.empty()) if (!TmpName.empty())
return false; return false;
Sess.begin();
Poco::Data::Statement Insert(Sess); Poco::Data::Statement Insert(Sess);
std::string St2{"INSERT INTO DefaultFirmwares ( " + DB_DefFirmware_SelectFields + std::string St2{"INSERT INTO DefaultFirmwares ( " + DB_DefFirmware_SelectFields +
@@ -95,7 +94,6 @@ namespace OpenWifi {
Insert << ConvertParams(St2), Insert << ConvertParams(St2),
Poco::Data::Keywords::use(R); Poco::Data::Keywords::use(R);
Insert.execute(); Insert.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -109,7 +107,6 @@ namespace OpenWifi {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
Poco::toLowerInPlace(deviceType); Poco::toLowerInPlace(deviceType);
@@ -117,7 +114,7 @@ namespace OpenWifi {
Delete << ConvertParams(St), Poco::Data::Keywords::use(deviceType); Delete << ConvertParams(St), Poco::Data::Keywords::use(deviceType);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -128,9 +125,9 @@ namespace OpenWifi {
bool Storage::UpdateDefaultFirmware(GWObjects::DefaultFirmware &DefFirmware) { bool Storage::UpdateDefaultFirmware(GWObjects::DefaultFirmware &DefFirmware) {
try { try {
uint64_t Now = Utils::Now();
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
uint64_t Now = time(nullptr);
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
DefFirmware.LastModified = Now; DefFirmware.LastModified = Now;
Poco::toLowerInPlace(DefFirmware.deviceType); Poco::toLowerInPlace(DefFirmware.deviceType);
@@ -146,7 +143,7 @@ namespace OpenWifi {
Poco::Data::Keywords::use(R), Poco::Data::Keywords::use(R),
Poco::Data::Keywords::use(DefFirmware.deviceType); Poco::Data::Keywords::use(DefFirmware.deviceType);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),

View File

@@ -19,39 +19,37 @@ namespace OpenWifi {
"Models TEXT, " "Models TEXT, "
"Description TEXT, " "Description TEXT, "
"Created BIGINT , " "Created BIGINT , "
"LastModified BIGINT, Platform TEXT )"}; "LastModified BIGINT)"};
const static std::string DB_DefConfig_SelectFields{"Name, " const static std::string DB_DefConfig_SelectFields{"Name, "
"Configuration, " "Configuration, "
"Models, " "Models, "
"Description, " "Description, "
"Created, " "Created, "
"LastModified, Platform "}; "LastModified "};
const static std::string DB_DefConfig_InsertValues{"?,?,?,?,?,?,?"}; const static std::string DB_DefConfig_InsertValues{"?,?,?,?,?,?"};
typedef Poco::Tuple<std::string, std::string, std::string, std::string, uint64_t, uint64_t, std::string> typedef Poco::Tuple<std::string, std::string, std::string, std::string, uint64_t, uint64_t>
DefConfigRecordTuple; DefConfigRecordTuple;
typedef std::vector<DefConfigRecordTuple> DefConfigRecordList; typedef std::vector<DefConfigRecordTuple> DefConfigRecordList;
void Convert(const DefConfigRecordTuple &R, GWObjects::DefaultConfiguration &T) { void Convert(const DefConfigRecordTuple &R, GWObjects::DefaultConfiguration &T) {
T.name = R.get<0>(); T.Name = R.get<0>();
T.configuration = R.get<1>(); T.Configuration = R.get<1>();
T.models = RESTAPI_utils::to_object_array(R.get<2>()); T.Models = RESTAPI_utils::to_object_array(R.get<2>());
T.description = R.get<3>(); T.Description = R.get<3>();
T.created = R.get<4>(); T.Created = R.get<4>();
T.lastModified = R.get<5>(); T.LastModified = R.get<5>();
T.platform = R.get<6>();
} }
void Convert(const GWObjects::DefaultConfiguration &R, DefConfigRecordTuple &T) { void Convert(const GWObjects::DefaultConfiguration &R, DefConfigRecordTuple &T) {
T.set<0>(R.name); T.set<0>(R.Name);
T.set<1>(R.configuration); T.set<1>(R.Configuration);
T.set<2>(RESTAPI_utils::to_string(R.models)); T.set<2>(RESTAPI_utils::to_string(R.Models));
T.set<3>(R.description); T.set<3>(R.Description);
T.set<4>(R.created); T.set<4>(R.Created);
T.set<5>(R.lastModified); T.set<5>(R.LastModified);
T.set<6>(R.platform);
} }
bool Storage::CreateDefaultConfiguration(std::string &Name, bool Storage::CreateDefaultConfiguration(std::string &Name,
@@ -71,10 +69,9 @@ namespace OpenWifi {
if (!TmpName.empty()) if (!TmpName.empty())
return false; return false;
Config::Config Cfg(DefConfig.configuration); Config::Config Cfg(DefConfig.Configuration);
if (Cfg.Valid()) { if (Cfg.Valid()) {
Sess.begin();
Poco::Data::Statement Insert(Sess); Poco::Data::Statement Insert(Sess);
std::string St{"INSERT INTO DefaultConfigs ( " + DB_DefConfig_SelectFields + std::string St{"INSERT INTO DefaultConfigs ( " + DB_DefConfig_SelectFields +
@@ -86,7 +83,6 @@ namespace OpenWifi {
Convert(DefConfig, R); Convert(DefConfig, R);
Insert << ConvertParams(St), Poco::Data::Keywords::use(R); Insert << ConvertParams(St), Poco::Data::Keywords::use(R);
Insert.execute(); Insert.execute();
Sess.commit();
return true; return true;
} else { } else {
poco_warning(Logger(), "Cannot create device: invalid configuration."); poco_warning(Logger(), "Cannot create device: invalid configuration.");
@@ -103,14 +99,13 @@ namespace OpenWifi {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St{"DELETE FROM DefaultConfigs WHERE Name=?"}; std::string St{"DELETE FROM DefaultConfigs WHERE Name=?"};
Delete << ConvertParams(St), Poco::Data::Keywords::use(Name); Delete << ConvertParams(St), Poco::Data::Keywords::use(Name);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -122,14 +117,14 @@ namespace OpenWifi {
bool Storage::UpdateDefaultConfiguration(std::string &Name, bool Storage::UpdateDefaultConfiguration(std::string &Name,
GWObjects::DefaultConfiguration &DefConfig) { GWObjects::DefaultConfiguration &DefConfig) {
try { try {
uint64_t Now = Utils::Now();
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
uint64_t Now = time(nullptr);
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
DefConfig.lastModified = Now; DefConfig.LastModified = Now;
std::string St{"UPDATE DefaultConfigs SET Name=?, Configuration=?, Models=?, " std::string St{"UPDATE DefaultConfigs SET Name=?, Configuration=?, Models=?, "
"Description=?, Created=? , LastModified=? , Platform=? WHERE Name=?"}; "Description=?, Created=? , LastModified=? WHERE Name=?"};
DefConfigRecordTuple R; DefConfigRecordTuple R;
Convert(DefConfig, R); Convert(DefConfig, R);
@@ -137,7 +132,6 @@ namespace OpenWifi {
Update << ConvertParams(St), Poco::Data::Keywords::use(R), Update << ConvertParams(St), Poco::Data::Keywords::use(R),
Poco::Data::Keywords::use(Name); Poco::Data::Keywords::use(Name);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -221,30 +215,31 @@ namespace OpenWifi {
return false; return false;
} }
bool Storage::FindDefaultConfigurationForModel(const std::string &DeviceModel, const std::string &Platform, bool Storage::FindDefaultConfigurationForModel(const std::string &Model,
GWObjects::DefaultConfiguration &Config) { GWObjects::DefaultConfiguration &DefConfig) {
try { try {
DefConfigRecordList DefConfigs; DefConfigRecordList Records;
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess); Poco::Data::Statement Select(Sess);
Select << "SELECT " + DB_DefConfig_SelectFields + " FROM DefaultConfigs", Select << "SELECT " + DB_DefConfig_SelectFields + " FROM DefaultConfigs",
Poco::Data::Keywords::into(DefConfigs); Poco::Data::Keywords::into(Records);
Select.execute(); Select.execute();
for (const auto &DefConfig : DefConfigs) { for (const auto &i : Records) {
GWObjects::DefaultConfiguration C; GWObjects::DefaultConfiguration Config;
Convert(DefConfig, C); Convert(i, Config);
for (const auto &Model : C.models) { for (const auto &j : Config.Models) {
if ((Model == "*" || Model == DeviceModel) && (Config.platform == Platform)){ if (j == "*" || j == Model) {
Config = C; DefConfig = Config;
return true; return true;
} }
} }
} }
Logger().information( Logger().information(
fmt::format("AUTO-PROVISIONING: no default configuration for model:{}", DeviceModel)); fmt::format("AUTO-PROVISIONING: no default configuration for model:{}", Model));
return false;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
E.displayText())); E.displayText()));

View File

@@ -172,18 +172,14 @@ namespace OpenWifi {
R.set<30>(D.connectReason); R.set<30>(D.connectReason);
} }
bool Storage::GetDeviceCount(uint64_t &Count, const std::string &platform) { bool Storage::GetDeviceCount(uint64_t &Count) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess); Poco::Data::Statement Select(Sess);
if(!platform.empty()) {
std::string st{"SELECT COUNT(*) FROM Devices WHERE DeviceType='" + platform + "'"};
Select << st, Poco::Data::Keywords::into(Count);
} else {
std::string st{"SELECT COUNT(*) FROM Devices"}; std::string st{"SELECT COUNT(*) FROM Devices"};
Select << st, Poco::Data::Keywords::into(Count); Select << st, Poco::Data::Keywords::into(Count);
}
Select.execute(); Select.execute();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -194,37 +190,16 @@ namespace OpenWifi {
bool Storage::GetDeviceSerialNumbers(uint64_t From, uint64_t HowMany, bool Storage::GetDeviceSerialNumbers(uint64_t From, uint64_t HowMany,
std::vector<std::string> &SerialNumbers, std::vector<std::string> &SerialNumbers,
const std::string &orderBy, const std::string &orderBy) {
const std::string &platform, bool includeProvisioned) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess); Poco::Data::Statement Select(Sess);
std::string st; std::string st;
std::string whereClause = "";
if(!platform.empty()) {
if (includeProvisioned == false) {
whereClause = fmt::format("WHERE entity='' and venue='' and DeviceType='" + platform + "'");
} else {
whereClause = fmt::format("WHERE DeviceType='" + platform + "'");
}
//st = "SELECT SerialNumber From Devices WHERE DeviceType='" + platform + "' ";
} else {
if (includeProvisioned == false) {
whereClause = fmt::format("WHERE entity='' and venue=''");
}
//st = "SELECT SerialNumber From Devices ";
}
st = fmt::format("SELECT SerialNumber From Devices {}", whereClause);
if (orderBy.empty()) if (orderBy.empty())
st += " ORDER BY SerialNumber ASC "; st = "SELECT SerialNumber From Devices ORDER BY SerialNumber ASC ";
else else
st += orderBy; st = "SELECT SerialNumber From Devices " + orderBy;
Select << st + ComputeRange(From, HowMany), Poco::Data::Keywords::into(SerialNumbers); Select << st + ComputeRange(From, HowMany), Poco::Data::Keywords::into(SerialNumbers);
Select.execute(); Select.execute();
@@ -235,7 +210,7 @@ namespace OpenWifi {
return false; return false;
} }
/* bool Storage::UpdateDeviceConfiguration(std::string &SerialNumber, std::string &Configuration, bool Storage::UpdateDeviceConfiguration(std::string &SerialNumber, std::string &Configuration,
uint64_t &NewUUID) { uint64_t &NewUUID) {
try { try {
@@ -280,7 +255,7 @@ namespace OpenWifi {
} }
return false; return false;
} }
*/
bool Storage::RollbackDeviceConfigurationChange(std::string & SerialNumber) { bool Storage::RollbackDeviceConfigurationChange(std::string & SerialNumber) {
try { try {
GWObjects::Device D; GWObjects::Device D;
@@ -290,10 +265,9 @@ namespace OpenWifi {
D.pendingUUID = 0; D.pendingUUID = 0;
D.LastConfigurationChange = Utils::Now(); D.LastConfigurationChange = Utils::Now();
SetCurrentConfigurationID(Utils::SerialNumberToInt(SerialNumber), D.UUID); ConfigurationCache().Add(Utils::SerialNumberToInt(SerialNumber), D.UUID);
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
DeviceRecordTuple R; DeviceRecordTuple R;
@@ -303,7 +277,6 @@ namespace OpenWifi {
Update << ConvertParams(St2), Poco::Data::Keywords::use(R), Update << ConvertParams(St2), Poco::Data::Keywords::use(R),
Poco::Data::Keywords::use(SerialNumber); Poco::Data::Keywords::use(SerialNumber);
Update.execute(); Update.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -313,43 +286,22 @@ namespace OpenWifi {
bool Storage::CompleteDeviceConfigurationChange(std::string & SerialNumber) { bool Storage::CompleteDeviceConfigurationChange(std::string & SerialNumber) {
try { try {
auto Session = Pool_->get();
return CompleteDeviceConfigurationChange(Session, SerialNumber);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::CompleteDeviceConfigurationChange(Poco::Data::Session & Session, std::string & SerialNumber) {
try {
GWObjects::Device D; GWObjects::Device D;
if (!GetDevice(SerialNumber, D)) if (!GetDevice(SerialNumber, D))
return false; return false;
if(!D.pendingConfiguration.empty()) { if(D.pendingConfiguration.empty())
return true;
D.Configuration = D.pendingConfiguration; D.Configuration = D.pendingConfiguration;
D.pendingConfiguration.clear(); D.pendingConfiguration.clear();
}
if(D.pendingUUID!=0) {
D.UUID = D.pendingUUID; D.UUID = D.pendingUUID;
D.pendingUUID = 0; D.pendingUUID = 0;
}
// if this is a broken device, fix it...
if(D.UUID==0) {
Config::Config Cfg(D.Configuration);
if(Cfg.Valid()) {
D.UUID = Cfg.UUID();
}
}
D.LastConfigurationChange = Utils::Now(); D.LastConfigurationChange = Utils::Now();
SetCurrentConfigurationID(Utils::SerialNumberToInt(SerialNumber), D.UUID);
Session.begin(); ConfigurationCache().Add(Utils::SerialNumberToInt(SerialNumber), D.UUID);
Poco::Data::Statement Update(Session);
Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Update(Sess);
DeviceRecordTuple R; DeviceRecordTuple R;
ConvertDeviceRecord(D, R); ConvertDeviceRecord(D, R);
@@ -358,7 +310,6 @@ namespace OpenWifi {
Update << ConvertParams(St2), Poco::Data::Keywords::use(R), Update << ConvertParams(St2), Poco::Data::Keywords::use(R),
Poco::Data::Keywords::use(SerialNumber); Poco::Data::Keywords::use(SerialNumber);
Update.execute(); Update.execute();
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -377,12 +328,13 @@ namespace OpenWifi {
return false; return false;
} }
Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess);
GWObjects::Device D; GWObjects::Device D;
if (!GetDevice(SerialNumber, D)) if (!GetDevice(SerialNumber, D))
return false; return false;
uint64_t Now = time(nullptr); uint64_t Now = time(nullptr);
if(NewUUID==0) { if(NewUUID==0) {
D.pendingUUID = NewUUID = (D.LastConfigurationChange == Now ? Now + 1 : Now); D.pendingUUID = NewUUID = (D.LastConfigurationChange == Now ? Now + 1 : Now);
@@ -391,8 +343,6 @@ namespace OpenWifi {
} }
if (Cfg.SetUUID(NewUUID)) { if (Cfg.SetUUID(NewUUID)) {
Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
D.pendingConfiguration = Cfg.get(); D.pendingConfiguration = Cfg.get();
@@ -403,7 +353,6 @@ namespace OpenWifi {
Update << ConvertParams(St2), Poco::Data::Keywords::use(R), Update << ConvertParams(St2), Poco::Data::Keywords::use(R),
Poco::Data::Keywords::use(SerialNumber); Poco::Data::Keywords::use(SerialNumber);
Update.execute(); Update.execute();
Sess.commit();
poco_information(Logger(), poco_information(Logger(),
fmt::format("DEVICE-PENDING-CONFIGURATION-UPDATED({}): New UUID is {}", fmt::format("DEVICE-PENDING-CONFIGURATION-UPDATED({}): New UUID is {}",
SerialNumber, NewUUID)); SerialNumber, NewUUID));
@@ -417,46 +366,37 @@ namespace OpenWifi {
return false; return false;
} }
bool Storage::SetDeviceLastRecordedContact(LockedDbSession &Session, std::string &SerialNumber, std::uint64_t lastRecordedContact) { bool Storage::SetDeviceLastRecordedContact(std::string &SerialNumber, std::uint64_t lastRecordedContact) {
try { try {
std::lock_guard Lock(Session.Mutex()); Poco::Data::Session Sess = Pool_->get();
return SetDeviceLastRecordedContact(Session.Session(), SerialNumber, lastRecordedContact); Poco::Data::Statement Update(Sess);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::SetDeviceLastRecordedContact(Poco::Data::Session &Session, std::string &SerialNumber, std::uint64_t lastRecordedContact) {
try {
Session.begin();
Poco::Data::Statement Update(Session);
std::string St{"UPDATE Devices SET lastRecordedContact=? WHERE SerialNumber=?"}; std::string St{"UPDATE Devices SET lastRecordedContact=? WHERE SerialNumber=?"};
Update << ConvertParams(St), Poco::Data::Keywords::use(lastRecordedContact), Update << ConvertParams(St), Poco::Data::Keywords::use(lastRecordedContact),
Poco::Data::Keywords::use(SerialNumber); Poco::Data::Keywords::use(SerialNumber);
Update.execute(); Update.execute();
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
} }
return false; return false;
} }
bool Storage::SetDeviceLastRecordedContact(std::string &SerialNumber, std::uint64_t lastRecordedContact) { bool Storage::CreateDevice(GWObjects::Device &DeviceDetails) {
try {
auto Session = Pool_->get();
return SetDeviceLastRecordedContact(Session, SerialNumber, lastRecordedContact);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::CreateDevice(Poco::Data::Session &Sess, GWObjects::Device &DeviceDetails) {
std::string SerialNumber; std::string SerialNumber;
try { try {
Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess);
std::string St{"SELECT SerialNumber FROM Devices WHERE SerialNumber=?"};
// Select << ConvertParams(St), Poco::Data::Keywords::into(SerialNumber),
// Poco::Data::Keywords::use(DeviceDetails.SerialNumber);
// Select.execute();
// if (Select.rowsExtracted() == 0) {
Config::Config Cfg(DeviceDetails.Configuration); Config::Config Cfg(DeviceDetails.Configuration);
uint64_t Now = Utils::Now(); uint64_t Now = Utils::Now();
@@ -467,7 +407,6 @@ namespace OpenWifi {
if (Cfg.Valid() && Cfg.SetUUID(DeviceDetails.UUID)) { if (Cfg.Valid() && Cfg.SetUUID(DeviceDetails.UUID)) {
DeviceDetails.Configuration = Cfg.get(); DeviceDetails.Configuration = Cfg.get();
Sess.begin();
Poco::Data::Statement Insert(Sess); Poco::Data::Statement Insert(Sess);
std::string St2{"INSERT INTO Devices ( " + DB_DeviceSelectFields + " ) " + std::string St2{"INSERT INTO Devices ( " + DB_DeviceSelectFields + " ) " +
@@ -478,13 +417,17 @@ namespace OpenWifi {
ConvertDeviceRecord(DeviceDetails, R); ConvertDeviceRecord(DeviceDetails, R);
Insert << ConvertParams(St2), Poco::Data::Keywords::use(R); Insert << ConvertParams(St2), Poco::Data::Keywords::use(R);
Insert.execute(); Insert.execute();
Sess.commit();
SetCurrentConfigurationID(DeviceDetails.SerialNumber, DeviceDetails.UUID); SetCurrentConfigurationID(DeviceDetails.SerialNumber, DeviceDetails.UUID);
SerialNumberCache()->AddSerialNumber(DeviceDetails.SerialNumber); SerialNumberCache()->AddSerialNumber(DeviceDetails.SerialNumber);
return true;
} else { } else {
poco_warning(Logger(), "Cannot create device: invalid configuration."); poco_warning(Logger(), "Cannot create device: invalid configuration.");
return false; return false;
} }
// } else {
// poco_warning(Logger(), fmt::format("Device {} already exists.", SerialNumber));
// return false;
// }
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -492,26 +435,6 @@ namespace OpenWifi {
return false; return false;
} }
bool Storage::CreateDevice(LockedDbSession &Session, GWObjects::Device &DeviceDetails) {
try {
std::lock_guard Lock(Session.Mutex());
return CreateDevice(Session.Session(), DeviceDetails);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::CreateDevice(GWObjects::Device &DeviceDetails) {
try {
auto Session = Pool_->get();
return CreateDevice(Session, DeviceDetails);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
static std::string InsertRadiosCountyRegulation(std::string &Config, static std::string InsertRadiosCountyRegulation(std::string &Config,
const Poco::Net::IPAddress &IPAddress) { const Poco::Net::IPAddress &IPAddress) {
std::string FoundCountry; std::string FoundCountry;
@@ -561,18 +484,19 @@ namespace OpenWifi {
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
} }
return false; return true;
} }
bool Storage::CreateDefaultDevice(Poco::Data::Session &Session, std::string &SerialNumber, const Config::Capabilities &Caps, #define __DBGLOG__ std::cout << __LINE__ << std::endl;
bool Storage::CreateDefaultDevice(std::string &SerialNumber, const Config::Capabilities &Caps,
std::string &Firmware, std::string &Firmware,
const Poco::Net::IPAddress &IPAddress, const Poco::Net::IPAddress &IPAddress,
bool simulated) { bool simulated) {
GWObjects::Device D; GWObjects::Device D;
poco_information(Logger(), fmt::format("AUTO-CREATION({})", SerialNumber));
// poco_information(Logger(), fmt::format("AUTO-CREATION({}): Start.", SerialNumber)); uint64_t Now = time(nullptr);
uint64_t Now = Utils::Now();
GWObjects::DefaultConfiguration DefConfig; GWObjects::DefaultConfiguration DefConfig;
if (!Caps.Platform().empty() && !Caps.Compatible().empty()) { if (!Caps.Platform().empty() && !Caps.Compatible().empty()) {
@@ -593,31 +517,21 @@ namespace OpenWifi {
} }
if (!Found && AP_WS_Server()->UseDefaults() && if (!Found && AP_WS_Server()->UseDefaults() &&
FindDefaultConfigurationForModel(Caps.Compatible(), Caps.Platform(), DefConfig)) { FindDefaultConfigurationForModel(Caps.Compatible(), DefConfig)) {
Config::Config NewConfig(DefConfig.configuration); Config::Config NewConfig(DefConfig.Configuration);
NewConfig.SetUUID(Now); NewConfig.SetUUID(Now);
D.Configuration = NewConfig.get(); D.Configuration = NewConfig.get();
} else if (!Found) { } else if (!Found) {
if(Caps.Platform()==Platforms::AP) {
Config::Config NewConfig; Config::Config NewConfig;
NewConfig.SetUUID(Now); NewConfig.SetUUID(Now);
D.Configuration = NewConfig.get(); D.Configuration = NewConfig.get();
} else {
Poco::JSON::Object Obj;
Obj.set("uuid", Now);
std::ostringstream os;
Obj.stringify(os);
D.Configuration = os.str();
}
} }
// We need to insert the country code according to the IP in the radios section... // We need to insert the country code according to the IP in the radios section...
D.locale = InsertRadiosCountyRegulation(D.Configuration, IPAddress); D.locale = InsertRadiosCountyRegulation(D.Configuration, IPAddress);
D.SerialNumber = Poco::toLower(SerialNumber); D.SerialNumber = Poco::toLower(SerialNumber);
D.Compatible = Caps.Compatible(); D.Compatible = Caps.Compatible();
if(D.Compatible.empty()) D.DeviceType = Daemon()->IdentifyDevice(D.Compatible);
D.Compatible = Caps.Model();
D.DeviceType = Poco::toLower(Caps.Platform());
D.MACAddress = Utils::SerialToMAC(SerialNumber); D.MACAddress = Utils::SerialToMAC(SerialNumber);
D.Manufacturer = Caps.Model(); D.Manufacturer = Caps.Model();
D.Firmware = Firmware; D.Firmware = Firmware;
@@ -625,13 +539,12 @@ namespace OpenWifi {
D.Notes = SecurityObjects::NoteInfoVec{ D.Notes = SecurityObjects::NoteInfoVec{
SecurityObjects::NoteInfo{(uint64_t)Utils::Now(), "", "Auto-provisioned."}}; SecurityObjects::NoteInfo{(uint64_t)Utils::Now(), "", "Auto-provisioned."}};
CreateDeviceCapabilities(Session, SerialNumber, Caps); CreateDeviceCapabilities(SerialNumber, Caps);
auto Result = CreateDevice(Session, D);
poco_information(Logger(), fmt::format("AUTO-CREATION({}): Done, Result={}", SerialNumber, Result)); return CreateDevice(D);
return Result;
} }
/* bool Storage::GetDeviceFWUpdatePolicy(std::string &SerialNumber, std::string &Policy) { bool Storage::GetDeviceFWUpdatePolicy(std::string &SerialNumber, std::string &Policy) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess); Poco::Data::Statement Select(Sess);
@@ -646,19 +559,16 @@ namespace OpenWifi {
} }
return false; return false;
} }
*/
bool Storage::SetDevicePassword(LockedDbSession &Sess, std::string &SerialNumber, std::string &Password) {
try {
std::lock_guard Lock(Sess.Mutex());
Sess.Session().begin();
Poco::Data::Statement Update(Sess.Session()); bool Storage::SetDevicePassword(std::string &SerialNumber, std::string &Password) {
try {
Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Update(Sess);
std::string St{"UPDATE Devices SET DevicePassword=? WHERE SerialNumber=?"}; std::string St{"UPDATE Devices SET DevicePassword=? WHERE SerialNumber=?"};
Update << ConvertParams(St), Poco::Data::Keywords::use(Password), Update << ConvertParams(St), Poco::Data::Keywords::use(Password),
Poco::Data::Keywords::use(SerialNumber); Poco::Data::Keywords::use(SerialNumber);
Update.execute(); Update.execute();
Sess.Session().commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
@@ -666,20 +576,34 @@ namespace OpenWifi {
return false; return false;
} }
std::string Storage::GetPlatform(const std::string &SerialNumber) { bool Storage::SetConnectInfo(std::string &SerialNumber, std::string &Firmware) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess); Poco::Data::Statement Select(Sess);
std::string St = fmt::format("SELECT DeviceType FROM Devices WHERE SerialNumber='{}'", SerialNumber); // Get the old version and if they do not match, set the last date
std::string Platform; std::string St{"SELECT Firmware FROM Devices WHERE SerialNumber=?"};
Select << ConvertParams(St), Poco::Data::Keywords::into(Platform); std::string TmpFirmware;
Select << ConvertParams(St), Poco::Data::Keywords::into(TmpFirmware),
Poco::Data::Keywords::use(SerialNumber);
Select.execute(); Select.execute();
return Platform;
if (TmpFirmware != Firmware) {
Poco::Data::Statement Update(Sess);
std::string St2{
"UPDATE Devices SET Firmware=?, LastFWUpdate=? WHERE SerialNumber=?"};
uint64_t Now = Utils::Now();
Update << ConvertParams(St2), Poco::Data::Keywords::use(Firmware),
Poco::Data::Keywords::use(Now), Poco::Data::Keywords::use(SerialNumber);
Update.execute();
return true;
}
return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);
} }
return ""; return false;
} }
bool Storage::DeleteDevice(std::string &SerialNumber) { bool Storage::DeleteDevice(std::string &SerialNumber) {
@@ -690,14 +614,12 @@ namespace OpenWifi {
for (const auto &tableName : TableNames) { for (const auto &tableName : TableNames) {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St = fmt::format("DELETE FROM {} WHERE SerialNumber='{}'", tableName, SerialNumber); std::string St = fmt::format("DELETE FROM {} WHERE SerialNumber='{}'", tableName, SerialNumber);
try { try {
Delete << St; Delete << St;
Delete.execute(); Delete.execute();
Sess.commit();
} catch (...) { } catch (...) {
} }
} }
@@ -770,14 +692,19 @@ namespace OpenWifi {
return false; return false;
} }
bool Storage::GetDevice(Poco::Data::Session &Session, const std::string &SerialNumber, GWObjects::Device &DeviceDetails) { bool Storage::GetDevice(std::string &SerialNumber, GWObjects::Device &DeviceDetails) {
try { try {
Poco::Data::Statement Select(Session); Poco::Data::Session Sess = Pool_->get();
std::string St = fmt::format("SELECT {} FROM Devices WHERE SerialNumber='{}'", DB_DeviceSelectFields, SerialNumber); Poco::Data::Statement Select(Sess);
std::string St{"SELECT " + DB_DeviceSelectFields +
" FROM Devices WHERE SerialNumber=?"};
DeviceRecordTuple R; DeviceRecordTuple R;
Select << St, Poco::Data::Keywords::into(R); Select << ConvertParams(St), Poco::Data::Keywords::into(R),
Poco::Data::Keywords::use(SerialNumber);
Select.execute(); Select.execute();
if (Select.rowsExtracted() == 0) if (Select.rowsExtracted() == 0)
return false; return false;
ConvertDeviceRecord(R, DeviceDetails); ConvertDeviceRecord(R, DeviceDetails);
@@ -788,26 +715,6 @@ namespace OpenWifi {
return false; return false;
} }
bool Storage::GetDevice(const std::string &SerialNumber, GWObjects::Device &DeviceDetails) {
try {
auto Sess = Pool_->get();
return GetDevice(Sess, SerialNumber, DeviceDetails);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::GetDevice(LockedDbSession &Session, const std::string &SerialNumber, GWObjects::Device &DeviceDetails) {
try {
std::lock_guard Lock(Session.Mutex());
return GetDevice(Session.Session(), SerialNumber, DeviceDetails);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::DeviceExists(std::string &SerialNumber) { bool Storage::DeviceExists(std::string &SerialNumber) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
@@ -834,26 +741,6 @@ namespace OpenWifi {
bool Storage::UpdateDevice(GWObjects::Device &NewDeviceDetails) { bool Storage::UpdateDevice(GWObjects::Device &NewDeviceDetails) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
return UpdateDevice(Sess, NewDeviceDetails);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::UpdateDevice(LockedDbSession &Session, GWObjects::Device &NewDeviceDetails) {
try {
std::lock_guard Lock(Session.Mutex());
return UpdateDevice(Session.Session(), NewDeviceDetails);
} catch (const Poco::Exception &E) {
Logger().log(E);
}
return false;
}
bool Storage::UpdateDevice(Poco::Data::Session &Sess, GWObjects::Device &NewDeviceDetails) {
try {
Sess.begin();
Poco::Data::Statement Update(Sess); Poco::Data::Statement Update(Sess);
DeviceRecordTuple R; DeviceRecordTuple R;
@@ -866,7 +753,6 @@ namespace OpenWifi {
Update << ConvertParams(St2), Poco::Data::Keywords::use(R), Update << ConvertParams(St2), Poco::Data::Keywords::use(R),
Poco::Data::Keywords::use(NewDeviceDetails.SerialNumber); Poco::Data::Keywords::use(NewDeviceDetails.SerialNumber);
Update.execute(); Update.execute();
Sess.commit();
// GetDevice(NewDeviceDetails.SerialNumber,NewDeviceDetails); // GetDevice(NewDeviceDetails.SerialNumber,NewDeviceDetails);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -876,38 +762,18 @@ namespace OpenWifi {
} }
bool Storage::GetDevices(uint64_t From, uint64_t HowMany, bool Storage::GetDevices(uint64_t From, uint64_t HowMany,
std::vector<GWObjects::Device> &Devices, const std::string &orderBy, const std::string &platform, std::vector<GWObjects::Device> &Devices, const std::string &orderBy) {
bool includeProvisioned) {
DeviceRecordList Records; DeviceRecordList Records;
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Select(Sess); Poco::Data::Statement Select(Sess);
std::string st; // std::string st{"SELECT " + DB_DeviceSelectFields + " FROM Devices " + orderBy.empty()
std::string whereClause = ""; // ? " ORDER BY SerialNumber ASC " + ComputeRange(From, HowMany)};
if(platform.empty()) { std::string st = fmt::format("SELECT {} FROM Devices {} {}", DB_DeviceSelectFields,
if (includeProvisioned == false) {
whereClause = fmt::format("WHERE entity='' and venue=''");
}
} else {
if (includeProvisioned == false) {
whereClause = fmt::format("WHERE DeviceType='{}' and entity='' and venue=''",platform);
} else {
whereClause = fmt::format("WHERE DeviceType='{}'", platform);
}
}
st =
fmt::format("SELECT {} FROM Devices {} {} {}", DB_DeviceSelectFields, whereClause,
orderBy.empty() ? " ORDER BY SerialNumber ASC " : orderBy, orderBy.empty() ? " ORDER BY SerialNumber ASC " : orderBy,
ComputeRange(From, HowMany)); ComputeRange(From, HowMany));
//Logger().information(fmt::format(" GetDevices st is {} ", st));
Select << ConvertParams(st), Poco::Data::Keywords::into(Records); Select << ConvertParams(st), Poco::Data::Keywords::into(Records);
Select.execute(); Select.execute();
@@ -1135,9 +1001,9 @@ namespace OpenWifi {
} }
} }
uint64_t Associations_2G, Associations_5G, Associations_6G, uptime; uint64_t Associations_2G, Associations_5G, Associations_6G;
StateUtils::ComputeAssociations(RawObject, Associations_2G, Associations_5G, StateUtils::ComputeAssociations(RawObject, Associations_2G, Associations_5G,
Associations_6G, uptime); Associations_6G);
UpdateCountedMap(Dashboard.associations, "2G", Associations_2G); UpdateCountedMap(Dashboard.associations, "2G", Associations_2G);
UpdateCountedMap(Dashboard.associations, "5G", Associations_5G); UpdateCountedMap(Dashboard.associations, "5G", Associations_5G);
UpdateCountedMap(Dashboard.associations, "6G", Associations_6G); UpdateCountedMap(Dashboard.associations, "6G", Associations_6G);
@@ -1161,25 +1027,4 @@ namespace OpenWifi {
FieldList.push_back(field); FieldList.push_back(field);
} }
void Storage::FixDeviceTypeBug() {
try {
std::vector<std::string> ScriptLines{
"update devices set devicetype='ap' where devicetype='AP';",
"update devices set devicetype='switch' where devicetype='SWITCH';",
"update devices set devicetype='ap' where devicetype!='ap' and devicetype!='switch';"
};
for (const auto &ScriptLine : ScriptLines) {
try {
Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement SqlStatement(Sess);
SqlStatement << ScriptLine, Poco::Data::Keywords::now;
} catch (...) {
}
}
} catch (const Poco::Exception &E) {
Logger().log(E);
}
}
} // namespace OpenWifi } // namespace OpenWifi

View File

@@ -35,11 +35,10 @@ namespace OpenWifi {
R.set<4>(H.Recorded); R.set<4>(H.Recorded);
} }
bool Storage::AddHealthCheckData(LockedDbSession &Session, const GWObjects::HealthCheck &Check) { bool Storage::AddHealthCheckData(const GWObjects::HealthCheck &Check) {
try { try {
std::lock_guard Guard(Session.Mutex()); Poco::Data::Session Sess = Pool_->get();
Session.Session().begin(); Poco::Data::Statement Insert(Sess);
Poco::Data::Statement Insert(Session.Session());
std::string St{"INSERT INTO HealthChecks ( " + DB_HealthCheckSelectFields + std::string St{"INSERT INTO HealthChecks ( " + DB_HealthCheckSelectFields +
" ) VALUES( " + DB_HealthCheckInsertValues + " )"}; " ) VALUES( " + DB_HealthCheckInsertValues + " )"};
@@ -48,7 +47,6 @@ namespace OpenWifi {
ConvertHealthCheckRecord(Check, R); ConvertHealthCheckRecord(Check, R);
Insert << ConvertParams(St), Poco::Data::Keywords::use(R); Insert << ConvertParams(St), Poco::Data::Keywords::use(R);
Insert.execute(); Insert.execute();
Session.Session().commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -136,7 +134,7 @@ namespace OpenWifi {
uint64_t ToDate) { uint64_t ToDate) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
bool DatesIncluded = (FromDate != 0 || ToDate != 0); bool DatesIncluded = (FromDate != 0 || ToDate != 0);
std::string Prefix{"DELETE FROM HealthChecks "}; std::string Prefix{"DELETE FROM HealthChecks "};
@@ -160,7 +158,7 @@ namespace OpenWifi {
Delete << Statement + DateSelector; Delete << Statement + DateSelector;
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -172,13 +170,11 @@ namespace OpenWifi {
bool Storage::RemoveHealthChecksRecordsOlderThan(uint64_t Date) { bool Storage::RemoveHealthChecksRecordsOlderThan(uint64_t Date) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St1{"delete from HealthChecks where recorded<?"}; std::string St1{"delete from HealthChecks where recorded<?"};
Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date); Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),

View File

@@ -39,11 +39,11 @@ namespace OpenWifi {
R.set<6>(Log.UUID); R.set<6>(Log.UUID);
} }
bool Storage::AddLog(LockedDbSession &Session, const GWObjects::DeviceLog &Log) { bool Storage::AddLog(const GWObjects::DeviceLog &Log) {
try { try {
std::lock_guard Guard(Session.Mutex());
Session.Session().begin(); Poco::Data::Session Sess = Pool_->get();
Poco::Data::Statement Insert(Session.Session()); Poco::Data::Statement Insert(Sess);
std::string St{"INSERT INTO DeviceLogs (" + DB_LogsSelectFields + ") values( " + std::string St{"INSERT INTO DeviceLogs (" + DB_LogsSelectFields + ") values( " +
DB_LogsInsertValues + " )"}; DB_LogsInsertValues + " )"};
@@ -53,7 +53,6 @@ namespace OpenWifi {
Insert << ConvertParams(St), Poco::Data::Keywords::use(R); Insert << ConvertParams(St), Poco::Data::Keywords::use(R);
Insert.execute(); Insert.execute();
Session.Session().commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -115,7 +114,7 @@ namespace OpenWifi {
uint64_t Type) { uint64_t Type) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
bool DatesIncluded = (FromDate != 0 || ToDate != 0); bool DatesIncluded = (FromDate != 0 || ToDate != 0);
bool HasWhere = DatesIncluded || !SerialNumber.empty(); bool HasWhere = DatesIncluded || !SerialNumber.empty();
@@ -142,7 +141,7 @@ namespace OpenWifi {
Delete << StatementStr + DateSelector + TypeSelector; Delete << StatementStr + DateSelector + TypeSelector;
Delete.execute(); Delete.execute();
Sess.commit(); Delete.reset(Sess);
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
@@ -184,13 +183,11 @@ namespace OpenWifi {
bool Storage::RemoveDeviceLogsRecordsOlderThan(uint64_t Date) { bool Storage::RemoveDeviceLogsRecordsOlderThan(uint64_t Date) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
Poco::Data::Statement Delete(Sess); Poco::Data::Statement Delete(Sess);
std::string St1{"delete from DeviceLogs where recorded<?"}; std::string St1{"delete from DeviceLogs where recorded<?"};
Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date); Delete << ConvertParams(St1), Poco::Data::Keywords::use(Date);
Delete.execute(); Delete.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),

View File

@@ -8,6 +8,7 @@
#include "AP_WS_Server.h" #include "AP_WS_Server.h"
#include "StorageService.h" #include "StorageService.h"
#include "fmt/format.h" #include "fmt/format.h"
namespace OpenWifi { namespace OpenWifi {
@@ -32,10 +33,10 @@ namespace OpenWifi {
R.set<3>(Stats.Recorded); R.set<3>(Stats.Recorded);
} }
bool Storage::AddStatisticsData(Poco::Data::Session &Session, const GWObjects::Statistics &Stats) { bool Storage::AddStatisticsData(const GWObjects::Statistics &Stats) {
try { try {
Session.begin(); Poco::Data::Session Sess(Pool_->get());
Poco::Data::Statement Insert(Session); Poco::Data::Statement Insert(Sess);
poco_trace(Logger(), fmt::format("{}: Adding stats. Size={}", Stats.SerialNumber, poco_trace(Logger(), fmt::format("{}: Adding stats. Size={}", Stats.SerialNumber,
std::to_string(Stats.Data.size()))); std::to_string(Stats.Data.size())));
@@ -45,7 +46,6 @@ namespace OpenWifi {
ConvertStatsRecord(Stats, R); ConvertStatsRecord(Stats, R);
Insert << ConvertParams(St), Poco::Data::Keywords::use(R); Insert << ConvertParams(St), Poco::Data::Keywords::use(R);
Insert.execute(); Insert.execute();
Session.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), fmt::format("{}: Failed with: {}", std::string(__func__),
@@ -167,7 +167,7 @@ namespace OpenWifi {
uint64_t ToDate) { uint64_t ToDate) {
try { try {
Poco::Data::Session Sess = Pool_->get(); Poco::Data::Session Sess = Pool_->get();
Sess.begin();
bool DatesIncluded = (FromDate != 0 || ToDate != 0); bool DatesIncluded = (FromDate != 0 || ToDate != 0);
std::string Prefix{"DELETE FROM Statistics "}; std::string Prefix{"DELETE FROM Statistics "};
@@ -189,7 +189,7 @@ namespace OpenWifi {
Poco::Data::Statement Select(Sess); Poco::Data::Statement Select(Sess);
Select << Statement + DateSelector; Select << Statement + DateSelector;
Select.execute(); Select.execute();
Sess.commit();
return true; return true;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
poco_warning(Logger(), (fmt::format("{}: Failed with: {}", std::string(__func__), poco_warning(Logger(), (fmt::format("{}: Failed with: {}", std::string(__func__),

View File

@@ -275,21 +275,9 @@ namespace OpenWifi {
"Models TEXT, " "Models TEXT, "
"Description TEXT, " "Description TEXT, "
"Created BIGINT , " "Created BIGINT , "
"LastModified BIGINT, Platform TEXT)", "LastModified BIGINT)",
Poco::Data::Keywords::now; Poco::Data::Keywords::now;
} }
std::vector<std::string> Script{
"alter table DefaultConfigs add column Platform text"
};
for (const auto &i : Script) {
try {
Sess << i, Poco::Data::Keywords::now;
} catch (...) {
}
}
return 0; return 0;
} catch (const Poco::Exception &E) { } catch (const Poco::Exception &E) {
Logger().log(E); Logger().log(E);