add fits-header to weather_proxy, fixed something

This commit is contained in:
Edward Emelianov
2026-04-24 23:12:29 +03:00
parent 5acd1cd97d
commit 7b2d93299d
5 changed files with 119 additions and 57 deletions

View File

@@ -16,7 +16,6 @@
* along with this program. If not, see <http://www.gnu.org/licenses/>. * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/ */
#include "weather_data.h"
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
@@ -27,8 +26,7 @@
#include <semaphore.h> #include <semaphore.h>
#include <unistd.h> #include <unistd.h>
#define SHM_NAME "/weather_shm" #include "weather_data.h"
#define SEM_NAME "/weather_sem"
int get_weather_data(weather_data_t *data) { int get_weather_data(weather_data_t *data) {
int shm_fd; int shm_fd;

View File

@@ -22,8 +22,12 @@
int main() { int main() {
weather_data_t wd; weather_data_t wd;
if(get_weather_data(&wd) == 0){ if(get_weather_data(&wd) == 0){
printf("Weather: %d, Max wind: %.1f, Wind: %.1f, Temp: %.1f; updated @%zd\n", char strt[64];
wd.weather, wd.windmax, wd.wind, wd.exttemp, wd.last_update); struct tm *T = localtime(&wd.last_update);
strftime(strt, 63, "%F %T", T);
printf("Prohibited: %d\nWeather: %d\nMax wind: %.1f\nWind: %.1f\nTemp: %.1f\nPressure: %.1f\nHumidity: %.1f\nupdated @%zd (%s)\n",
wd.prohibited, wd.weather, wd.windmax, wd.wind, wd.exttemp, wd.pressure, wd.humidity,
wd.last_update, strt);
}else{ }else{
fprintf(stderr, "Failed to get weather data\n"); fprintf(stderr, "Failed to get weather data\n");
} }

View File

@@ -15,19 +15,12 @@
* You should have received a copy of the GNU General Public License * You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>. * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/ */
#if 0
#include <stdlib.h>
#include <errno.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdarg.h>
#endif
#include <fcntl.h> #include <fcntl.h>
#include <semaphore.h> #include <semaphore.h>
#include <signal.h> #include <signal.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h>
#include <string.h> #include <string.h>
#include <sys/mman.h> #include <sys/mman.h>
#include <sys/prctl.h> #include <sys/prctl.h>
@@ -43,19 +36,18 @@
#define DEFAULT_PID "/tmp/weather_proxy.pid" #define DEFAULT_PID "/tmp/weather_proxy.pid"
#define DEAD_TMOUT 15 // if we have no fresh data more than `RECONN_TMOUT`, try to reconect
#define RECONN_TMOUT 5 #define RECONN_TMOUT 5
// don't ask new data less than `WEAT_TMOUT` seconds
#define WEAT_TMOUT 1 #define WEAT_TMOUT 1
#define SHM_NAME "/weather_shm"
#define SEM_NAME "/weather_sem"
typedef struct{ typedef struct{
char *node; // node of server char *node; // node of server
int isunix; // use UNIX-sockets instead of net int isunix; // use UNIX-sockets instead of net
char *logfile; // logfile name char *logfile; // logfile name
int verb; // verbocity level int verb; // verbocity level
char *pidfile; // pidfile name char *pidfile; // pidfile name
char *fitsheader; // FITS-header with collected weather data
} glob_pars; } glob_pars;
static pid_t childpid; static pid_t childpid;
@@ -64,14 +56,15 @@ static int forbidden = 0;
static sem_t *sem = NULL; static sem_t *sem = NULL;
static weather_data_t *shared_data = NULL; static weather_data_t *shared_data = NULL;
static volatile int running = 1; static volatile int running = 1;
static glob_pars G = {0}; static glob_pars G = {.pidfile = DEFAULT_PID};
static sl_option_t opts[] = { static sl_option_t opts[] = {
{"node", NEED_ARG, NULL, 'n', arg_string, APTR(&G.node), "node to connect (host:port or UNIX socket name)"}, {"node", NEED_ARG, NULL, 'n', arg_string, APTR(&G.node), "node to connect (host:port or UNIX socket name)"},
{"logfile", NEED_ARG, NULL, 'l', arg_string, APTR(&G.logfile), "save logs to file"}, {"logfile", NEED_ARG, NULL, 'l', arg_string, APTR(&G.logfile), "save logs to file"},
{"pidfile", NEED_ARG, NULL, 'p', arg_string, APTR(&G.pidfile), "pidfile name (default: " DEFAULT_PID ")"}, {"pidfile", NEED_ARG, NULL, 'p', arg_string, APTR(&G.pidfile), "pidfile name (default: " DEFAULT_PID ")"},
{"isunix", NO_ARGS, NULL, 'u', arg_string, APTR(&G.isunix), "use UNIX socket instead of network"}, {"isunix", NO_ARGS, NULL, 'u', arg_string, APTR(&G.isunix), "use UNIX socket instead of network"},
{"verbose", NO_ARGS, NULL, 'v', arg_int, APTR(&G.verb), "verbose level (each -v increases)"}, {"verbose", NO_ARGS, NULL, 'v', arg_none, APTR(&G.verb), "verbose level (each -v increases)"},
{"fitsheader",NEED_ARG, NULL, 'f', arg_string, APTR(&G.fitsheader),"fits-header for weather data"},
end_option end_option
}; };
@@ -96,13 +89,14 @@ void signals(int signo){
if(childpid){ // master if(childpid){ // master
LOGERR("Main process exits with status %d", signo); LOGERR("Main process exits with status %d", signo);
if(G.pidfile) unlink(G.pidfile); if(G.pidfile) unlink(G.pidfile);
if(G.fitsheader) unlink(G.fitsheader);
exit(1); exit(1);
}else{ // child }else{ // child
LOGERR("Killed with status %d", signo); if(running){
LOGERR("Stop running");
running = 0; // let make cleanup running = 0; // let make cleanup
} }
sleep(1); }
exit(signo); // force exit if stubs
} }
static int init_ipc(void){ static int init_ipc(void){
@@ -166,10 +160,54 @@ static void cleanup_ipc(void){
} }
} }
// update record of FITS header
static void FITS_update(const char *line, int finish){
static FILE *tmp = NULL;
static char templ[32]; // temporary file name
if(!tmp){ // try to create new temporary file
sprintf(templ, "/tmp/fitshdrXXXXXX");
int fd = mkstemp(templ);
if(fd < 0){
WARN("mkstemp()");
LOGERR("Can't create temporary file!");
return;
}
tmp = fdopen(fd, "w");
if(!tmp){
WARN("fdopen()");
LOGERR("Error in fdopen()");
return;
}
}
fprintf(tmp, "%s\n", line);
if(finish){ // move temporary file into new location
fclose(tmp);
tmp = NULL;
chmod(templ, 0644);
if(rename(templ, G.fitsheader) < 0){
WARN("rename(%s, %s)", templ, G.fitsheader);
LOGERR("Error in rename()");
}
}
}
static void update_shm(weather_data_t *data){
if(sem_wait(sem) == -1){
LOGWARN("sem_wait failed: %s", strerror(errno));
}else{
memcpy(shared_data, data, sizeof(weather_data_t));
sem_post(sem);
LOGDBG("Weather data updated");
}
}
static void parse_line(const char *line, weather_data_t *data) { static void parse_line(const char *line, weather_data_t *data) {
char key[64]; char key[SL_KEY_LEN];
char value[256]; char value[SL_VAL_LEN];
if (sscanf(line, "%63[^=]=%255s", key, value) == 2) {
int update = 0; // 0 for updating, 1 for finishing, -1 for error
if(sl_get_keyval(line, key, value)){
if(strcmp(key, "WEATHER") == 0){ if(strcmp(key, "WEATHER") == 0){
data->weather = (weather_condition_t) atoi(value); data->weather = (weather_condition_t) atoi(value);
printf("got weather: %d\n", data->weather); printf("got weather: %d\n", data->weather);
@@ -196,18 +234,16 @@ static void parse_line(const char *line, weather_data_t *data) {
printf("got humidity: %g\n", data->humidity); printf("got humidity: %g\n", data->humidity);
}else if (strcmp(key, "PROHIBIT") == 0){ }else if (strcmp(key, "PROHIBIT") == 0){
data->prohibited = atoi(value); data->prohibited = atoi(value);
} else if (strcmp(key, "TMEAS") == 0) { }else if (strcmp(key, "TMEAS") == 0){ // last line in message -> update
data->last_update = atof(value); data->last_update = atof(value);
if(data->weather == WEATHER_PROHIBITED || forbidden) data->prohibited = 1; if(data->weather == WEATHER_PROHIBITED || forbidden) data->prohibited = 1;
else if(data->weather < WEATHER_TERRIBLE) data->prohibited = 0; else if(data->weather < WEATHER_PROHIBITED) data->prohibited = 0;
// update all // update all
if (sem_wait(sem) == -1) { update_shm(data);
LOGWARN("sem_wait failed: %s", strerror(errno)); update = 1;
} else { }else update = -1;
memcpy(shared_data, data, sizeof(weather_data_t)); if(update > -1 && G.fitsheader){
sem_post(sem); FITS_update(line, update);
LOGMSG("Weather data updated");
}
} }
} }
} }
@@ -222,7 +258,7 @@ static int request_weather_data(sl_sock_t *sock){
DBG("try to send request: '%s", request); DBG("try to send request: '%s", request);
if(sl_sock_sendstrmessage(sock, request) < 1){ if(sl_sock_sendstrmessage(sock, request) < 1){
LOGERR("Can't poll new data"); LOGWARN("Can't poll new data");
return -1; return -1;
} }
return 0; return 0;
@@ -239,32 +275,52 @@ static void run_daemon(){
LOGERR("Can't connect to meteodaemon over socket with node %s", G.node); LOGERR("Can't connect to meteodaemon over socket with node %s", G.node);
return; return;
} }
LOGMSG("Connected to meteodaemon %s", G.node);
memcpy(&new_data, shared_data, sizeof(weather_data_t)); memcpy(&new_data, shared_data, sizeof(weather_data_t));
time_t lastert = time(NULL); time_t lastert = time(NULL);
while(running){ while(running){
time_t tnow = time(NULL); time_t tnow = time(NULL);
if(!sock || request_weather_data(sock) == -1){ int req = -1;
if(tnow - lastert > RECONN_TMOUT){ // try to reconnect if(sock) req = request_weather_data(sock);
if(req == -1){
int diff = tnow - lastert;
DBG("diff = %d", diff);
if(diff > RECONN_TMOUT){ // try to reconnect
LOGERR("Failed to request weather data, retry"); LOGERR("Failed to request weather data, retry");
if(sock) sl_sock_delete(&sock); if(sock) sl_sock_delete(&sock);
if(!(sock = sl_sock_run_client(stype, G.node, 4096))) lastert += 5; if(!(sock = sl_sock_run_client(stype, G.node, 4096))){
else lastert = tnow; new_data.weather = WEATHER_TERRIBLE; // no connection to weather server, don't allow to open
update_shm(&new_data);
lastert += RECONN_TMOUT;
}else{
LOGMSG("Reconnected to %s", G.node);
lastert = tnow;
} }
}else lastert = tnow; }
}else if(req == 0) lastert = tnow;
while(sl_sock_readline(sock, line, 255) > 0){ while(sl_sock_readline(sock, line, 255) > 0){
DBG("Parse '%s'", line); DBG("Parse '%s'", line);
parse_line(line, &new_data); parse_line(line, &new_data);
} }
usleep(500000);
} }
sl_sock_delete(&sock); // disconnect and clear memory sl_sock_delete(&sock); // disconnect and clear memory
DBG("run_daemon() exited");
} }
int main(int argc, char *argv[]){ int main(int argc, char *argv[]){
sl_init(); sl_init();
sl_parseargs(&argc, &argv, opts); sl_parseargs(&argc, &argv, opts);
if(!G.node) ERRX("Point node to connect"); if(!G.node) ERRX("Point node to connect");
if(G.fitsheader){
FILE *fitsfile = fopen(G.fitsheader, "w");
if(!fitsfile){
WARN("Can't create FITS header %s", G.fitsheader);
FREE(G.fitsheader);
}else fclose(fitsfile);
}
sl_check4running(NULL, G.pidfile); sl_check4running(NULL, G.pidfile);
if(G.logfile){ if(G.logfile){
sl_loglevel_e lvl = LOGLEVEL_ERR + G.verb; sl_loglevel_e lvl = LOGLEVEL_ERR + G.verb;
@@ -304,7 +360,8 @@ int main(int argc, char *argv[]){
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
run_daemon(); run_daemon();
LOGDBG("Daemon is dead");
cleanup_ipc(); cleanup_ipc();
LOGMSG("Daemon is dead"); LOGDBG("IPC cleaned");
return 0; return 0;
} }

View File

@@ -3,11 +3,14 @@
#include <stdint.h> #include <stdint.h>
#include <time.h> #include <time.h>
#define SHM_NAME "/weather_shm"
#define SEM_NAME "/weather_sem"
typedef enum { typedef enum {
WEATHER_GOOD = 0, WEATHER_GOOD = 0, // may start observations
WEATHER_BAD = 1, WEATHER_BAD = 1, // cannot start but can continue if want
WEATHER_TERRIBLE = 2, WEATHER_TERRIBLE = 2, // close & park: wind, precipitation, humidity etc.
WEATHER_PROHIBITED = 3, WEATHER_PROHIBITED = 3, // force closing & parking; power off equipment, ready to power off computer
} weather_condition_t; } weather_condition_t;
typedef struct { typedef struct {
@@ -19,7 +22,7 @@ typedef struct {
float pressure; // atm. pressure, mmHg: "PRESSURE" float pressure; // atm. pressure, mmHg: "PRESSURE"
float humidity; // humidity, percents: "HUMIDITY" float humidity; // humidity, percents: "HUMIDITY"
int rain; // ==1 when rainy: "PRECIP" int rain; // ==1 when rainy: "PRECIP"
int prohibited; // ==1 if "weather == prohibited" or rain == 1 int prohibited; // ==1 if "weather == prohibited" or got `prohibited` signal -> ready to power off
time_t last_update; // value of "TMEAS" time_t last_update; // value of "TMEAS"
} weather_data_t; } weather_data_t;

View File

@@ -32,7 +32,7 @@ int help;
// default values for Gdefault & help // default values for Gdefault & help
#define DEFAULT_PORT "12345" #define DEFAULT_PORT "12345"
#define DEFAULT_PID "/tmp/weatherdaemon.pid" #define DEFAULT_PID "/tmp/superweatherdaemon.pid"
// DEFAULTS // DEFAULTS
// default global parameters // default global parameters