main.c deleted because it was not written in the same spirit
KaneRoot
parent
f6254d0cf9
commit
2fe12ac7dc
312
main.c
312
main.c
|
|
@ -1,312 +0,0 @@
|
|||
#include <stdio.h>
|
||||
#include <mysql/mysql.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <stdarg.h>
|
||||
#include <time.h>
|
||||
#include <unistd.h>
|
||||
#include <signal.h>
|
||||
#define HOST "10.8.0.10"
|
||||
#define USER "vpsm"
|
||||
#define PASS "bla"
|
||||
#define BASE "vpsm"
|
||||
|
||||
int my_atoi (char * str )
|
||||
{
|
||||
int res=atoi(str);
|
||||
free(str);
|
||||
return res;
|
||||
}
|
||||
|
||||
char **my_explode(char *str, char separator)
|
||||
{
|
||||
char **res = NULL;
|
||||
int nbstr = 1;
|
||||
int len;
|
||||
int from = 0;
|
||||
int i;
|
||||
int j;
|
||||
|
||||
res = (char **) malloc(sizeof (char *));
|
||||
len = strlen(str);
|
||||
for (i = 0; i <= len; ++i)
|
||||
{
|
||||
if ((i == len) || (str[i] == separator))
|
||||
{
|
||||
res = (char **) realloc(res, ++nbstr * sizeof (char *));
|
||||
res[nbstr - 2] = (char *) malloc((i - from + 1) * sizeof (char));
|
||||
for (j = 0; j < (i - from); ++j)
|
||||
res[nbstr - 2][j] = str[j + from];
|
||||
res[nbstr - 2][i - from] = '\0';
|
||||
from = i + 1;
|
||||
++i;
|
||||
}
|
||||
}
|
||||
res[nbstr - 1] = NULL;
|
||||
return res;
|
||||
}
|
||||
/**
|
||||
* The va_list method for str() above
|
||||
**/
|
||||
|
||||
char *vstr(char *format, va_list args)
|
||||
{
|
||||
char *buffer = NULL;
|
||||
int va_size = 256;
|
||||
/* Parse the format and generate a proper string output */
|
||||
do
|
||||
{
|
||||
va_size *= 2;
|
||||
buffer = (char*)realloc(buffer, va_size);
|
||||
memset(buffer, 0, va_size);
|
||||
} while (vsnprintf(buffer, va_size, format, args) < 0);
|
||||
|
||||
/* Trim any white space at the end */
|
||||
char *output = (char*)malloc(strlen(buffer) + 1);
|
||||
strcpy(output, buffer);
|
||||
free(buffer);
|
||||
return output;
|
||||
}
|
||||
/**
|
||||
* Method to produce a malloc()'d char*, similar to printf()
|
||||
**/
|
||||
char *str(char *format, ...)
|
||||
{
|
||||
va_list args;
|
||||
va_start(args,format);
|
||||
char *ret = vstr(format, args);
|
||||
va_end(args);
|
||||
return ret;
|
||||
}
|
||||
// Fonction pour logger un message dans le fichier de logs
|
||||
void log_message(char * message)
|
||||
{
|
||||
FILE *file;
|
||||
char infotime[80];
|
||||
time_t rawtime;
|
||||
struct tm * timeinfo;
|
||||
time ( &rawtime );
|
||||
timeinfo = localtime ( &rawtime );
|
||||
strftime (infotime,80,"[%d/%m/%y - %X] ",timeinfo);
|
||||
if((file = fopen("log.txt","a"))) // Ouverture du fichier en append
|
||||
{
|
||||
message=str("%s%s\n",infotime,message);
|
||||
fwrite(message,sizeof(char),strlen(message),file);
|
||||
}
|
||||
free(message);
|
||||
fclose(file);
|
||||
}
|
||||
int do_query(char * query)
|
||||
{
|
||||
int res=-1;
|
||||
MYSQL mysql;
|
||||
mysql_init(&mysql);
|
||||
mysql_options(&mysql,MYSQL_READ_DEFAULT_GROUP,"option");
|
||||
if(mysql_real_connect(&mysql,HOST,USER,PASS,BASE,0,NULL,0))
|
||||
{
|
||||
if(mysql_query(&mysql,query))
|
||||
{
|
||||
res=1;
|
||||
}
|
||||
}
|
||||
mysql_close(&mysql);
|
||||
free(query);
|
||||
return res;
|
||||
}
|
||||
void update_status()
|
||||
{
|
||||
|
||||
int vmid=0;
|
||||
int status=0;
|
||||
FILE *process;
|
||||
process = popen("{ vzlist && qm list; } | grep -vi 'status' | tr -s ' ' | cut -d ' ' -f 2,4 | sort -gk1 | sed 's/running/1/' | sed 's/stopped/2/'","r");
|
||||
char data[50];
|
||||
char **line;
|
||||
int vmids[30];
|
||||
int vmstatus[30];
|
||||
int i=0;
|
||||
char* tstatus[3];
|
||||
tstatus[0]="";
|
||||
tstatus[1]="Actif";
|
||||
tstatus[2]="Éteind";
|
||||
while(fgets(data,50,process))
|
||||
{
|
||||
line=my_explode(data,' ');
|
||||
vmid=atoi(line[0]);
|
||||
status=atoi(line[1]);
|
||||
vmids[i]=vmid;
|
||||
vmstatus[i]=status;
|
||||
//printf("%d is %d \n",vmid,status);
|
||||
i++;
|
||||
free(line[0]);
|
||||
free(line[1]);
|
||||
free(line);
|
||||
}
|
||||
fclose(process);
|
||||
int length=i;
|
||||
MYSQL mysql;
|
||||
mysql_init(&mysql);
|
||||
mysql_options(&mysql,MYSQL_READ_DEFAULT_GROUP,"option");
|
||||
mysql_real_connect(&mysql,HOST,USER,PASS,BASE,0,NULL,0);
|
||||
mysql_query(&mysql,"SELECT vmid,status FROM vps ORDER BY vmid");
|
||||
MYSQL_RES *result=NULL;
|
||||
MYSQL_ROW row;
|
||||
result=mysql_use_result(&mysql);
|
||||
i=0;
|
||||
//printf("test database : \n");
|
||||
while((i<length) && (row=mysql_fetch_row(result)))
|
||||
{
|
||||
unsigned long *lengths=mysql_fetch_lengths(result);
|
||||
vmid=my_atoi(str("%.*s", (int) lengths[0], row[0]));
|
||||
status=my_atoi(str("%.*s", (int) lengths[1], row[1]));
|
||||
//printf("i = %d, vmid=%d, status=%d, dvmid=%d, vstatus=%d\n",i,vmid,status,vmids[i],vmstatus[i]);
|
||||
while(i<length && vmids[i]<vmid)
|
||||
{
|
||||
//log_message(str("VPS %d non trouvé dans la base, passage au VPS %d",vmids[i],vmids[i+1]));
|
||||
i++;
|
||||
}
|
||||
if(vmids[i]==vmid)
|
||||
{
|
||||
if(vmstatus[i]!=status)
|
||||
{
|
||||
log_message(str("Changement d'état du VPS %d à %s",vmid,tstatus[vmstatus[i]]));
|
||||
do_query(str("UPDATE vps SET status=%d WHERE vmid=%d",vmstatus[i],vmid));
|
||||
//printf("%d status updated to %d",vmid,vmstatus[i]);
|
||||
}
|
||||
i++;
|
||||
}
|
||||
else if (status==1)
|
||||
{
|
||||
log_message(str("Statut du VPS %d non récupéré, passage a État %s",vmid,tstatus[2]));
|
||||
do_query(str("UPDATE vps SET status=2 WHERE vmid=%d",vmid));
|
||||
//printf("vps %d not found, setting to 2",vmid);
|
||||
}
|
||||
|
||||
}
|
||||
mysql_free_result(result);
|
||||
mysql_close(&mysql);
|
||||
}
|
||||
int exec_process(char * command)
|
||||
{
|
||||
FILE * process = popen(str("%s &>/dev/stdout",command),"r");
|
||||
char data[100];
|
||||
char **line;
|
||||
while(fgets(data,100,process))
|
||||
{
|
||||
line=my_explode(data,'\n');
|
||||
log_message(str("%s: %s",command, line[0]));
|
||||
free(line[0]);
|
||||
free(line);
|
||||
}
|
||||
int res=pclose(process);
|
||||
return res;
|
||||
}
|
||||
void exec_task(char * command,int id)
|
||||
{
|
||||
do_query(str("UPDATE tasks SET status=3 WHERE id=%d",id));
|
||||
log_message(str("[Tâche %d] Exécution de : %s",id,command));
|
||||
int res=exec_process(command);
|
||||
if(res) // Si commande échouée
|
||||
{
|
||||
log_message(str("[Tâche %d] Une Erreur est survenue",id));
|
||||
do_query(str("UPDATE tasks SET status=2 WHERE id=%d",id));
|
||||
}
|
||||
else
|
||||
{
|
||||
//exec_process("php update.php");
|
||||
update_status();
|
||||
log_message(str("[Tâche %d] Exécution Terminée",id));
|
||||
do_query(str("UPDATE tasks SET status=1 WHERE id=%d",id));
|
||||
}
|
||||
free(command);
|
||||
|
||||
}
|
||||
|
||||
int continuer=1;
|
||||
|
||||
void sighandler(int signum)
|
||||
{
|
||||
continuer=0;
|
||||
log_message(str("Signal %d attrapé, Fermeture...",signum));
|
||||
exit(0);
|
||||
}
|
||||
|
||||
int main()
|
||||
{
|
||||
|
||||
struct sigaction act;
|
||||
|
||||
act.sa_handler = sighandler;
|
||||
sigfillset(&act.sa_mask);
|
||||
act.sa_flags = 0;
|
||||
sigaction(15, &act, NULL);
|
||||
sigaction(11, &act, NULL);
|
||||
|
||||
char* vctype[3]; // Type du contrôlleur de VPS
|
||||
vctype[0]=""; // Rien
|
||||
vctype[1]="vzctl"; // OpenVZ
|
||||
vctype[2]="qm"; // KVM
|
||||
char* ttype[4]; // Type de la commande (tâche)
|
||||
ttype[0]=""; //Rien
|
||||
ttype[1]="start"; // Démarrage de la VM
|
||||
ttype[2]="stop"; // Éteindre la VM
|
||||
ttype[3]="shutdown"; // Éteindre la VM par ACPI
|
||||
MYSQL mysql;
|
||||
mysql_init(&mysql);
|
||||
mysql_options(&mysql,MYSQL_READ_DEFAULT_GROUP,"option");
|
||||
int id=0;
|
||||
int vid=0;
|
||||
int type=0;
|
||||
int status=0;
|
||||
int vtype=0;
|
||||
unsigned long *lengths;
|
||||
if(mysql_real_connect(&mysql,HOST,USER,PASS,BASE,0,NULL,0))
|
||||
{
|
||||
log_message("Connexion réussie");
|
||||
// Création du daemon
|
||||
int p=fork();
|
||||
if(p!=0) // Si c'est le parent
|
||||
{
|
||||
// On détache...
|
||||
printf("VPSMd: Démarrage et détachement de la console...\n");
|
||||
exit(0);
|
||||
}
|
||||
else
|
||||
{
|
||||
// On commence le daemon
|
||||
log_message("Daemon démarré");
|
||||
while(continuer)
|
||||
{
|
||||
// exec_process("php update.php");
|
||||
update_status();
|
||||
// Rercherche des tâches à effectuer (status 0 : à traiter)
|
||||
mysql_query(&mysql,"SELECT t.id,t.vid,t.type,t.status,v.type AS vtype FROM tasks t JOIN vps v ON t.vid=v.vmid WHERE t.status=0");
|
||||
MYSQL_RES *result=NULL;
|
||||
MYSQL_ROW row;
|
||||
|
||||
result=mysql_use_result(&mysql);
|
||||
while((row=mysql_fetch_row(result)))
|
||||
{
|
||||
lengths=mysql_fetch_lengths(result);
|
||||
id=my_atoi(str("%.*s", (int) lengths[0], row[0]));
|
||||
vid=my_atoi(str("%.*s", (int) lengths[1], row[1]));
|
||||
type=my_atoi(str("%.*s", (int) lengths[2], row[2]));
|
||||
status=my_atoi(str("%.*s", (int) lengths[3], row[3]));
|
||||
vtype=my_atoi(str("%.*s", (int) lengths[4], row[4]));
|
||||
log_message(str("Tâche numéro %d pour le VPS numero %d, de type %d, statut %d",id,vid,type,status));
|
||||
exec_task(str("%s %s %d",vctype[vtype],ttype[type],vid),id);
|
||||
}
|
||||
mysql_free_result(result);
|
||||
sleep(3);
|
||||
}
|
||||
}
|
||||
mysql_close(&mysql);
|
||||
log_message("Connexion fermée");
|
||||
log_message("Fermeture du daemon");
|
||||
}
|
||||
else
|
||||
{
|
||||
log_message("Erreur de connexion ! Fermeture du daemon");
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue