mirror of https://github.com/BOINC/boinc.git
some utils
git-svn-id: svn+ssh://cvs.lpds.sztaki.hu/var/lib/svn/szdg/dcapi/trunk@1014 a7169a2c-3604-0410-bc95-c702d8d87f7a
This commit is contained in:
drotos
Adam Visegradi
parent
16ed6e17e8
commit
103c7c549b
|
|
@ -0,0 +1,500 @@
|
|||
/*
|
||||
* condor/condor_utils.c
|
||||
*
|
||||
* DC-API usefull functions
|
||||
*
|
||||
* (c) Daniel Drotos, 2006
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <sys/stat.h>
|
||||
#include <dirent.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
#include <stdlib.h>
|
||||
#include <ctype.h>
|
||||
|
||||
#include "dc_internal.h"
|
||||
#include "dc_common.h"
|
||||
|
||||
#include "local_utils.h"
|
||||
#include "local_common.h"
|
||||
|
||||
|
||||
void
|
||||
_DC_init_utils(void)
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
int
|
||||
_DC_mkdir_with_parents(char *dn, mode_t mode)
|
||||
{
|
||||
char *p= dn;
|
||||
int res;
|
||||
|
||||
/*fprintf(stderr, "making %s (EEXIST=%d)\n", dn, EEXIST);*/
|
||||
if (!dn ||
|
||||
!*dn)
|
||||
return(0);
|
||||
while (*p)
|
||||
{
|
||||
while (*p != '/' &&
|
||||
*p != '\0')
|
||||
p++;
|
||||
if (*p == '/')
|
||||
{
|
||||
*p= '\0';
|
||||
if (*dn)
|
||||
{
|
||||
/*fprintf(stderr, "mkdir(%s,%d)=",dn,mode);*/
|
||||
res= mkdir(dn, mode);
|
||||
/*fprintf(stderr, "%d %d\n", res, errno);*/
|
||||
}
|
||||
*p= '/';
|
||||
p++;
|
||||
}
|
||||
}
|
||||
/*fprintf(stderr, "mkdir(%s,%d)=",dn,mode);*/
|
||||
res= mkdir(dn, mode);
|
||||
/*fprintf(stderr, "%d %d\n", res, errno);*/
|
||||
if (res != 0 &&
|
||||
errno == EEXIST)
|
||||
return(DC_OK);
|
||||
return(res?DC_ERR_SYSTEM:DC_OK);
|
||||
}
|
||||
|
||||
|
||||
static int
|
||||
_DC_rmsubdir(char *name)
|
||||
{
|
||||
DIR *d;
|
||||
struct dirent *de;
|
||||
int i= 0;
|
||||
|
||||
if ((d= opendir(name)) == NULL)
|
||||
{
|
||||
DC_log(LOG_DEBUG, "Open %s: %s", name, strerror(errno));
|
||||
return(0);
|
||||
}
|
||||
while ((de= readdir(d)))
|
||||
{
|
||||
if (strcmp(de->d_name, ".") != 0 &&
|
||||
strcmp(de->d_name, "..") != 0)
|
||||
{
|
||||
char *s= malloc(strlen(name)+strlen(de->d_name) + 10);
|
||||
strcpy(s, name);
|
||||
strcat(s, "/");
|
||||
strcat(s, de->d_name);
|
||||
i+= _DC_rm(s);
|
||||
free(s);
|
||||
}
|
||||
}
|
||||
closedir(d);
|
||||
if (rmdir(name))
|
||||
{
|
||||
DC_log(LOG_ERR, "Failed to rmdir %s: %s",
|
||||
name, strerror(errno));
|
||||
}
|
||||
return(i);
|
||||
}
|
||||
|
||||
int
|
||||
_DC_rm(char *name)
|
||||
{
|
||||
struct stat s;
|
||||
int i;
|
||||
|
||||
if (!name ||
|
||||
!*name)
|
||||
return(0);
|
||||
i= lstat(name, &s);
|
||||
if (i != 0)
|
||||
{
|
||||
return(0);
|
||||
}
|
||||
if (S_ISDIR(s.st_mode) &&
|
||||
strcmp(name, ".") != 0 &&
|
||||
strcmp(name, "..") != 0)
|
||||
{
|
||||
return(_DC_rmsubdir(name));
|
||||
}
|
||||
else
|
||||
{
|
||||
if (remove(name))
|
||||
DC_log(LOG_ERR, "Failed to remove %s: %s",
|
||||
name, strerror(errno));
|
||||
}
|
||||
return(1);
|
||||
}
|
||||
|
||||
|
||||
int
|
||||
_DC_file_exists(char *fn)
|
||||
{
|
||||
return(access(fn, F_OK) == 0);
|
||||
}
|
||||
|
||||
|
||||
int
|
||||
_DC_file_empty(char *fn)
|
||||
{
|
||||
struct stat s;
|
||||
|
||||
if (stat(fn, &s))
|
||||
return(1);
|
||||
return(s.st_size == 0);
|
||||
}
|
||||
|
||||
|
||||
int
|
||||
_DC_create_file(char *fn, char *what)
|
||||
{
|
||||
FILE *f= fopen(fn, "w");
|
||||
if (f)
|
||||
{
|
||||
if (what)
|
||||
fprintf(f, "%s", what);
|
||||
fclose(f);
|
||||
return(DC_OK);
|
||||
}
|
||||
else
|
||||
return(DC_ERR_SYSTEM);
|
||||
}
|
||||
|
||||
|
||||
char *
|
||||
_DC_get_file(char *fn)
|
||||
{
|
||||
FILE *f;
|
||||
char *buf= NULL;
|
||||
|
||||
if ((f= fopen(fn, "r")) != NULL)
|
||||
{
|
||||
int bs= 100, i;
|
||||
char c;
|
||||
|
||||
buf= malloc(bs);
|
||||
i= 0;
|
||||
buf[i]= '\0';
|
||||
while ((c= fgetc(f)) != EOF)
|
||||
{
|
||||
if (i > bs-2)
|
||||
{
|
||||
bs+= 100;
|
||||
buf= realloc(buf, bs);
|
||||
}
|
||||
buf[i]= c;
|
||||
i++;
|
||||
buf[i]= '\0';
|
||||
}
|
||||
fclose(f);
|
||||
}
|
||||
return(buf);
|
||||
}
|
||||
|
||||
|
||||
static int _DC_message_id= 0;
|
||||
|
||||
int
|
||||
_DC_create_message(char *box,
|
||||
char *name,
|
||||
const char *message,
|
||||
char *msgfile)
|
||||
{
|
||||
char *fn, *fn2;
|
||||
FILE *f;
|
||||
int ret= DC_OK;
|
||||
|
||||
/*DC_log(LOG_DEBUG, "DC_sendMessage(%s)", message);*/
|
||||
if (!box ||
|
||||
!name)
|
||||
return(DC_ERR_BADPARAM);
|
||||
mkdir(box, S_IRWXU|S_IRGRP|S_IXGRP|S_IROTH|S_IXOTH);
|
||||
_DC_message_id++;
|
||||
fn= malloc(strlen(box)+strlen(name)+100);
|
||||
sprintf(fn, "%s/%s_creating.%d", box, name, _DC_message_id);
|
||||
if ((f= fopen(fn, "w")) != NULL)
|
||||
{
|
||||
if (message)
|
||||
fprintf(f, "%s", message);
|
||||
else if (msgfile)
|
||||
{
|
||||
if ((ret= _DC_copyFile(msgfile, fn)) != 0)
|
||||
DC_log(LOG_ERR, "copyFile(%s,%s) "
|
||||
"error: %s", msgfile, fn,
|
||||
strerror(errno));
|
||||
}
|
||||
else
|
||||
{
|
||||
DC_log(LOG_WARNING, "No message specified for "
|
||||
"creation");
|
||||
}
|
||||
if (fsync(fileno(f)) != 0)
|
||||
DC_log(LOG_ERR, "Sync of %s: %d %s", fn,
|
||||
errno, strerror(errno));
|
||||
fclose(f);
|
||||
/*DC_log(LOG_DEBUG, "Message %d created", _DC_message_id);*/
|
||||
}
|
||||
else
|
||||
{
|
||||
DC_log(LOG_ERR, "Error creating message file (%s)",
|
||||
fn);
|
||||
ret= DC_ERR_SYSTEM;
|
||||
}
|
||||
fn2= malloc(strlen(box)+strlen(name)+100);
|
||||
sprintf(fn2, "%s/%s.%d", box, name, _DC_message_id);
|
||||
rename(fn, fn2);
|
||||
DC_log(LOG_DEBUG, "Message created: %s", fn2);
|
||||
free(fn2);
|
||||
free(fn);
|
||||
return(ret);
|
||||
}
|
||||
|
||||
|
||||
int
|
||||
_DC_nuof_messages(char *box, char *name)
|
||||
{
|
||||
DIR *d;
|
||||
struct dirent *de;
|
||||
int nuof= 0;
|
||||
char *s;
|
||||
|
||||
if (!box)
|
||||
return(0);
|
||||
if ((d= opendir(box)) == NULL)
|
||||
return(0);
|
||||
s= malloc(100 + (name?strlen(name):strlen(_DCAPI_MSG_MESSAGE)));
|
||||
strcpy(s, name?name:_DCAPI_MSG_MESSAGE);
|
||||
strcat(s, ".");
|
||||
while ((de= readdir(d)) != NULL)
|
||||
{
|
||||
char *found= strstr(de->d_name, s);
|
||||
if (found == de->d_name)
|
||||
nuof++;
|
||||
}
|
||||
if (d)
|
||||
closedir(d);
|
||||
free(s);
|
||||
return(nuof);
|
||||
}
|
||||
|
||||
|
||||
char *
|
||||
_DC_message_name(char *box, char *name)
|
||||
{
|
||||
char *dn, *n;
|
||||
DIR *d;
|
||||
struct dirent *de;
|
||||
int min_id;
|
||||
|
||||
if (!box ||
|
||||
!name)
|
||||
return(NULL);
|
||||
if (!*box ||
|
||||
!*name)
|
||||
return(NULL);
|
||||
/*DC_log(LOG_DEBUG, "_DC_message_name(%s,%s)", box, name);*/
|
||||
dn= strdup(box);
|
||||
n= malloc(strlen(name)+10);
|
||||
strcpy(n, name);
|
||||
strcat(n, ".");
|
||||
/*DC_log(LOG_DEBUG, "opendir(%s)=", dn);*/
|
||||
d= opendir(dn);
|
||||
/*DC_log(LOG_DEBUG, "%p", d);*/
|
||||
min_id= -1;
|
||||
while (d &&
|
||||
(de= readdir(d)) != NULL)
|
||||
{
|
||||
char *found= strstr(de->d_name, n);
|
||||
/*DC_log(LOG_DEBUG, "%s <-> %s", de->d_name, n);*/
|
||||
if (found == de->d_name)
|
||||
{
|
||||
char *pos= strrchr(de->d_name, '.');
|
||||
/*DC_log(LOG_DEBUG, "found");*/
|
||||
if (pos)
|
||||
{
|
||||
int id= 0;
|
||||
pos++;
|
||||
id= strtol(pos, NULL, 10);
|
||||
/*DC_log(LOG_DEBUG, "id=%d", id);*/
|
||||
if (id > 0)
|
||||
{
|
||||
if (min_id < 0)
|
||||
min_id= id;
|
||||
else
|
||||
if (id < min_id)
|
||||
min_id= id;
|
||||
}
|
||||
}
|
||||
}
|
||||
/*else
|
||||
DC_log(LOG_DEBUG, "not found");*/
|
||||
}
|
||||
if (d)
|
||||
closedir(d);
|
||||
if (min_id >= 0)
|
||||
{
|
||||
dn= realloc(dn, strlen(box)+strlen(name)+100);
|
||||
sprintf(dn, "%s/%s.%d", box, name, min_id);
|
||||
free(n);
|
||||
/*DC_log(LOG_DEBUG, "return: %s", dn);*/
|
||||
return(dn);
|
||||
}
|
||||
if (dn)
|
||||
free(dn);
|
||||
if (n)
|
||||
free(n);
|
||||
/*DC_log(LOG_DEBUG, "return null");*/
|
||||
return(NULL);
|
||||
}
|
||||
|
||||
|
||||
char *
|
||||
_DC_read_message(char *box, char *name, int del_msg)
|
||||
{
|
||||
FILE *f;
|
||||
char *fn= _DC_message_name(box, name);
|
||||
char *buf= NULL;
|
||||
|
||||
if (fn==NULL)
|
||||
return(NULL);
|
||||
/*DC_log(LOG_DEBUG, "Reading message from %s", fn);*/
|
||||
if ((f= fopen(fn, "r")) != NULL)
|
||||
{
|
||||
int bs= 100, i;
|
||||
char c;
|
||||
|
||||
buf= malloc(bs);
|
||||
i= 0;
|
||||
buf[i]= '\0';
|
||||
while ((c= fgetc(f)) != EOF)
|
||||
{
|
||||
if (i > bs-2)
|
||||
{
|
||||
bs+= 100;
|
||||
buf= realloc(buf, bs);
|
||||
}
|
||||
buf[i]= c;
|
||||
i++;
|
||||
buf[i]= '\0';
|
||||
}
|
||||
fclose(f);
|
||||
if (del_msg)
|
||||
unlink(fn);
|
||||
}
|
||||
if (fn!=NULL)
|
||||
free(fn);
|
||||
return(buf);
|
||||
}
|
||||
|
||||
|
||||
static struct {
|
||||
DC_WUState state;
|
||||
char *name;
|
||||
}
|
||||
_DC_state_names[]= {
|
||||
{ DC_WU_READY, "READY" },
|
||||
{ DC_WU_RUNNING, "RUNNING" },
|
||||
{ DC_WU_FINISHED, "FINISHED" },
|
||||
{ DC_WU_SUSPENDED, "SUSPENDED" },
|
||||
{ DC_WU_ABORTED, "ABORTED" },
|
||||
{ DC_WU_UNKNOWN, "UNKNOWN" },
|
||||
{ 0, NULL }
|
||||
};
|
||||
|
||||
char *
|
||||
_DC_state_name(DC_WUState state)
|
||||
{
|
||||
int i;
|
||||
for (i= 0; _DC_state_names[i].name; i++)
|
||||
if (_DC_state_names[i].state == state)
|
||||
return(_DC_state_names[i].name);
|
||||
return("(unknown)");
|
||||
}
|
||||
|
||||
|
||||
char *
|
||||
_DC_quote_string(char *str)
|
||||
{
|
||||
int i, n;
|
||||
char *s, *p;
|
||||
|
||||
if (!str)
|
||||
{
|
||||
s= strdup("\\nnn");
|
||||
return(s);
|
||||
}
|
||||
if (!*str)
|
||||
{
|
||||
s= strdup("\\000");
|
||||
return(s);
|
||||
}
|
||||
|
||||
for (i= n= 0; str[i]; i++)
|
||||
{
|
||||
if (!isalnum(str[i]))
|
||||
n++;
|
||||
}
|
||||
|
||||
s= calloc(strlen(str)+10*n+10, 1);
|
||||
for (i= 0, p= s; str[i]; i++)
|
||||
{
|
||||
if (isalnum(str[i]))
|
||||
{
|
||||
*p= str[i];
|
||||
p++;
|
||||
}
|
||||
else
|
||||
{
|
||||
sprintf(p, "\\%03o", str[i]);
|
||||
p+= strlen(p);
|
||||
}
|
||||
}
|
||||
return(s);
|
||||
}
|
||||
|
||||
|
||||
char *
|
||||
_DC_unquote_string(char *str)
|
||||
{
|
||||
char *s, *p, *i;
|
||||
|
||||
if (!str)
|
||||
return(NULL);
|
||||
s= calloc(strlen(str)+1, 1);
|
||||
p= str;
|
||||
i= s;
|
||||
while (*p != '\0')
|
||||
{
|
||||
if (*p != '\\')
|
||||
*i= *p;
|
||||
else
|
||||
{
|
||||
char c, *x= p+4;
|
||||
int l;
|
||||
c= *x;
|
||||
*x= '\0';
|
||||
if (strcmp(p+1, "nnn") == 0)
|
||||
{
|
||||
free(s);
|
||||
return(NULL);
|
||||
}
|
||||
l= strtol(p+1, 0, 8);
|
||||
*i= (char)l;
|
||||
*x= c;
|
||||
p= x-1;
|
||||
}
|
||||
p++;
|
||||
i++;
|
||||
}
|
||||
return(s);
|
||||
}
|
||||
|
||||
|
||||
/* End of condor_utils.c */
|
||||
|
||||
/* Local variables: */
|
||||
/* c-file-style: "linux" */
|
||||
/* End: */
|
||||
|
|
@ -0,0 +1,55 @@
|
|||
/*
|
||||
* condor/condor_utils.h
|
||||
*
|
||||
* DC-API usefull functions
|
||||
*
|
||||
* (c) Daniel Drotos, 2006
|
||||
*/
|
||||
|
||||
#ifndef __DC_API_CONDOR_UTILS_H_
|
||||
#define __DC_API_CONDOR_UTILS_H_
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
#include <fcntl.h>
|
||||
|
||||
#include "local_common.h"
|
||||
|
||||
|
||||
extern void _DC_init_utils(void);
|
||||
|
||||
/* Usefull funcs */
|
||||
extern int _DC_mkdir_with_parents(char *dn, mode_t mode);
|
||||
extern int _DC_rm(char *name);
|
||||
extern int _DC_file_exists(char *fn);
|
||||
extern int _DC_file_empty(char *fn);
|
||||
extern int _DC_create_file(char *fn, char *what);
|
||||
extern char *_DC_get_file(char *fn);
|
||||
|
||||
/* Message passing utilities */
|
||||
extern int _DC_create_message(char *box,
|
||||
char *name,
|
||||
const char *message,
|
||||
char *msgfile);
|
||||
extern int _DC_nuof_messages(char *box, char *name);
|
||||
extern char *_DC_message_name(char *box, char *name);
|
||||
extern char *_DC_read_message(char *box, char *name, int del_msg);
|
||||
|
||||
extern char *_DC_state_name(DC_WUState state);
|
||||
extern char *_DC_quote_string(char *str);
|
||||
extern char *_DC_unquote_string(char *str);
|
||||
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
||||
|
||||
/* End of condor/condor_utils.h */
|
||||
|
||||
/* Local variables: */
|
||||
/* c-file-style: "linux" */
|
||||
/* End: */
|
||||
Loading…
Reference in New Issue