581 lines
20 KiB
C
581 lines
20 KiB
C
#include<stdio.h>
|
|
#include<stdlib.h>
|
|
#include<curl/curl.h>
|
|
#include<string.h>
|
|
#include<dbus/dbus.h>
|
|
#include<stdbool.h>
|
|
#include<time.h>
|
|
#include<unistd.h>
|
|
#include<pwd.h>
|
|
|
|
#include"config.h"
|
|
|
|
#define HELP_TEXT "requires a command\n--\ngetSendStatus [objectpath]\nlistUnits\nconfig [configfile / -d]\nset token with GOTIFY_TOKEN=\"sjdfoiasfd\"\n--\n"
|
|
|
|
typedef struct service
|
|
{
|
|
char *name;
|
|
char *targetPath;
|
|
char *host;
|
|
bool alerted;
|
|
int priority;
|
|
} Service;
|
|
|
|
int gotify_message(char *,char *,char *);
|
|
int systemdCall(DBusConnection *,char *, char *);
|
|
int checkIsActive(DBusConnection *connection, char *service_name);
|
|
int systemdUnitCall(DBusConnection *connection,char *target,char *method,char *param,char **result);
|
|
int GetSendStatus(DBusConnection *connection,char *objectpath);
|
|
int ServiceGetSendStatus(DBusConnection *connection,Service *service);
|
|
int checkState(char *ServiceName,char *result,Service *service);
|
|
int getPath(char *configPath,int size);
|
|
/*
|
|
*sytemctl show gotify-server | grep PID
|
|
*sudo busctl tree org.freedesktop.systemd1 > get object paths
|
|
//org/freedesktop/systemd1
|
|
//gotify_message("https://5ccppi.org:4433/message?token=","anothertitle","anothermessage");
|
|
* */
|
|
|
|
|
|
int main(int argc,char *argv[])
|
|
{
|
|
DBusConnection *connection=NULL;
|
|
DBusError error;
|
|
clock_t time;
|
|
|
|
|
|
time = clock();
|
|
|
|
printf("time:%f\n",(float)time / CLOCKS_PER_SEC);
|
|
|
|
dbus_error_init(&error);
|
|
connection = dbus_bus_get(DBUS_BUS_SYSTEM,&error);//DBUS_BUS_SYSTEM DBUS_BUS_SESSION
|
|
if(dbus_error_is_set(&error))
|
|
{
|
|
fprintf(stderr,"%s",error.message);
|
|
return -1;
|
|
}
|
|
printf("This is my unique dbus name: %s\n",dbus_bus_get_unique_name(connection));
|
|
|
|
if(argc>1)
|
|
{
|
|
if(strcmp(argv[1],"getSendStatus")==0)
|
|
{
|
|
if(argc>2)
|
|
{
|
|
if(argv[2][0]=='/')
|
|
{
|
|
GetSendStatus(connection,argv[2]);
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr,"\n must be in the form of /org.freedesktop.systemd1/...\n");
|
|
return 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr,"%s needs 1 argument\n",argv[1]);
|
|
return 2;
|
|
}
|
|
}
|
|
else if(strcmp(argv[1],"listUnits")==0)
|
|
{
|
|
systemdCall(connection,"ListUnits",NULL);
|
|
}
|
|
else if(strcmp(argv[1],"config")==0)
|
|
{
|
|
if(argc==3 && strcmp(argv[2],"-d")!=0)
|
|
{
|
|
char *str_entry[64]={'\0'};
|
|
char *host=NULL;
|
|
int i=0;
|
|
|
|
loadConfig(argv[2],str_entry,&host,64);
|
|
for(i=0;str_entry[i];i++)
|
|
{
|
|
printf("entry in config:%s\n",str_entry[i]);
|
|
GetSendStatus(connection,str_entry[i]);
|
|
}
|
|
}
|
|
else if(argc==3 && strcmp(argv[2],"-d")==0)
|
|
{
|
|
printf("Running as a daemon\n");
|
|
char *str_entry[64]={'\0'};
|
|
int i=0;
|
|
Service services[64]={'\0'};
|
|
char *ptr_name=NULL;
|
|
char path[256];
|
|
char *host=NULL;
|
|
|
|
getPath(path,256);
|
|
printf("[main]configPath:%s\n",path);
|
|
|
|
loadConfig(path,str_entry,&host,64);
|
|
printf("[main]host:%s\n",host);
|
|
for(i=0;str_entry[i];i++)
|
|
{
|
|
services[i].targetPath = strdup(str_entry[i]);
|
|
ptr_name = strrchr(services[i].targetPath,'/');
|
|
if(ptr_name!=NULL)
|
|
{
|
|
services[i].name = strdup(ptr_name);
|
|
services[i].name = services[i].name + 1;
|
|
}
|
|
else{
|
|
fprintf(stderr,"does not seem to be a valid target\n");
|
|
services[i].name = strdup("Undefined");
|
|
}
|
|
services[i].alerted = false;
|
|
services[i].priority = 6;
|
|
services[i].host = host;
|
|
printf("Name:%s\ntarget:%s\n",services[i].name,services[i].targetPath);
|
|
}
|
|
if(services[0].name)
|
|
{
|
|
while(true)
|
|
{
|
|
for(i=0;services[i].name;i++)
|
|
{
|
|
printf("entry in config:%s\n",services[i].name);
|
|
ServiceGetSendStatus(connection,&services[i]);
|
|
}
|
|
sleep(10);
|
|
}
|
|
}else
|
|
{
|
|
fprintf(stderr,"No service found in config file, please check %s\n",path);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr,"config requires a config file as argument\nAppend -d for daemonizing\n");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr,"Unknown command\n%s",HELP_TEXT);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr,HELP_TEXT);
|
|
}
|
|
return 0;
|
|
}
|
|
int getPath(char *configPath,int size)
|
|
{
|
|
uid_t myid = getuid();
|
|
printf("My uid:%d\n",myid);
|
|
struct passwd *info = getpwuid(myid);
|
|
printf("My dir:%s\n",info->pw_dir);
|
|
snprintf(configPath,size,"%s/.config/systemd-gotify/config",info->pw_dir);
|
|
printf("ConfigPath:%s\n",configPath);
|
|
return 0;
|
|
}
|
|
int DBusOptions(DBusConnection *connection,DBusError **error) //not tested check pointer of error!!
|
|
{
|
|
int ret=0,i=0;
|
|
dbus_bool_t allow_user=0;
|
|
char **objects = NULL;
|
|
unsigned long uid;
|
|
|
|
ret = dbus_bus_request_name(connection,"test.method.client",DBUS_NAME_FLAG_REPLACE_EXISTING,*error);
|
|
if(dbus_error_is_set(*error))
|
|
{
|
|
fprintf(stderr,"%s",(**error).message);
|
|
return -1;
|
|
}else
|
|
{
|
|
printf("name requested from dbus: %d\n",ret);
|
|
}
|
|
|
|
dbus_connection_set_allow_anonymous(connection,allow_user);
|
|
printf("user auth dbus requ: %d\n",allow_user);
|
|
if(!dbus_connection_list_registered(connection,"/",&objects))
|
|
{
|
|
for(i=0;objects[i];i++)
|
|
{
|
|
printf("\nobject dbus: %s",objects[i]);
|
|
}
|
|
}else{
|
|
printf("dbus_connection_list_registered error\n");
|
|
}
|
|
|
|
if((ret=dbus_connection_get_unix_user(connection,&uid))==1)
|
|
{
|
|
printf("found USERID:%ld",uid);
|
|
}else
|
|
{
|
|
printf("NO userID set in dbus session, servers normali do not authenthicate against users/slaves: %d\n",ret);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int GetSendStatus(DBusConnection *connection,char *objectpath)
|
|
{
|
|
char *result=0;
|
|
int ret = systemdUnitCall(connection,objectpath,"Get","SubState",&result);
|
|
if(ret == 0)
|
|
{
|
|
//checkState(objectpath,result,NULL);
|
|
} else
|
|
{
|
|
fprintf(stderr,"could not call systemdunit\n");
|
|
}
|
|
return ret;
|
|
}
|
|
int ServiceGetSendStatus(DBusConnection *connection,Service *service)
|
|
{
|
|
char *result=0;
|
|
char *objectpath = service->targetPath;
|
|
int ret = systemdUnitCall(connection,objectpath,"Get","SubState",&result);
|
|
if(ret == 0)
|
|
{
|
|
checkState(service->name,result,service);
|
|
} else
|
|
{
|
|
fprintf(stderr,"could not call systemdunit\n");
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int checkState(char *ServiceName,char *result,Service *service)
|
|
{
|
|
int i=0;
|
|
char *error_strings[]={"dead","inactive","exited"};
|
|
printf("\n\nresult systemdUnitCall: %s\n",result);
|
|
for(i=0;i<(sizeof(error_strings)/sizeof(error_strings[0]));i++)
|
|
{
|
|
//printf("%s %ld\n",error_strings[i],sizeof(error_strings)/sizeof(error_strings[0]));
|
|
if(strcmp(result,error_strings[i])==0 && service->alerted == false)
|
|
{
|
|
service->alerted=true;
|
|
gotify_message(service->host,ServiceName,"is not running");
|
|
printf("!!system unit died!! Alert gotify server!\n");
|
|
}
|
|
else if(strcmp(result,"active")==0 || strcmp(result,"running")==0)
|
|
{
|
|
if(service->alerted == true)
|
|
{
|
|
gotify_message(service->host,ServiceName,"Running again, after crash");
|
|
}
|
|
service->alerted=false;
|
|
printf("!!service seems okay!!\n");
|
|
return 0;
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
int gotify_message(char *url,char *title,char *message)
|
|
{
|
|
int ret;
|
|
CURL *curl;
|
|
CURLcode res,res2;
|
|
char post_buffer[128];
|
|
char url_buffer[128];
|
|
long response_code;
|
|
|
|
curl = curl_easy_init();
|
|
|
|
const char* token = NULL;
|
|
if(NULL==(token = getenv("GOTIFY_TOKEN")))
|
|
{
|
|
fprintf(stderr,"Pleas set a token using GOTIFY_TOKEN=\"abcde123\"\n");
|
|
}
|
|
else
|
|
{
|
|
printf("Got token from env\n");
|
|
ret = snprintf(url_buffer,sizeof(url_buffer),"%s/message?token=%s",url,token);
|
|
if(ret < sizeof(url_buffer))
|
|
{
|
|
printf("url:%s",url);
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr,"error connecting strings for url and token\n");
|
|
}
|
|
}
|
|
|
|
|
|
ret = snprintf(post_buffer,sizeof(post_buffer),"title=%s&message=%s&priority=6",title,message);
|
|
if(ret < sizeof(post_buffer))
|
|
{
|
|
printf("%s %d\n",post_buffer,ret);
|
|
|
|
if(curl)
|
|
{
|
|
curl_easy_setopt(curl,CURLOPT_URL,url_buffer);
|
|
curl_easy_setopt(curl,CURLOPT_POSTFIELDS,post_buffer);
|
|
curl_easy_setopt(curl,CURLOPT_FOLLOWLOCATION,1L);
|
|
|
|
res = curl_easy_perform(curl);
|
|
if(res != CURLE_OK)
|
|
{
|
|
fprintf(stderr,"curl_easy_perform failed: %s\n",curl_easy_strerror(res));
|
|
}
|
|
res2 = curl_easy_getinfo(curl,CURLINFO_RESPONSE_CODE,&response_code);
|
|
if((res2 == CURLE_OK) && response_code)
|
|
{
|
|
printf("Server returned:%ld\n",response_code);
|
|
}else
|
|
{
|
|
fprintf(stderr,"Error getting info; %ld\n",(long int)res2);
|
|
}
|
|
|
|
curl_easy_cleanup(curl);
|
|
}else
|
|
{
|
|
fprintf(stderr,"Error initializing curl context\n");
|
|
}
|
|
}else
|
|
{
|
|
printf("Error on string concelblalbla: %d\n",ret);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int systemdCall(DBusConnection *connection,char *method,char *param)
|
|
{
|
|
|
|
DBusError error;
|
|
DBusMessage *msg;
|
|
DBusMessage *ret_msg;
|
|
DBusMessageIter iter;
|
|
DBusMessageIter array_iter;
|
|
|
|
char *reply_msg=NULL;
|
|
int ret1,ret2;
|
|
|
|
dbus_error_init(&error);
|
|
|
|
msg = dbus_message_new_method_call("org.freedesktop.systemd1","/org/freedesktop/systemd1","org.freedesktop.systemd1.Manager",method);
|
|
if(NULL == msg)
|
|
{
|
|
fprintf(stderr,"Message null, no message created");
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
printf("message created!\n");
|
|
}
|
|
|
|
|
|
if(param != NULL)
|
|
{
|
|
dbus_message_iter_init_append(msg,&iter);
|
|
if(!dbus_message_iter_append_basic(&iter,DBUS_TYPE_STRING,¶m))
|
|
{
|
|
fprintf(stderr,"Error appending");
|
|
return -1;
|
|
}
|
|
else{
|
|
printf("message appended\n");
|
|
}
|
|
}
|
|
|
|
ret_msg = dbus_connection_send_with_reply_and_block(connection,msg,DBUS_TIMEOUT_INFINITE,&error);
|
|
if(ret_msg == NULL)
|
|
{
|
|
fprintf(stderr,"error on receiving message from peer\n");
|
|
}
|
|
else
|
|
{
|
|
DBusMessageIter reply_iter;
|
|
dbus_message_iter_init(ret_msg,&reply_iter);
|
|
ret1=dbus_message_iter_get_arg_type(&reply_iter);
|
|
|
|
while(ret1 != DBUS_TYPE_INVALID)
|
|
{
|
|
if(ret1 == *DBUS_TYPE_OBJECT_PATH_AS_STRING)
|
|
{
|
|
printf("succesfull received message\narg type %c\n",ret1);
|
|
dbus_message_iter_get_basic(&reply_iter,&reply_msg);
|
|
printf("returned value: %s\n",reply_msg);
|
|
}
|
|
if(ret1 == DBUS_TYPE_STRING)
|
|
{
|
|
printf("succesfull received message\narg type %c\n",ret1);
|
|
dbus_message_iter_get_basic(&reply_iter,&reply_msg);
|
|
printf("Received message:%s\n",reply_msg);
|
|
}
|
|
if(ret1 == DBUS_TYPE_ARRAY)
|
|
{
|
|
int element_count = dbus_message_iter_get_element_count(&reply_iter);
|
|
printf("Array:Elem count %d\n",element_count);
|
|
//ret2 = dbus_message_iter_get_element_type(&reply_iter);
|
|
dbus_message_iter_recurse(&reply_iter,&array_iter);
|
|
int i2=0;
|
|
ret2 = dbus_message_iter_get_arg_type(&array_iter);
|
|
while(ret2 != DBUS_TYPE_INVALID)
|
|
{
|
|
|
|
//printf("type:%c\n",ret2);
|
|
|
|
//dbus_message_iter_get_basic(&array_iter,&reply_msg);
|
|
//printf("Received message:%s\n",reply_msg);
|
|
ret2 = dbus_message_iter_get_arg_type(&array_iter);
|
|
|
|
DBusMessageIter sub_iter;
|
|
int ret3;
|
|
dbus_message_iter_recurse(&array_iter,&sub_iter);
|
|
ret3 = dbus_message_iter_get_arg_type(&sub_iter);
|
|
while(ret3 != DBUS_TYPE_INVALID)
|
|
{
|
|
ret3 = dbus_message_iter_get_arg_type(&sub_iter);
|
|
//printf("arr_sub_iter:%c\n",ret3);
|
|
if(ret3 == DBUS_TYPE_STRING)
|
|
{
|
|
dbus_message_iter_get_basic(&sub_iter,&reply_msg);
|
|
printf("%s\n",reply_msg);
|
|
}
|
|
ret3=dbus_message_iter_next(&sub_iter);
|
|
}
|
|
|
|
|
|
ret2=dbus_message_iter_next(&array_iter);
|
|
i2++;
|
|
//printf("i2:%d\n",i2);
|
|
}
|
|
}
|
|
//printf("iter:%c\n",ret1);
|
|
ret1=dbus_message_iter_next(&reply_iter);
|
|
|
|
}
|
|
|
|
}
|
|
if(dbus_error_is_set(&error))
|
|
{
|
|
fprintf(stderr,"%s\n",error.message);
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int systemdUnitCall(DBusConnection *connection,char *target,char *method,char *param,char **result)
|
|
{
|
|
|
|
DBusError error;
|
|
DBusMessage *msg;
|
|
DBusMessage *ret_msg;
|
|
DBusMessageIter iter;
|
|
DBusMessageIter array_iter;
|
|
DBusMessageIter variant_iter;
|
|
DBusMessageIter sub_iter;
|
|
int ret3;
|
|
char *reply_msg=NULL;
|
|
int ret1,ret2;
|
|
char *param2 = "org.freedesktop.systemd1.Unit";
|
|
|
|
dbus_error_init(&error);
|
|
//dest(servcice/bus) ,path(object) ,iface method
|
|
msg = dbus_message_new_method_call("org.freedesktop.systemd1",target,"org.freedesktop.DBus.Properties",method);
|
|
if(NULL == msg)
|
|
{
|
|
fprintf(stderr,"Message null, no message created\n");
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
printf("message created!\n");
|
|
}
|
|
|
|
if(param2 != NULL)
|
|
{
|
|
dbus_message_iter_init_append(msg,&iter);
|
|
if(!dbus_message_iter_append_basic(&iter,DBUS_TYPE_STRING,¶m2))
|
|
{
|
|
fprintf(stderr,"Error appending\n");
|
|
return -1;
|
|
}
|
|
else{
|
|
printf("message appended\n");
|
|
}
|
|
if(!dbus_message_iter_append_basic(&iter,DBUS_TYPE_STRING,¶m))
|
|
{
|
|
fprintf(stderr,"Error appending\n");
|
|
return -1;
|
|
}
|
|
else{
|
|
printf("message appended\n");
|
|
}
|
|
}
|
|
|
|
ret_msg = dbus_connection_send_with_reply_and_block(connection,msg,DBUS_TIMEOUT_INFINITE,&error);
|
|
if(ret_msg == NULL)
|
|
{
|
|
fprintf(stderr,"error ond receiving message from peer\n");
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
DBusMessageIter reply_iter;
|
|
dbus_message_iter_init(ret_msg,&reply_iter);
|
|
ret1=dbus_message_iter_get_arg_type(&reply_iter);
|
|
printf("ret1 type:%c\n",ret1);
|
|
while(ret1 != DBUS_TYPE_INVALID)
|
|
{
|
|
if(ret1 == DBUS_TYPE_OBJECT_PATH)
|
|
{
|
|
printf("succesfull received message\narg type %c\n",ret1);
|
|
dbus_message_iter_get_basic(&reply_iter,&reply_msg);
|
|
printf("returned value: %s\n",reply_msg);
|
|
}
|
|
if(ret1 == DBUS_TYPE_STRING)
|
|
{
|
|
printf("succesfull received message\narg type %c\n",ret1);
|
|
printf("[func]reply_msg:%p , &reply_msg:%p , *reply_msg:%d\n", reply_msg,&reply_msg,*reply_msg);
|
|
dbus_message_iter_get_basic(&reply_iter,&reply_msg);
|
|
printf("Received message:%s\n",reply_msg);
|
|
}
|
|
if(ret1 == DBUS_TYPE_VARIANT)
|
|
{
|
|
dbus_message_iter_recurse(&reply_iter,&variant_iter);
|
|
char *type = dbus_message_iter_get_signature(&variant_iter);
|
|
printf("str_val:%s\n",type);
|
|
if(type[0] == DBUS_TYPE_STRING)
|
|
{
|
|
printf("[func]&result:%p , result:%p",&result,result);
|
|
dbus_message_iter_get_basic(&variant_iter,result);
|
|
printf("\n--\n*result:%s",*result);
|
|
}
|
|
}
|
|
if(ret1 == DBUS_TYPE_ARRAY)
|
|
{
|
|
int element_count = dbus_message_iter_get_element_count(&reply_iter);
|
|
printf("Array:Elem count %d\n",element_count);
|
|
//ret2 = dbus_message_iter_get_element_type(&reply_iter);
|
|
dbus_message_iter_recurse(&reply_iter,&array_iter);
|
|
ret2 = dbus_message_iter_get_arg_type(&array_iter);
|
|
while(ret2 != DBUS_TYPE_INVALID)
|
|
{
|
|
ret2 = dbus_message_iter_get_arg_type(&array_iter);
|
|
dbus_message_iter_recurse(&array_iter,&sub_iter);
|
|
ret3 = dbus_message_iter_get_arg_type(&sub_iter);
|
|
while(ret3 != DBUS_TYPE_INVALID)
|
|
{
|
|
ret3 = dbus_message_iter_get_arg_type(&sub_iter);
|
|
if(ret3 == DBUS_TYPE_STRING)
|
|
{
|
|
dbus_message_iter_get_basic(&sub_iter,&reply_msg);
|
|
printf("%s\n",reply_msg);
|
|
}
|
|
ret3=dbus_message_iter_next(&sub_iter);
|
|
}
|
|
ret2=dbus_message_iter_next(&array_iter);
|
|
}
|
|
}
|
|
ret1=dbus_message_iter_next(&reply_iter);
|
|
}
|
|
}
|
|
if(dbus_error_is_set(&error))
|
|
{
|
|
fprintf(stderr,"%s\n",error.message);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|