From: fredrik@DOLDA2000.COM Date: Fri, 4 Mar 2005 02:27:32 +0000 (+0000) Subject: Incremental commit. X-Git-Url: http://git.dolda2000.com/gitweb/?a=commitdiff_plain;h=e73ae1346f7b795d6f89a193dbb5d4b5fca9b39a;p=utils.git Incremental commit. git-svn-id: svn+ssh://svn.dolda2000.com/srv/svn/repos/src/utils@168 959494ce-11ee-0310-bf91-de5d638817bd --- diff --git a/krb5-agent.c b/krb5-agent.c index 6cbb783..7edd369 100644 --- a/krb5-agent.c +++ b/krb5-agent.c @@ -2,8 +2,150 @@ #include #include #include +#include +#include +#include +#include + +#define STD_INTERVAL (3600 * 8) + +volatile int died = 0; +pid_t child; +int execmode = 0, failsafe = 0; +int verbose = 0, quiet = 0; +krb5_context context = NULL; +krb5_ccache ccache = NULL; + +void cleanup_krb5(void) +{ + if(ccache != NULL) + krb5_cc_close(context, ccache); + if(context != NULL) + krb5_free_context(context); +} + +void sighandler(int sig) +{ + switch(sig) { + case SIGCHLD: + died = 1; + break; + } +} + +void renew(void) +{ + int ret; + + if(ccache == NULL) + return; +} int main(int argc, char **argv) { - return(0); + char *p; + int c; + time_t interval, last, now; + pid_t wpid; + int ret, status; + + interval = STD_INTERVAL; + while((c = getopt(argc, argv, "+hi:vqf")) != -1) { + switch(c) { + case 'v': + verbose++; + break; + case 'q': + quiet = 1; + break; + case 'f': + failsafe = 1; + break; + case 'i': + p = optarg + strlen(optarg) - 1; + if((*p >= 'a') || (*p <= 'z')) { + if(*p == 'm') + interval = 60; + else if(*p == 'h') + interval = 3600; + else if(*p == 'd') + interval = 86400; + else + interval = 1; + *p = 0; + } else { + interval = 1; + } + interval *= atoi(optarg); + break; + case 'h': + case '?': + case ':': + default: + fprintf(stderr, "usage: krb5-agent [-hvqf] [-i interval] [program args...]\n"); + exit((c == 'h')?0:1); + } + } + + atexit(cleanup_krb5); + if((ret = krb5_init_context(&context)) != 0) { + if(!quiet) + fprintf(stderr, "could not initialize Kerberos context: %s\n", error_message(ret)); + if(!failsafe) + exit(1); + } + if(context != NULL) { + if((ret = krb5_cc_default(context, &ccache)) != 0) { + if(!quiet) + fprintf(stderr, "could not initialize Kerberos context: %s\n", error_message(ret)); + if(!failsafe) + exit(1); + } + } + + if(optind < argc) { + execmode = 1; + signal(SIGCHLD, sighandler); + if((child = fork()) < 0) { + perror("fork"); + exit(1); + } + if(child == 0) { + char buf[80]; + snprintf(buf, 80, "KRB5_AGENT_PID=%i", getpid()); + putenv(buf); + execvp(argv[optind], argv + optind); + perror(argv[optind]); + exit(255); + } + } + now = last = time(NULL); + while(1) { + if(died) { + wpid = waitpid(-1, &status, WNOHANG); + if(wpid < 0) { + perror("waitpid"); + } else if(execmode && (wpid == child)) { + /* Try to preserve exit status as best as we can... */ + if(WIFEXITED(status)) { + exit(WEXITSTATUS(status)); + } else { + signal(WTERMSIG(status), SIG_DFL); + kill(getpid(), WTERMSIG(status)); + exit(255); + } + } + died = 0; + } + sleep((last + interval) - now); + now = time(NULL); + if(now >= last + interval) + renew(); + } } + +/* + * Local Variables: + * compile-command: "gcc -Wall -g -o krb5-agent krb5-agent.c -lkrb5" + * End: + */