spark_counter/rpi/spark_counter_receiver.cpp

124 lines
4.7 KiB
C++

/* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
/* Copyright (c) 2015 King Kévin <kingkevin@cuvoodoo.info> */
/* the spark counter receives electricity measurements over radio
* they are then received using an nRF2L01+ transceiver
* to communicate with the transceiver the RF24 library is requiered
* they are then stored in an InfluxDB time series database using the HTTP API
* to send values to the database the curl library is required
*/
#include <unistd.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <RF24/RF24.h> // http://tmrh20.github.io/RF24 library to communicate to the nRF24L01+
#include <curl/curl.h> // curl library to send the measurement data to the influxDB.
#include "aes.h" // AES library (from tiny-AES128-C)
// Setup for RPi B1 GPIO 22 CE and CE0 CSN with SPI Speed @ 8Mhz
RF24 radio(RPI_V2_GPIO_P1_22, BCM2835_SPI_CS0, BCM2835_SPI_SPEED_8MHZ);
// nRF24L01+ addresses
const uint8_t tx_addr[] = {1,'h','o','m','e'};
const uint8_t rx_addr[] = {0,'h','o','m','e'};
CURL *curl; // curl handle to post data to influxbd using the HTTP API
// key material
const uint8_t key[16] = {0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff};
uint8_t iv[16] = {0xff,0xee,0xdd,0xcc,0xbb,0xaa,0x99,0x88,0x77,0x66,0x55,0x44,0x33,0x22,0x11,0x00};
int main(int argc, char** argv){
// configure influxdb connection
curl = curl_easy_init();
if(curl) {
curl_easy_setopt(curl, CURLOPT_URL, "http://localhost:8086/write?db=electricity");
}
// configure nRF24L01+ radio
radio.begin();
radio.setChannel(42);
radio.setPALevel(RF24_PA_MAX);
radio.setDataRate(RF24_1MBPS);
radio.setAutoAck(true);
radio.enableDynamicPayloads();
radio.setRetries(2,15);
radio.setCRCLength(RF24_CRC_8);
radio.openWritingPipe(tx_addr);
radio.openReadingPipe(1,rx_addr);
//radio.printDetails();
printf("wait for packet to arrive\n");
radio.startListening();
while (1) { // forever loop
while (radio.available()) {
time_t ltime; // calendar time
ltime=time(NULL); // get current cal time
char stime[64]; // do display the time
strftime(stime, sizeof(stime), "%F %T %z %Z", localtime(&ltime));
printf("%s: payload received\n",stime);
uint8_t payload[32]; // buffer to save the payload
uint8_t size = radio.getDynamicPayloadSize();
radio.read(&payload,size);
/*
printf("got %d bytes:",size);
for (uint8_t i=0; i<size; i++) {
printf(" %02x",payload[i]);
}
printf("\n");
*/
// got through payload
if (size!=1+16) {
continue;
}
uint8_t id = payload[0]; // the meter id (0 is myself, used for unknown source)
float voltage, current, power, energy; // the values coming from the meter
uint8_t values[16]; // the encrypted values block
AES128_ECB_decrypt(&payload[1], key, values); // decrypt payload
for (uint8_t i=0; i<sizeof(values) && i<sizeof(iv); i++) { // use CBC mode
values[i] ^= iv[i]; // XOR with last IV
}
memcpy(iv,&payload[1],sizeof(iv)); // save next IV
memcpy(&voltage,&values[0],4); // read voltage
memcpy(&current,&values[4],4); // read current
memcpy(&power,&values[8],4); // read power
memcpy(&energy,&values[12],4); // read energy
printf("meter: %d, voltage: %f V, current: %f A, power: %f W, energy: %f Wh\n",id,voltage,current,power,energy);
if (curl) {
CURLcode res = CURLE_OK; // curl response
char post[128*4] = {0}; // string to submit data to DB using POST request
snprintf(post, sizeof(post), "voltage,meter=%d value=%f\ncurrent,meter=%d value=%f\npower,meter=%d value=%f\nenergy,meter=%d value=%f\n", id, voltage, id, current, id, power, id, energy);
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, post);
res = curl_easy_perform(curl);
if (res!= CURLE_OK) {
fprintf(stderr, "curl_easy_perform() failed: %s\n", curl_easy_strerror(res));
}
}
}
usleep(100000); // wait 100ms before request if data is available again (the IRQ signal is not used)
}
if (curl) {
curl_easy_cleanup(curl);
}
}