Mercurial > hg > Applications > virsh-wrapper
view ie-virsh.c @ 26:76bd473f3b78
change how to define and be used name
author | taiki <taiki@cr.ie.u-ryukyu.ac.jp> |
---|---|
date | Wed, 13 Aug 2014 15:01:32 -1000 |
parents | 66a88f51993f |
children | b731ce75cd35 |
line wrap: on
line source
#include <stdlib.h> #include <unistd.h> #include <stdio.h> #include <sys/types.h> #include <unistd.h> #include <signal.h> #include <string.h> #include <sys/types.h> #include <regex.h> /* Creative Commons Attribution-Noncommercial-Share Alike 3.0 United States License */ /* #define command "/usr/bin/virsh" */ #define command "/usr/local/bin/virsh" /* #define list_command "/usr/bin/virsh list --all" */ #define list_command "/usr/local/bin/virsh list --all" #define start_command "start" #define stop_command "destroy" #define define_command "define" #define undefine_command "undefine" #define dumpxml_command "dumpxml" #define console_command "console" const char *guests[] = {"mata"}; const char *managers[] = {"taiki"}; const char students_sym[] = "students"; const char managers_sym[] = "managers"; const char guests_sym[] = "guests"; const char delimiter[] = "_"; static char bad_name[] = "Bad vmname. Try students/e11/e115711/01 or teachers/kono/02\n"; enum { NAME_LENGTH = 50, VM_NAME_LENGTH = 50 }; enum { STUDENTS, GUESTS, MANAGERS }; #define VMNAME_MAX (512) typedef struct vmlist { char name[VMNAME_MAX]; struct vmlist *next; } VMLIST, *VMLISTPTR; #define NEW(type) ((type*)malloc(sizeof(type))) /* Define global variables */ VMLISTPTR get_vmlist(regex_t *list_pattern) { VMLISTPTR list = NEW(VMLIST); VMLISTPTR p = list ; p->name[0] = 0; p->next = 0; FILE *fp = popen(list_command,"r"); while(fgets(p->name,VMNAME_MAX,fp)!=NULL) { if (regexec(list_pattern, p->name, (size_t) 0, NULL, 0)) continue; p->next = NEW(VMLIST); p = p->next; } p->name[0] = 0; pclose(fp); return list; } void print_vmlist(VMLISTPTR list) { for(;list && list->name[0]; list=list->next) { fprintf(stdout, " %s\n",list->name); } } int check_vmlist_name(VMLISTPTR list, char *arg) { for(;list && list->name[0]; list=list->next) { if (strstr(list->name,arg)!=0) return 1; } return 0; } int check_name(const char *p) { if (!p) return 1; for(;*p;p++) { char c = *p; if (c<=' ') return 1; if (('a'<=c && c<='z') || ('0'<=c && c<='9') || ('/'==c ) || ('-'==c )) continue; return 1; printf("%c", c); } return 0; } void usage() { printf("Usage: COMMAND [define|list|start|destroy|xmldump|undefine|console] [vm-name]\n"); printf(" vm-name should be students/e11/e115711/01 or teachers/kono/02\n"); } int check_user_name(const char *account_name) { const char *regex = "[ek]([0-9]{6})"; regex_t *pattern = (regex_t *)malloc(sizeof(regex_t*)); int ret = 1; if (regcomp(pattern, regex, REG_EXTENDED|REG_NEWLINE) != 0) { exit(0); } ret = regexec(pattern, account_name, (size_t) 0, NULL, 0); regfree(pattern); if (!ret){ return STUDENTS; } const int managers_num = sizeof(managers) / sizeof(managers[0]); int i = 0; for (; i < managers_num; i++) { if (strncmp(account_name, managers[i], NAME_LENGTH) == 0) { return MANAGERS; } } const int guests_num = sizeof(guests) / sizeof(guests[0]); int j = 0; for (; j < guests_num; j++) { if (strncmp(account_name, guests[j], NAME_LENGTH) == 0) { return GUESTS; } } return -1; } void bind_name(char *name, const char *first, const char *second) { strncat(name, first, VM_NAME_LENGTH); strncat(name, delimiter, VM_NAME_LENGTH); strncat(name, second, VM_NAME_LENGTH); strncat(name, delimiter, VM_NAME_LENGTH); return; } void make_vm_name(char *vm_name, const int account_type, const char *account_name, const char *vm_num) { switch(account_type) { case STUDENTS: strncat(vm_name, students_sym, VM_NAME_LENGTH); strncat(vm_name, delimiter, VM_NAME_LENGTH); strncat(vm_name, account_name, 3); strncat(vm_name, delimiter, VM_NAME_LENGTH); strncat(vm_name, account_name, VM_NAME_LENGTH); strncat(vm_name, delimiter, VM_NAME_LENGTH); break; case GUESTS: bind_name(vm_name, guests_sym, account_name); break; case MANAGERS: bind_name(vm_name, managers_sym, account_name); break; default : fprintf(stderr, "Error: no registered type name."); return; } strncat(vm_name, vm_num, VM_NAME_LENGTH); printf("%s\n", vm_name); } /* main(int argc, char **argv) - main process loop */ int main(int argc, char **argv) { int gid; int uid; /* Set euid and egid to actual user */ char *name = getlogin(); uid = getuid(); gid = getgid(); printf("uid %d gid %d name %s\n", uid,gid,name); setegid(getgid()); seteuid(getuid()); regex_t *pattern = NEW(regex_t); if (regcomp(pattern, name, 0) != 0) { exit(0); } int account_type = check_user_name(name); char *vm_name = (char *)malloc(sizeof(char) * VM_NAME_LENGTH); make_vm_name(vm_name, account_type, name, argv[1]); /* Confirm user is in GROUP(999) group */ /* if ( gid != 999 ) { printf("User Not Authorized! Exiting...\n"); exit(1); } */ /* Set uid, gid, euid and egid to root */ setegid(0); seteuid(0); setgid(0); setuid(0); if (argc>=3) { if ( strncmp(argv[1], "define", 6) == 0 ) { if (regexec(pattern, argv[2], (size_t) 0, NULL, 0)) { fprintf(stderr, bad_name); exit(0); } if (check_name(argv[2])) { fprintf(stderr, bad_name); exit(0); } char exec[1024]; // strncpy(exec, "/usr/local/bin/newvm.py -c /etc/libvirt/qemu/fedora16.xml -n ",900); strncpy(exec, "/usr/local/bin/newvm.py -c /etc/libvirt/qemu/fedora19.xml -n ",900); strncat(exec, argv[2],1000); fprintf(stdout, "excuting %s\n",exec ); system(exec); } } VMLISTPTR vmlist = get_vmlist(pattern); char name_xml[1024]; name_xml[0] = 0; if (argc>=3) { if ( strncmp(argv[1], "define", 6) == 0 ) { strncpy(name_xml,argv[2],900); strncat(name_xml,".xml",1000); } else if (check_vmlist_name(vmlist, argv[2])==0) { fprintf(stderr, bad_name); print_vmlist(vmlist); exit(0); } } else if (argc<2) { print_vmlist(vmlist); usage(); exit(0); } /* Check argv for proper arguments and run * the corresponding script, if invoked. */ if (argv[1]==0 || strncmp(argv[1], "list", 4) == 0 ) { print_vmlist(vmlist); } else if (strncmp(argv[1], start_command, 5) == 0) { if (execl(command, command, start_command, argv[2], NULL) < 0) { perror("Execl:"); } } else if ( strncmp(argv[1], stop_command, 4) == 0 ) { if (execl(command, command, stop_command, argv[2], NULL) < 0) { perror("Execl:"); } } else if ( strncmp(argv[1], dumpxml_command, 7) == 0 ) { if (execl(command, command, dumpxml_command, argv[2], NULL) < 0) { perror("Execl:"); } } else if ( strncmp(argv[1], console_command, 8) == 0 ) { if (execl(command, command, console_command, argv[2], NULL) < 0) { perror("Execl:"); } } else if (strncmp(argv[1], define_command, 6) == 0 ) { chdir("/usr/local/etc/libvirt/qemu"); if (execl(command, command, define_command, name_xml, NULL) < 0) { perror("Execl:"); } } else if ( strncmp(argv[1], undefine_command, 8) == 0 ) { chdir("/usr/local/etc/libvirt/qemu"); if (execl(command, command, undefine_command, argv[2], NULL) < 0) { perror("Execl:"); } } else { usage(); exit(1); } exit(0); } /* end */