Updates for diskwatcher

This commit is contained in:
Michiel Broek 2004-03-21 12:55:45 +00:00
parent f55ff6280f
commit f10a39a9e7
17 changed files with 67 additions and 82 deletions

View File

@ -471,20 +471,30 @@ unsigned long sequencer()
*/ */
int enoughspace(unsigned long needed) int enoughspace(unsigned long needed)
{ {
char *buf, *res; char *buf, *res;
int rc = 3; int rc = 3, cnt;
unsigned long avail = 0L;
buf = calloc(SS_BUFSIZE, sizeof(char)); buf = calloc(SS_BUFSIZE, sizeof(char));
sprintf(buf, "DSPC:1,%ld;", needed); sprintf(buf, "DSPC:1,%ld;", needed);
if (socket_send(buf) == 0) { if (socket_send(buf) == 0) {
free(buf); sprintf(buf, "%s", socket_receive());
buf = socket_receive(); res = strtok(buf, ":");
res = strtok(buf, ","); cnt = atoi(strtok(NULL, ","));
res = strtok(NULL, ";"); if (cnt == 1) {
rc = atoi(res); rc = atoi(strtok(NULL, ";"));
} else if (cnt == 2) {
rc = atoi(strtok(NULL, ","));
avail = atol(strtok(NULL, ";"));
if (rc == 0)
Syslog('+', "Only %ld MBytes diskspace, need %ld MBytes", avail, needed);
} else {
Syslog('-', "Error in enoughspace()");
}
} }
free(buf);
return rc; return rc;
} }

View File

@ -346,10 +346,8 @@ int main(int argc, char *argv[])
/* /*
* Don't do outbound calls if low diskspace * Don't do outbound calls if low diskspace
*/ */
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
WriteError("Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (addr == NULL) { if (addr == NULL) {
WriteError("Calling mbcico without node address not supported anymore"); WriteError("Calling mbcico without node address not supported anymore");

View File

@ -143,10 +143,8 @@ int main(int argc, char **argv)
if (!do_quiet) if (!do_quiet)
printf("\n"); printf("\n");
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (lockprogram((char *)"mbaff")) { if (lockprogram((char *)"mbaff")) {
if (!do_quiet) if (!do_quiet)

View File

@ -167,10 +167,8 @@ int main(int argc, char **argv)
printf("\n"); printf("\n");
} }
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
/* /*
* Extract work directory from the first commandline parameter * Extract work directory from the first commandline parameter

View File

@ -424,10 +424,8 @@ int main(int argc, char **argv)
/* /*
* Not yet locked, if anything goes wrong, exit with die(MBERR_NO_PROGLOCK) * Not yet locked, if anything goes wrong, exit with die(MBERR_NO_PROGLOCK)
*/ */
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (do_mail) { if (do_mail) {
/* /*

View File

@ -234,10 +234,8 @@ int main(int argc, char **argv)
if (!do_quiet) if (!do_quiet)
printf("\n"); printf("\n");
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (lockprogram((char *)"mbfile")) { if (lockprogram((char *)"mbfile")) {
if (!do_quiet) if (!do_quiet)

View File

@ -173,10 +173,8 @@ void ImportFiles(int Area)
/* /*
* Check diskspace * Check diskspace
*/ */
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
Files++; Files++;
memset(&f_db, 0, sizeof(f_db)); memset(&f_db, 0, sizeof(f_db));

View File

@ -358,10 +358,8 @@ void ReqIndex(void)
if (area.Available) { if (area.Available) {
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (!do_quiet) { if (!do_quiet) {
printf("\r%4ld => %-44s \b\b\b\b", i, area.Name); printf("\r%4ld => %-44s \b\b\b\b", i, area.Name);
@ -599,10 +597,8 @@ void HtmlIndex(char *Lang)
if (area.Available) { if (area.Available) {
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (!do_quiet) { if (!do_quiet) {
printf("\r%4ld => %-44s \b\b\b\b", i, area.Name); printf("\r%4ld => %-44s \b\b\b\b", i, area.Name);

View File

@ -86,10 +86,8 @@ void Kill(void)
if ((area.Available) && (area.DLdays || area.FDdays) && (!area.CDrom)) { if ((area.Available) && (area.DLdays || area.FDdays) && (!area.CDrom)) {
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (!do_quiet) { if (!do_quiet) {
printf("\r%4d => %-44s \b\b\b\b", i, area.Name); printf("\r%4d => %-44s \b\b\b\b", i, area.Name);

View File

@ -79,10 +79,8 @@ void PackFileBase(void)
if (area.Available && !area.CDrom) { if (area.Available && !area.CDrom) {
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (!do_quiet) { if (!do_quiet) {
printf("\r%4d => %-44s", i, area.Name); printf("\r%4d => %-44s", i, area.Name);

View File

@ -158,10 +158,8 @@ void SortFileBase(int Area)
if (area.Available) { if (area.Available) {
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (!do_quiet) { if (!do_quiet) {
printf("Sorting area %d: %-44s", Area, area.Name); printf("Sorting area %d: %-44s", Area, area.Name);

View File

@ -177,10 +177,8 @@ int main(int argc,char *argv[])
Syslog(' ', cmd); Syslog(' ', cmd);
free(cmd); free(cmd);
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (lockprogram((char *)"mbindex")) { if (lockprogram((char *)"mbindex")) {
if (!do_quiet) if (!do_quiet)

View File

@ -277,10 +277,8 @@ void DoMsgBase()
fread(&msgs, msgshdr.recsize, 1, pAreas); fread(&msgs, msgshdr.recsize, 1, pAreas);
if (msgs.Active) { if (msgs.Active) {
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
if (!do_quiet) { if (!do_quiet) {
colour(3, 0); colour(3, 0);
@ -308,10 +306,8 @@ void DoMsgBase()
arearec++; arearec++;
if (msgs.Active) { if (msgs.Active) {
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
die(MBERR_DISK_FULL); die(MBERR_DISK_FULL);
}
Nopper(); Nopper();
if (!do_quiet) { if (!do_quiet) {

View File

@ -268,7 +268,7 @@ void system_stat(void)
mvprintw(12,62, "Load avg"); mvprintw(12,62, "Load avg");
hor_lin(13,30,8); hor_lin(13,30,8);
hor_lin(13,45,8); hor_lin(13,45,8);
mvprintw(13,62, "Disk stat"); mvprintw(13,62, "Diskspace");
mvprintw(14, 6, "Client connects"); mvprintw(14, 6, "Client connects");
mvprintw(15, 6, "Peak connections"); mvprintw(15, 6, "Peak connections");
mvprintw(16, 6, "Protocol syntax errors"); mvprintw(16, 6, "Protocol syntax errors");

View File

@ -100,10 +100,8 @@ int main(int argc, char **argv)
Syslog(' ', cmd); Syslog(' ', cmd);
free(cmd); free(cmd);
if (enoughspace(CFG.freespace) == 0) { if (enoughspace(CFG.freespace) == 0)
Syslog('+', "Low diskspace, abort");
ExitClient(MBERR_DISK_FULL); ExitClient(MBERR_DISK_FULL);
}
if (lockprogram((char *)"mbuser")) { if (lockprogram((char *)"mbuser")) {
if (!do_quiet) if (!do_quiet)

View File

@ -378,8 +378,8 @@ char *exe_cmd(char *in)
/* /*
* DSPC:1,n; Check free space in MBytes * DSPC:1,n; Check free space in MBytes
* 100:1,0; No * 100:2,0,n; No, n = lowest size in MBytes
* 100:1,1; Yes * 100:2,1,n; Yes, n = lowest size in MBytes
* 100:1,2; Unknown * 100:1,2; Unknown
* 100:1,3; Error * 100:1,3; Error
*/ */

View File

@ -40,6 +40,7 @@ typedef struct _mfs_list {
char *fstype; /* FS type */ char *fstype; /* FS type */
unsigned long size; /* Size in MB */ unsigned long size; /* Size in MB */
unsigned long avail; /* Available in MB */ unsigned long avail; /* Available in MB */
unsigned ro : 1; /* Read-Only fs. */
} mfs_list; } mfs_list;
mfs_list *mfs = NULL; /* List of filesystems */ mfs_list *mfs = NULL; /* List of filesystems */
@ -72,6 +73,7 @@ void tidy_mfslist(mfs_list **fap)
for (tmp = *fap; tmp; tmp = old) { for (tmp = *fap; tmp; tmp = old) {
old = tmp->next; old = tmp->next;
free(tmp->mountpoint); free(tmp->mountpoint);
free(tmp->fstype);
free(tmp); free(tmp);
} }
*fap = NULL; *fap = NULL;
@ -105,6 +107,7 @@ void fill_mfslist(mfs_list **fap, char *mountpoint, char *fstype)
tmp->fstype = xstrcpy(fstype); tmp->fstype = xstrcpy(fstype);
tmp->size = 0L; tmp->size = 0L;
tmp->avail = 0L; tmp->avail = 0L;
tmp->ro = TRUE; // Read-only, statfs will set real value.
*fap = tmp; *fap = tmp;
} }
@ -132,20 +135,12 @@ char *disk_check(char *token)
{ {
static char buf[SS_BUFSIZE]; static char buf[SS_BUFSIZE];
mfs_list *tmp; mfs_list *tmp;
unsigned long needed; unsigned long needed, lowest = 0xffffffff;
int rc; int rc;
strtok(token, ","); strtok(token, ",");
needed = atol(strtok(NULL, ";")); needed = atol(strtok(NULL, ";"));
if (! needed) {
/*
* Answer enough space
*/
sprintf(buf, "100:1,1");
return buf;
}
if (mfs == NULL) { if (mfs == NULL) {
/* /*
* Answer Error * Answer Error
@ -158,24 +153,18 @@ char *disk_check(char *token)
Syslog('!', "disk_check() mutex_lock failed rc=%d", rc); Syslog('!', "disk_check() mutex_lock failed rc=%d", rc);
for (tmp = mfs; tmp; tmp = tmp->next) { for (tmp = mfs; tmp; tmp = tmp->next) {
if (tmp->avail < needed) { if (!tmp->ro && (tmp->avail < lowest))
/* lowest = tmp->avail;
* Answer Not enough space
*/
if ((rc = pthread_mutex_unlock(&a_mutex)))
Syslog('!', "disk_check() mutex_unlock failed rc=%d", rc);
sprintf(buf, "100:1,0");
return buf;
}
} }
if ((rc = pthread_mutex_unlock(&a_mutex))) if ((rc = pthread_mutex_unlock(&a_mutex)))
Syslog('!', "disk_check() mutex_unlock failed rc=%d", rc); Syslog('!', "disk_check() mutex_unlock failed rc=%d", rc);
/* if (lowest < needed) {
* Enough space sprintf(buf, "100:2,0,%ld;", lowest);
*/ } else {
sprintf(buf, "100:1,1"); sprintf(buf, "100:2,1,%ld;", lowest);
}
return buf; return buf;
} }
@ -247,6 +236,16 @@ void update_diskstat(void)
temp = (unsigned long)(sfs.f_bsize / 512L); temp = (unsigned long)(sfs.f_bsize / 512L);
tmp->size = (unsigned long)(sfs.f_blocks * temp) / 2048L; tmp->size = (unsigned long)(sfs.f_blocks * temp) / 2048L;
tmp->avail = (unsigned long)(sfs.f_bavail * temp) / 2048L; tmp->avail = (unsigned long)(sfs.f_bavail * temp) / 2048L;
#if defined(__linux__)
/*
* The struct statfs (or statvfs) seems to have no information
* about read-only filesystems, so we guess it based on fstype.
* See man 2 statvfs about what approximately is defined.
*/
tmp->ro = (strstr(tmp->fstype, "iso") != NULL);
#elif defined(__FreeBSD__) || defined(__NetBSD__)
Syslog('-', "%s %d", tmp->mountpoint, sfs.f_flags);
#endif
} }
} }
@ -283,6 +282,7 @@ void add_path(char *path)
if (S_ISDIR(sb.st_mode)) { if (S_ISDIR(sb.st_mode)) {
#if defined(__linux__) #if defined(__linux__)
if ((fp = fopen((char *)"/etc/mtab", "r"))) { if ((fp = fopen((char *)"/etc/mtab", "r"))) {
mtab = calloc(PATH_MAX, sizeof(char)); mtab = calloc(PATH_MAX, sizeof(char));
fsname = calloc(PATH_MAX, sizeof(char)); fsname = calloc(PATH_MAX, sizeof(char));
@ -297,7 +297,6 @@ void add_path(char *path)
* mounted filesystem that matches must be the one we seek. * mounted filesystem that matches must be the one we seek.
*/ */
if (strncmp(fs, path, strlen(fs)) == 0) { if (strncmp(fs, path, strlen(fs)) == 0) {
Syslog('d', "Found fs %s", fs);
sprintf(fsname, "%s", fs); sprintf(fsname, "%s", fs);
fs = strtok(NULL, " \t"); fs = strtok(NULL, " \t");
sprintf(fstype, "%s", fs); sprintf(fstype, "%s", fs);
@ -318,9 +317,7 @@ void add_path(char *path)
fstype = calloc(PATH_MAX, sizeof(char)); fstype = calloc(PATH_MAX, sizeof(char));
for (i = 0; i < mntsize; i++) { for (i = 0; i < mntsize; i++) {
Syslog('d', "Check fs %s", mntbuf[i].f_mntonname);
if (strncmp(mntbuf[i].f_mntonname, path, strlen(mntbuf[i].f_mntonname)) == 0) { if (strncmp(mntbuf[i].f_mntonname, path, strlen(mntbuf[i].f_mntonname)) == 0) {
Syslog('d', "Found fs %s", mntbuf[i].f_mntonname);
sprintf(fsname, "%s", mntbuf[i].f_mntonname); sprintf(fsname, "%s", mntbuf[i].f_mntonname);
sprintf(fstype, "%s", mntbuf[i].f_fstypename); sprintf(fstype, "%s", mntbuf[i].f_fstypename);
} }
@ -331,10 +328,18 @@ void add_path(char *path)
free(fstype); free(fstype);
} }
#else
#error "Unknow OS - don't know what to do"
#endif #endif
} else { } else {
Syslog('d', "**** Is not a dir"); /*
* Not a directory, maybe a file.
*/
if ((p = strrchr(path, '/')))
*p = '\0';
Syslog('d', "Recursive add name %s", path);
add_path(path);
} }
} else { } else {
/* /*
@ -513,7 +518,7 @@ void *disk_thread(void)
Syslog('d', "All directories added"); Syslog('d', "All directories added");
for (tmp = mfs; tmp; tmp = tmp->next) { for (tmp = mfs; tmp; tmp = tmp->next) {
Syslog('+', "%s %s %ld %ld", tmp->mountpoint, tmp->fstype, tmp->size, tmp->avail); Syslog('+', "Found filesystem: %s type: %s", tmp->mountpoint, tmp->fstype);
} }
if ((rc = pthread_mutex_unlock(&a_mutex))) if ((rc = pthread_mutex_unlock(&a_mutex)))