2000-02-19 13:36:48 +01:00
|
|
|
/*
|
|
|
|
* tools.c: Various tools
|
|
|
|
*
|
2000-04-24 09:46:05 +02:00
|
|
|
* See the main source file 'vdr.c' for copyright information and
|
2000-02-19 13:36:48 +01:00
|
|
|
* how to reach the author.
|
|
|
|
*
|
2006-01-08 11:44:37 +01:00
|
|
|
* $Id: tools.c 1.109 2006/01/08 11:40:35 kls Exp $
|
2000-02-19 13:36:48 +01:00
|
|
|
*/
|
|
|
|
|
|
|
|
#include "tools.h"
|
2000-07-23 15:01:31 +02:00
|
|
|
#include <ctype.h>
|
2000-04-15 17:38:11 +02:00
|
|
|
#include <dirent.h>
|
2000-03-11 11:22:37 +01:00
|
|
|
#include <errno.h>
|
2005-12-29 11:24:02 +01:00
|
|
|
extern "C" {
|
|
|
|
#ifdef boolean
|
|
|
|
#define HAVE_BOOLEAN
|
|
|
|
#endif
|
|
|
|
#include <jpeglib.h>
|
|
|
|
#undef boolean
|
|
|
|
}
|
2005-08-27 16:42:28 +02:00
|
|
|
#include <stdarg.h>
|
2001-08-11 11:15:41 +02:00
|
|
|
#include <stdlib.h>
|
2000-02-19 13:36:48 +01:00
|
|
|
#include <sys/time.h>
|
2002-03-31 20:51:06 +02:00
|
|
|
#include <sys/vfs.h>
|
2001-08-26 15:52:17 +02:00
|
|
|
#include <time.h>
|
2000-04-15 17:38:11 +02:00
|
|
|
#include <unistd.h>
|
2005-09-25 13:49:31 +02:00
|
|
|
#include <utime.h>
|
2001-08-25 13:27:26 +02:00
|
|
|
#include "i18n.h"
|
2000-02-19 13:36:48 +01:00
|
|
|
|
2000-04-15 17:38:11 +02:00
|
|
|
int SysLogLevel = 3;
|
|
|
|
|
2004-01-11 15:54:37 +01:00
|
|
|
int BCD2INT(int x)
|
|
|
|
{
|
|
|
|
return ((1000000 * BCDCHARTOINT((x >> 24) & 0xFF)) +
|
|
|
|
(10000 * BCDCHARTOINT((x >> 16) & 0xFF)) +
|
|
|
|
(100 * BCDCHARTOINT((x >> 8) & 0xFF)) +
|
|
|
|
BCDCHARTOINT( x & 0xFF));
|
|
|
|
}
|
|
|
|
|
2001-08-12 15:22:48 +02:00
|
|
|
ssize_t safe_read(int filedes, void *buffer, size_t size)
|
|
|
|
{
|
|
|
|
for (;;) {
|
|
|
|
ssize_t p = read(filedes, buffer, size);
|
|
|
|
if (p < 0 && errno == EINTR) {
|
2002-05-13 16:35:49 +02:00
|
|
|
dsyslog("EINTR while reading from file handle %d - retrying", filedes);
|
2001-08-12 15:22:48 +02:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
return p;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t safe_write(int filedes, const void *buffer, size_t size)
|
|
|
|
{
|
2002-03-23 15:48:08 +01:00
|
|
|
ssize_t p = 0;
|
|
|
|
ssize_t written = size;
|
2002-03-22 15:18:58 +01:00
|
|
|
const unsigned char *ptr = (const unsigned char *)buffer;
|
|
|
|
while (size > 0) {
|
|
|
|
p = write(filedes, ptr, size);
|
|
|
|
if (p < 0) {
|
|
|
|
if (errno == EINTR) {
|
2002-05-13 16:35:49 +02:00
|
|
|
dsyslog("EINTR while writing to file handle %d - retrying", filedes);
|
2002-03-22 15:18:58 +01:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
ptr += p;
|
|
|
|
size -= p;
|
|
|
|
}
|
2002-03-23 15:48:08 +01:00
|
|
|
return p < 0 ? p : written;
|
2001-08-12 15:22:48 +02:00
|
|
|
}
|
|
|
|
|
2000-04-15 17:38:11 +02:00
|
|
|
void writechar(int filedes, char c)
|
|
|
|
{
|
2001-08-12 15:22:48 +02:00
|
|
|
safe_write(filedes, &c, sizeof(c));
|
2000-04-15 17:38:11 +02:00
|
|
|
}
|
|
|
|
|
2005-01-16 12:02:39 +01:00
|
|
|
int WriteAllOrNothing(int fd, const uchar *Data, int Length, int TimeoutMs, int RetryMs)
|
|
|
|
{
|
|
|
|
int written = 0;
|
|
|
|
while (Length > 0) {
|
|
|
|
int w = write(fd, Data + written, Length);
|
|
|
|
if (w > 0) {
|
|
|
|
Length -= w;
|
|
|
|
written += w;
|
|
|
|
}
|
|
|
|
else if (written > 0 && !FATALERRNO) {
|
|
|
|
// we've started writing, so we must finish it!
|
|
|
|
cTimeMs t;
|
|
|
|
cPoller Poller(fd, true);
|
|
|
|
Poller.Poll(RetryMs);
|
|
|
|
if (TimeoutMs > 0 && (TimeoutMs -= t.Elapsed()) <= 0)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
// nothing written yet (or fatal error), so we can just return the error code:
|
|
|
|
return w;
|
|
|
|
}
|
|
|
|
return written;
|
|
|
|
}
|
|
|
|
|
2001-09-22 13:07:43 +02:00
|
|
|
char *strcpyrealloc(char *dest, const char *src)
|
|
|
|
{
|
|
|
|
if (src) {
|
|
|
|
int l = max(dest ? strlen(dest) : 0, strlen(src)) + 1; // don't let the block get smaller!
|
|
|
|
dest = (char *)realloc(dest, l);
|
|
|
|
if (dest)
|
|
|
|
strcpy(dest, src);
|
|
|
|
else
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: out of memory");
|
2001-09-22 13:07:43 +02:00
|
|
|
}
|
2001-10-05 20:40:53 +02:00
|
|
|
else {
|
2002-08-11 13:32:23 +02:00
|
|
|
free(dest);
|
2001-10-05 20:40:53 +02:00
|
|
|
dest = NULL;
|
|
|
|
}
|
2001-09-22 13:07:43 +02:00
|
|
|
return dest;
|
|
|
|
}
|
|
|
|
|
2000-09-09 14:57:43 +02:00
|
|
|
char *strn0cpy(char *dest, const char *src, size_t n)
|
|
|
|
{
|
|
|
|
char *s = dest;
|
|
|
|
for ( ; --n && (*dest = *src) != 0; dest++, src++) ;
|
|
|
|
*dest = 0;
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2000-07-16 15:02:33 +02:00
|
|
|
char *strreplace(char *s, char c1, char c2)
|
|
|
|
{
|
|
|
|
char *p = s;
|
|
|
|
|
2001-02-11 11:29:22 +01:00
|
|
|
while (p && *p) {
|
2000-07-16 15:02:33 +02:00
|
|
|
if (*p == c1)
|
|
|
|
*p = c2;
|
|
|
|
p++;
|
|
|
|
}
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2002-02-03 15:55:04 +01:00
|
|
|
char *strreplace(char *s, const char *s1, const char *s2)
|
|
|
|
{
|
|
|
|
char *p = strstr(s, s1);
|
|
|
|
if (p) {
|
|
|
|
int of = p - s;
|
|
|
|
int l = strlen(s);
|
|
|
|
int l1 = strlen(s1);
|
|
|
|
int l2 = strlen(s2);
|
|
|
|
if (l2 > l1)
|
|
|
|
s = (char *)realloc(s, strlen(s) + l2 - l1 + 1);
|
|
|
|
if (l2 != l1)
|
|
|
|
memmove(s + of + l2, s + of + l1, l - of - l1 + 1);
|
|
|
|
strncpy(s + of, s2, l2);
|
|
|
|
}
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2000-10-29 13:17:22 +01:00
|
|
|
char *skipspace(const char *s)
|
2000-07-23 15:01:31 +02:00
|
|
|
{
|
|
|
|
while (*s && isspace(*s))
|
|
|
|
s++;
|
2000-10-29 13:17:22 +01:00
|
|
|
return (char *)s;
|
|
|
|
}
|
|
|
|
|
2000-11-11 16:38:41 +01:00
|
|
|
char *stripspace(char *s)
|
|
|
|
{
|
|
|
|
if (s && *s) {
|
|
|
|
for (char *p = s + strlen(s) - 1; p >= s; p--) {
|
|
|
|
if (!isspace(*p))
|
|
|
|
break;
|
|
|
|
*p = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2001-08-17 13:19:10 +02:00
|
|
|
char *compactspace(char *s)
|
|
|
|
{
|
|
|
|
if (s && *s) {
|
|
|
|
char *t = stripspace(skipspace(s));
|
|
|
|
char *p = t;
|
|
|
|
while (p && *p) {
|
|
|
|
char *q = skipspace(p);
|
|
|
|
if (q - p > 1)
|
|
|
|
memmove(p + 1, q, strlen(q) + 1);
|
|
|
|
p++;
|
|
|
|
}
|
|
|
|
if (t != s)
|
|
|
|
memmove(s, t, strlen(t) + 1);
|
|
|
|
}
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
cString strescape(const char *s, const char *chars)
|
|
|
|
{
|
|
|
|
char *buffer;
|
|
|
|
const char *p = s;
|
|
|
|
char *t = NULL;
|
|
|
|
while (*p) {
|
|
|
|
if (strchr(chars, *p)) {
|
|
|
|
if (!t) {
|
|
|
|
buffer = MALLOC(char, 2 * strlen(s) + 1);
|
|
|
|
t = buffer + (p - s);
|
|
|
|
s = strcpy(buffer, s);
|
|
|
|
}
|
|
|
|
*t++ = '\\';
|
|
|
|
}
|
|
|
|
if (t)
|
|
|
|
*t++ = *p;
|
|
|
|
p++;
|
|
|
|
}
|
|
|
|
if (t)
|
|
|
|
*t = 0;
|
2005-02-05 10:12:14 +01:00
|
|
|
return cString(s, t != NULL);
|
2004-12-26 12:45:22 +01:00
|
|
|
}
|
|
|
|
|
2001-08-17 13:19:10 +02:00
|
|
|
bool startswith(const char *s, const char *p)
|
|
|
|
{
|
|
|
|
while (*p) {
|
|
|
|
if (*p++ != *s++)
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2002-01-20 16:47:09 +01:00
|
|
|
bool endswith(const char *s, const char *p)
|
|
|
|
{
|
|
|
|
const char *se = s + strlen(s) - 1;
|
|
|
|
const char *pe = p + strlen(p) - 1;
|
|
|
|
while (pe >= p) {
|
|
|
|
if (*pe-- != *se-- || (se < s && pe >= p))
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2000-10-29 13:17:22 +01:00
|
|
|
bool isempty(const char *s)
|
|
|
|
{
|
|
|
|
return !(s && *skipspace(s));
|
2000-07-23 15:01:31 +02:00
|
|
|
}
|
|
|
|
|
2002-10-19 15:33:37 +02:00
|
|
|
int numdigits(int n)
|
|
|
|
{
|
|
|
|
char buf[16];
|
|
|
|
snprintf(buf, sizeof(buf), "%d", n);
|
|
|
|
return strlen(buf);
|
|
|
|
}
|
|
|
|
|
2000-07-23 15:01:31 +02:00
|
|
|
bool isnumber(const char *s)
|
|
|
|
{
|
2002-11-24 15:56:24 +01:00
|
|
|
if (!*s)
|
|
|
|
return false;
|
2000-07-23 15:01:31 +02:00
|
|
|
while (*s) {
|
|
|
|
if (!isdigit(*s))
|
|
|
|
return false;
|
|
|
|
s++;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
cString AddDirectory(const char *DirName, const char *FileName)
|
|
|
|
{
|
|
|
|
char *buf;
|
|
|
|
asprintf(&buf, "%s/%s", DirName && *DirName ? DirName : ".", FileName);
|
2005-02-05 10:12:14 +01:00
|
|
|
return cString(buf, true);
|
2004-12-26 12:45:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
cString itoa(int n)
|
|
|
|
{
|
|
|
|
char buf[16];
|
|
|
|
snprintf(buf, sizeof(buf), "%d", n);
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
2002-01-27 13:11:23 +01:00
|
|
|
int FreeDiskSpaceMB(const char *Directory, int *UsedMB)
|
2000-07-29 15:21:42 +02:00
|
|
|
{
|
2002-01-27 13:11:23 +01:00
|
|
|
if (UsedMB)
|
|
|
|
*UsedMB = 0;
|
|
|
|
int Free = 0;
|
2002-03-31 20:51:06 +02:00
|
|
|
struct statfs statFs;
|
|
|
|
if (statfs(Directory, &statFs) == 0) {
|
2002-05-01 16:22:41 +02:00
|
|
|
double blocksPerMeg = 1024.0 * 1024.0 / statFs.f_bsize;
|
2002-03-31 20:51:06 +02:00
|
|
|
if (UsedMB)
|
2002-05-01 16:22:41 +02:00
|
|
|
*UsedMB = int((statFs.f_blocks - statFs.f_bfree) / blocksPerMeg);
|
|
|
|
Free = int(statFs.f_bavail / blocksPerMeg);
|
2000-07-29 15:21:42 +02:00
|
|
|
}
|
|
|
|
else
|
2002-03-31 20:51:06 +02:00
|
|
|
LOG_ERROR_STR(Directory);
|
2000-07-29 15:21:42 +02:00
|
|
|
return Free;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool DirectoryOk(const char *DirName, bool LogErrors)
|
2000-07-28 13:44:31 +02:00
|
|
|
{
|
|
|
|
struct stat ds;
|
|
|
|
if (stat(DirName, &ds) == 0) {
|
|
|
|
if (S_ISDIR(ds.st_mode)) {
|
|
|
|
if (access(DirName, R_OK | W_OK | X_OK) == 0)
|
|
|
|
return true;
|
2000-07-29 15:21:42 +02:00
|
|
|
else if (LogErrors)
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: can't access %s", DirName);
|
2000-07-28 13:44:31 +02:00
|
|
|
}
|
2000-07-29 15:21:42 +02:00
|
|
|
else if (LogErrors)
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: %s is not a directory", DirName);
|
2000-07-28 13:44:31 +02:00
|
|
|
}
|
2000-07-29 15:21:42 +02:00
|
|
|
else if (LogErrors)
|
2000-07-28 13:44:31 +02:00
|
|
|
LOG_ERROR_STR(DirName);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2000-04-15 17:38:11 +02:00
|
|
|
bool MakeDirs(const char *FileName, bool IsDirectory)
|
2000-03-11 11:22:37 +01:00
|
|
|
{
|
|
|
|
bool result = true;
|
|
|
|
char *s = strdup(FileName);
|
|
|
|
char *p = s;
|
|
|
|
if (*p == '/')
|
|
|
|
p++;
|
2000-04-15 17:38:11 +02:00
|
|
|
while ((p = strchr(p, '/')) != NULL || IsDirectory) {
|
|
|
|
if (p)
|
|
|
|
*p = 0;
|
2000-03-11 11:22:37 +01:00
|
|
|
struct stat fs;
|
|
|
|
if (stat(s, &fs) != 0 || !S_ISDIR(fs.st_mode)) {
|
2002-05-13 16:35:49 +02:00
|
|
|
dsyslog("creating directory %s", s);
|
2005-08-06 09:56:08 +02:00
|
|
|
if (mkdir(s, ACCESSPERMS) == -1) {
|
2000-12-08 16:23:32 +01:00
|
|
|
LOG_ERROR_STR(s);
|
2000-03-11 11:22:37 +01:00
|
|
|
result = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2000-04-15 17:38:11 +02:00
|
|
|
if (p)
|
|
|
|
*p++ = '/';
|
|
|
|
else
|
|
|
|
break;
|
2000-03-11 11:22:37 +01:00
|
|
|
}
|
2002-08-11 13:32:23 +02:00
|
|
|
free(s);
|
2000-03-11 11:22:37 +01:00
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2000-07-29 15:21:42 +02:00
|
|
|
bool RemoveFileOrDir(const char *FileName, bool FollowSymlinks)
|
2000-04-15 17:38:11 +02:00
|
|
|
{
|
|
|
|
struct stat st;
|
|
|
|
if (stat(FileName, &st) == 0) {
|
|
|
|
if (S_ISDIR(st.st_mode)) {
|
2004-12-19 16:33:34 +01:00
|
|
|
cReadDir d(FileName);
|
|
|
|
if (d.Ok()) {
|
2000-04-15 17:38:11 +02:00
|
|
|
struct dirent *e;
|
2004-12-19 16:33:34 +01:00
|
|
|
while ((e = d.Next()) != NULL) {
|
2000-04-15 17:38:11 +02:00
|
|
|
if (strcmp(e->d_name, ".") && strcmp(e->d_name, "..")) {
|
|
|
|
char *buffer;
|
|
|
|
asprintf(&buffer, "%s/%s", FileName, e->d_name);
|
2000-07-29 15:21:42 +02:00
|
|
|
if (FollowSymlinks) {
|
|
|
|
int size = strlen(buffer) * 2; // should be large enough
|
2002-08-11 13:32:23 +02:00
|
|
|
char *l = MALLOC(char, size);
|
2000-07-29 15:21:42 +02:00
|
|
|
int n = readlink(buffer, l, size);
|
|
|
|
if (n < 0) {
|
|
|
|
if (errno != EINVAL)
|
|
|
|
LOG_ERROR_STR(buffer);
|
|
|
|
}
|
|
|
|
else if (n < size) {
|
|
|
|
l[n] = 0;
|
2002-05-13 16:35:49 +02:00
|
|
|
dsyslog("removing %s", l);
|
2000-07-29 15:21:42 +02:00
|
|
|
if (remove(l) < 0)
|
|
|
|
LOG_ERROR_STR(l);
|
|
|
|
}
|
|
|
|
else
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: symlink name length (%d) exceeded anticipated buffer size (%d)", n, size);
|
2002-08-11 13:32:23 +02:00
|
|
|
free(l);
|
2000-07-29 15:21:42 +02:00
|
|
|
}
|
2002-05-13 16:35:49 +02:00
|
|
|
dsyslog("removing %s", buffer);
|
2000-04-15 17:38:11 +02:00
|
|
|
if (remove(buffer) < 0)
|
2000-07-29 15:21:42 +02:00
|
|
|
LOG_ERROR_STR(buffer);
|
2002-08-11 13:32:23 +02:00
|
|
|
free(buffer);
|
2000-04-15 17:38:11 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
2000-07-29 15:21:42 +02:00
|
|
|
LOG_ERROR_STR(FileName);
|
2000-04-15 17:38:11 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
2002-05-13 16:35:49 +02:00
|
|
|
dsyslog("removing %s", FileName);
|
2001-02-04 12:36:32 +01:00
|
|
|
if (remove(FileName) < 0) {
|
|
|
|
LOG_ERROR_STR(FileName);
|
|
|
|
return false;
|
|
|
|
}
|
2000-04-15 17:38:11 +02:00
|
|
|
}
|
2000-12-28 12:57:16 +01:00
|
|
|
else if (errno != ENOENT) {
|
2000-07-29 15:21:42 +02:00
|
|
|
LOG_ERROR_STR(FileName);
|
2000-12-28 12:57:16 +01:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
2000-04-15 17:38:11 +02:00
|
|
|
}
|
|
|
|
|
2001-02-11 14:53:44 +01:00
|
|
|
bool RemoveEmptyDirectories(const char *DirName, bool RemoveThis)
|
|
|
|
{
|
2004-12-19 16:33:34 +01:00
|
|
|
cReadDir d(DirName);
|
|
|
|
if (d.Ok()) {
|
2001-02-11 14:53:44 +01:00
|
|
|
bool empty = true;
|
|
|
|
struct dirent *e;
|
2004-12-19 16:33:34 +01:00
|
|
|
while ((e = d.Next()) != NULL) {
|
2001-02-11 14:53:44 +01:00
|
|
|
if (strcmp(e->d_name, ".") && strcmp(e->d_name, "..") && strcmp(e->d_name, "lost+found")) {
|
|
|
|
char *buffer;
|
|
|
|
asprintf(&buffer, "%s/%s", DirName, e->d_name);
|
|
|
|
struct stat st;
|
|
|
|
if (stat(buffer, &st) == 0) {
|
|
|
|
if (S_ISDIR(st.st_mode)) {
|
|
|
|
if (!RemoveEmptyDirectories(buffer, true))
|
|
|
|
empty = false;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
empty = false;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
LOG_ERROR_STR(buffer);
|
2002-08-11 13:32:23 +02:00
|
|
|
free(buffer);
|
2001-02-11 14:53:44 +01:00
|
|
|
return false;
|
|
|
|
}
|
2002-08-11 13:32:23 +02:00
|
|
|
free(buffer);
|
2001-02-11 14:53:44 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if (RemoveThis && empty) {
|
2002-05-13 16:35:49 +02:00
|
|
|
dsyslog("removing %s", DirName);
|
2001-02-11 14:53:44 +01:00
|
|
|
if (remove(DirName) < 0) {
|
|
|
|
LOG_ERROR_STR(DirName);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return empty;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
LOG_ERROR_STR(DirName);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2005-12-18 10:41:26 +01:00
|
|
|
int DirSizeMB(const char *DirName)
|
|
|
|
{
|
|
|
|
cReadDir d(DirName);
|
|
|
|
if (d.Ok()) {
|
|
|
|
int size = 0;
|
|
|
|
struct dirent *e;
|
|
|
|
while (size >= 0 && (e = d.Next()) != NULL) {
|
|
|
|
if (strcmp(e->d_name, ".") && strcmp(e->d_name, "..")) {
|
|
|
|
char *buffer;
|
|
|
|
asprintf(&buffer, "%s/%s", DirName, e->d_name);
|
|
|
|
struct stat st;
|
|
|
|
if (stat(buffer, &st) == 0) {
|
|
|
|
if (S_ISDIR(st.st_mode)) {
|
|
|
|
int n = DirSizeMB(buffer);
|
|
|
|
if (n >= 0)
|
|
|
|
size += n;
|
|
|
|
else
|
|
|
|
size = -1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
size += st.st_size / MEGABYTE(1);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
LOG_ERROR_STR(buffer);
|
|
|
|
size = -1;
|
|
|
|
}
|
|
|
|
free(buffer);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return size;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
LOG_ERROR_STR(DirName);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2001-01-13 15:36:31 +01:00
|
|
|
char *ReadLink(const char *FileName)
|
|
|
|
{
|
2002-08-04 14:57:29 +02:00
|
|
|
char RealName[PATH_MAX];
|
2001-01-13 15:36:31 +01:00
|
|
|
const char *TargetName = NULL;
|
|
|
|
int n = readlink(FileName, RealName, sizeof(RealName) - 1);
|
|
|
|
if (n < 0) {
|
|
|
|
if (errno == ENOENT || errno == EINVAL) // file doesn't exist or is not a symlink
|
|
|
|
TargetName = FileName;
|
2002-02-02 13:44:24 +01:00
|
|
|
else // some other error occurred
|
2001-01-13 15:36:31 +01:00
|
|
|
LOG_ERROR_STR(FileName);
|
|
|
|
}
|
|
|
|
else if (n < int(sizeof(RealName))) { // got it!
|
|
|
|
RealName[n] = 0;
|
|
|
|
TargetName = RealName;
|
|
|
|
}
|
|
|
|
else
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: symlink's target name too long: %s", FileName);
|
2001-01-13 15:36:31 +01:00
|
|
|
return TargetName ? strdup(TargetName) : NULL;
|
|
|
|
}
|
|
|
|
|
2001-08-11 11:15:41 +02:00
|
|
|
bool SpinUpDisk(const char *FileName)
|
|
|
|
{
|
2004-12-19 16:33:34 +01:00
|
|
|
char *buf = NULL;
|
2001-08-11 11:15:41 +02:00
|
|
|
for (int n = 0; n < 10; n++) {
|
2002-08-11 13:32:23 +02:00
|
|
|
free(buf);
|
2001-08-11 11:15:41 +02:00
|
|
|
if (DirectoryOk(FileName))
|
|
|
|
asprintf(&buf, "%s/vdr-%06d", *FileName ? FileName : ".", n);
|
|
|
|
else
|
|
|
|
asprintf(&buf, "%s.vdr-%06d", FileName, n);
|
|
|
|
if (access(buf, F_OK) != 0) { // the file does not exist
|
|
|
|
timeval tp1, tp2;
|
|
|
|
gettimeofday(&tp1, NULL);
|
2005-08-06 09:56:08 +02:00
|
|
|
int f = open(buf, O_WRONLY | O_CREAT, DEFFILEMODE);
|
2001-08-11 11:15:41 +02:00
|
|
|
// O_SYNC doesn't work on all file systems
|
|
|
|
if (f >= 0) {
|
2005-02-19 13:51:44 +01:00
|
|
|
if (fdatasync(f) < 0)
|
|
|
|
LOG_ERROR_STR(buf);
|
2001-08-11 11:15:41 +02:00
|
|
|
close(f);
|
|
|
|
remove(buf);
|
|
|
|
gettimeofday(&tp2, NULL);
|
|
|
|
double seconds = (((long long)tp2.tv_sec * 1000000 + tp2.tv_usec) - ((long long)tp1.tv_sec * 1000000 + tp1.tv_usec)) / 1000000.0;
|
|
|
|
if (seconds > 0.5)
|
2005-11-04 14:22:04 +01:00
|
|
|
dsyslog("SpinUpDisk took %.2f seconds", seconds);
|
2004-12-19 16:33:34 +01:00
|
|
|
free(buf);
|
2001-08-11 11:15:41 +02:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
LOG_ERROR_STR(buf);
|
|
|
|
}
|
|
|
|
}
|
2004-12-19 16:33:34 +01:00
|
|
|
free(buf);
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: SpinUpDisk failed");
|
2001-08-11 11:15:41 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2005-09-25 13:49:31 +02:00
|
|
|
void TouchFile(const char *FileName)
|
|
|
|
{
|
|
|
|
if (utime(FileName, NULL) == -1 && errno != ENOENT)
|
|
|
|
LOG_ERROR_STR(FileName);
|
|
|
|
}
|
|
|
|
|
2004-06-13 20:26:51 +02:00
|
|
|
time_t LastModifiedTime(const char *FileName)
|
|
|
|
{
|
|
|
|
struct stat fs;
|
|
|
|
if (stat(FileName, &fs) == 0)
|
|
|
|
return fs.st_mtime;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2004-12-19 18:08:09 +01:00
|
|
|
// --- cTimeMs ---------------------------------------------------------------
|
|
|
|
|
|
|
|
cTimeMs::cTimeMs(void)
|
|
|
|
{
|
|
|
|
Set();
|
|
|
|
}
|
|
|
|
|
2005-01-04 11:06:45 +01:00
|
|
|
uint64 cTimeMs::Now(void)
|
|
|
|
{
|
|
|
|
struct timeval t;
|
|
|
|
if (gettimeofday(&t, NULL) == 0)
|
|
|
|
return (uint64(t.tv_sec)) * 1000 + t.tv_usec / 1000;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2004-12-19 18:08:09 +01:00
|
|
|
void cTimeMs::Set(int Ms)
|
|
|
|
{
|
|
|
|
begin = Now() + Ms;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cTimeMs::TimedOut(void)
|
|
|
|
{
|
|
|
|
return Now() >= begin;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64 cTimeMs::Elapsed(void)
|
|
|
|
{
|
|
|
|
return Now() - begin;
|
|
|
|
}
|
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
// --- cString ---------------------------------------------------------------
|
2004-12-19 16:33:34 +01:00
|
|
|
|
2005-02-05 10:12:14 +01:00
|
|
|
cString::cString(const char *S, bool TakePointer)
|
2004-12-19 16:33:34 +01:00
|
|
|
{
|
2005-02-05 10:12:14 +01:00
|
|
|
s = TakePointer ? (char *)S : S ? strdup(S) : NULL;
|
2004-12-19 16:33:34 +01:00
|
|
|
}
|
|
|
|
|
2005-11-26 14:16:02 +01:00
|
|
|
cString::cString(const cString &String)
|
|
|
|
{
|
|
|
|
s = String.s ? strdup(String.s) : NULL;
|
|
|
|
}
|
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
cString::~cString()
|
2004-12-19 16:33:34 +01:00
|
|
|
{
|
2004-12-26 12:45:22 +01:00
|
|
|
free(s);
|
2004-12-19 16:33:34 +01:00
|
|
|
}
|
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
cString &cString::operator=(const cString &String)
|
2004-12-19 16:33:34 +01:00
|
|
|
{
|
2005-11-26 14:16:02 +01:00
|
|
|
if (this == &String)
|
|
|
|
return *this;
|
2005-11-04 14:27:51 +01:00
|
|
|
free(s);
|
2004-12-26 12:45:22 +01:00
|
|
|
s = String.s ? strdup(String.s) : NULL;
|
|
|
|
return *this;
|
2004-12-19 16:33:34 +01:00
|
|
|
}
|
|
|
|
|
2005-08-27 16:42:28 +02:00
|
|
|
cString cString::sprintf(const char *fmt, ...)
|
|
|
|
{
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, fmt);
|
|
|
|
char *buffer;
|
|
|
|
vasprintf(&buffer, fmt, ap);
|
|
|
|
return cString(buffer, true);
|
|
|
|
}
|
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
cString WeekDayName(int WeekDay)
|
2002-02-17 13:05:05 +01:00
|
|
|
{
|
2004-12-26 12:45:22 +01:00
|
|
|
char buffer[4];
|
2002-02-17 13:05:05 +01:00
|
|
|
WeekDay = WeekDay == 0 ? 6 : WeekDay - 1; // we start with monday==0!
|
|
|
|
if (0 <= WeekDay && WeekDay <= 6) {
|
|
|
|
const char *day = tr("MonTueWedThuFriSatSun");
|
|
|
|
day += WeekDay * 3;
|
2004-12-19 16:33:34 +01:00
|
|
|
strn0cpy(buffer, day, sizeof(buffer));
|
2004-12-26 12:45:22 +01:00
|
|
|
return buffer;
|
2002-02-17 13:05:05 +01:00
|
|
|
}
|
|
|
|
else
|
2004-12-26 12:45:22 +01:00
|
|
|
return "???";
|
2002-02-17 13:05:05 +01:00
|
|
|
}
|
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
cString WeekDayName(time_t t)
|
|
|
|
{
|
|
|
|
struct tm tm_r;
|
|
|
|
return WeekDayName(localtime_r(&t, &tm_r)->tm_wday);
|
|
|
|
}
|
2004-05-22 13:23:22 +02:00
|
|
|
|
2004-12-26 12:45:22 +01:00
|
|
|
cString DayDateTime(time_t t)
|
2001-08-25 13:27:26 +02:00
|
|
|
{
|
2004-12-26 12:45:22 +01:00
|
|
|
char buffer[32];
|
2001-08-25 13:27:26 +02:00
|
|
|
if (t == 0)
|
|
|
|
time(&t);
|
2001-10-19 13:22:24 +02:00
|
|
|
struct tm tm_r;
|
|
|
|
tm *tm = localtime_r(&t, &tm_r);
|
2004-12-26 12:45:22 +01:00
|
|
|
snprintf(buffer, sizeof(buffer), "%s %2d.%02d %02d:%02d", *WeekDayName(tm->tm_wday), tm->tm_mday, tm->tm_mon + 1, tm->tm_hour, tm->tm_min);
|
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
|
|
|
|
cString TimeToString(time_t t)
|
|
|
|
{
|
|
|
|
char buffer[32];
|
|
|
|
if (ctime_r(&t, buffer)) {
|
|
|
|
buffer[strlen(buffer) - 1] = 0; // strip trailing newline
|
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
return "???";
|
2004-12-19 16:33:34 +01:00
|
|
|
}
|
|
|
|
|
2005-05-16 14:45:11 +02:00
|
|
|
cString DateString(time_t t)
|
|
|
|
{
|
|
|
|
char buf[32];
|
|
|
|
struct tm tm_r;
|
|
|
|
tm *tm = localtime_r(&t, &tm_r);
|
|
|
|
char *p = stpcpy(buf, WeekDayName(tm->tm_wday));
|
|
|
|
*p++ = ' ';
|
|
|
|
strftime(p, sizeof(buf) - (p - buf), "%d.%m.%Y", tm);
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
|
|
|
cString TimeString(time_t t)
|
|
|
|
{
|
|
|
|
char buf[25];
|
|
|
|
struct tm tm_r;
|
|
|
|
strftime(buf, sizeof(buf), "%R", localtime_r(&t, &tm_r));
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
2005-12-29 11:24:02 +01:00
|
|
|
// --- RgbToJpeg -------------------------------------------------------------
|
|
|
|
|
|
|
|
#define JPEGCOMPRESSMEM 500000
|
|
|
|
|
|
|
|
struct tJpegCompressData {
|
|
|
|
int size;
|
|
|
|
uchar *mem;
|
|
|
|
};
|
|
|
|
|
|
|
|
static void JpegCompressInitDestination(j_compress_ptr cinfo)
|
|
|
|
{
|
|
|
|
tJpegCompressData *jcd = (tJpegCompressData *)cinfo->client_data;
|
|
|
|
if (jcd) {
|
|
|
|
cinfo->dest->free_in_buffer = jcd->size = JPEGCOMPRESSMEM;
|
|
|
|
cinfo->dest->next_output_byte = jcd->mem = MALLOC(uchar, jcd->size);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static boolean JpegCompressEmptyOutputBuffer(j_compress_ptr cinfo)
|
|
|
|
{
|
|
|
|
tJpegCompressData *jcd = (tJpegCompressData *)cinfo->client_data;
|
|
|
|
if (jcd) {
|
|
|
|
int Used = jcd->size;
|
|
|
|
jcd->size += JPEGCOMPRESSMEM;
|
|
|
|
jcd->mem = (uchar *)realloc(jcd->mem, jcd->size);
|
|
|
|
if (jcd->mem) {
|
|
|
|
cinfo->dest->next_output_byte = jcd->mem + Used;
|
|
|
|
cinfo->dest->free_in_buffer = jcd->size - Used;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void JpegCompressTermDestination(j_compress_ptr cinfo)
|
|
|
|
{
|
|
|
|
tJpegCompressData *jcd = (tJpegCompressData *)cinfo->client_data;
|
|
|
|
if (jcd) {
|
|
|
|
int Used = cinfo->dest->next_output_byte - jcd->mem;
|
|
|
|
if (Used < jcd->size) {
|
|
|
|
jcd->size = Used;
|
|
|
|
jcd->mem = (uchar *)realloc(jcd->mem, jcd->size);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
uchar *RgbToJpeg(uchar *Mem, int Width, int Height, int &Size, int Quality)
|
|
|
|
{
|
|
|
|
if (Quality < 0)
|
|
|
|
Quality = 0;
|
|
|
|
else if (Quality > 100)
|
|
|
|
Quality = 100;
|
2006-01-08 11:44:37 +01:00
|
|
|
|
2005-12-29 11:24:02 +01:00
|
|
|
jpeg_destination_mgr jdm;
|
|
|
|
|
|
|
|
jdm.init_destination = JpegCompressInitDestination;
|
|
|
|
jdm.empty_output_buffer = JpegCompressEmptyOutputBuffer;
|
|
|
|
jdm.term_destination = JpegCompressTermDestination;
|
|
|
|
|
|
|
|
struct jpeg_compress_struct cinfo;
|
|
|
|
struct jpeg_error_mgr jerr;
|
|
|
|
cinfo.err = jpeg_std_error(&jerr);
|
|
|
|
jpeg_create_compress(&cinfo);
|
|
|
|
cinfo.dest = &jdm;
|
|
|
|
tJpegCompressData jcd;
|
|
|
|
cinfo.client_data = &jcd;
|
|
|
|
cinfo.image_width = Width;
|
|
|
|
cinfo.image_height = Height;
|
|
|
|
cinfo.input_components = 3;
|
|
|
|
cinfo.in_color_space = JCS_RGB;
|
|
|
|
|
|
|
|
jpeg_set_defaults(&cinfo);
|
|
|
|
jpeg_set_quality(&cinfo, Quality, true);
|
|
|
|
jpeg_start_compress(&cinfo, true);
|
|
|
|
|
|
|
|
int rs = Width * 3;
|
|
|
|
JSAMPROW rp[Height];
|
|
|
|
for (int k = 0; k < Height; k++)
|
|
|
|
rp[k] = &Mem[rs * k];
|
|
|
|
jpeg_write_scanlines(&cinfo, rp, Height);
|
|
|
|
jpeg_finish_compress(&cinfo);
|
|
|
|
jpeg_destroy_compress(&cinfo);
|
|
|
|
|
|
|
|
Size = jcd.size;
|
|
|
|
return jcd.mem;
|
|
|
|
}
|
|
|
|
|
2005-12-29 16:02:37 +01:00
|
|
|
// --- cBase64Encoder --------------------------------------------------------
|
|
|
|
|
|
|
|
const char *cBase64Encoder::b64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
|
|
|
|
|
|
|
cBase64Encoder::cBase64Encoder(const uchar *Data, int Length, int MaxResult)
|
|
|
|
{
|
|
|
|
data = Data;
|
|
|
|
length = Length;
|
|
|
|
maxResult = MaxResult;
|
|
|
|
i = 0;
|
|
|
|
result = MALLOC(char, maxResult + 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
cBase64Encoder::~cBase64Encoder()
|
|
|
|
{
|
|
|
|
free(result);
|
|
|
|
}
|
|
|
|
|
|
|
|
const char *cBase64Encoder::NextLine(void)
|
|
|
|
{
|
|
|
|
int r = 0;
|
|
|
|
while (i < length && r < maxResult - 3) {
|
|
|
|
result[r++] = b64[(data[i] >> 2) & 0x3F];
|
|
|
|
char c = (data[i] << 4) & 0x3F;
|
|
|
|
if (++i < length)
|
|
|
|
c |= (data[i] >> 4) & 0x0F;
|
|
|
|
result[r++] = b64[c];
|
|
|
|
if (i < length) {
|
|
|
|
c = (data[i] << 2) & 0x3F;
|
|
|
|
if (++i < length)
|
|
|
|
c |= (data[i] >> 6) & 0x03;
|
|
|
|
result[r++] = b64[c];
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
i++;
|
|
|
|
result[r++] = '=';
|
|
|
|
}
|
|
|
|
if (i < length) {
|
|
|
|
c = data[i] & 0x3F;
|
|
|
|
result[r++] = b64[c];
|
|
|
|
}
|
|
|
|
else
|
|
|
|
result[r++] = '=';
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
if (r > 0) {
|
|
|
|
result[r] = 0;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2004-12-19 16:33:34 +01:00
|
|
|
// --- cReadLine -------------------------------------------------------------
|
|
|
|
|
2005-11-04 17:18:33 +01:00
|
|
|
cReadLine::cReadLine(void)
|
|
|
|
{
|
|
|
|
size = 0;
|
|
|
|
buffer = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
cReadLine::~cReadLine()
|
|
|
|
{
|
|
|
|
free(buffer);
|
|
|
|
}
|
|
|
|
|
2004-12-19 16:33:34 +01:00
|
|
|
char *cReadLine::Read(FILE *f)
|
|
|
|
{
|
2005-11-04 17:18:33 +01:00
|
|
|
int n = getline(&buffer, &size, f);
|
|
|
|
if (n > 0) {
|
|
|
|
n--;
|
|
|
|
if (buffer[n] == '\n')
|
|
|
|
buffer[n] = 0;
|
2004-12-19 16:33:34 +01:00
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
return NULL;
|
2001-08-25 13:27:26 +02:00
|
|
|
}
|
|
|
|
|
2002-08-16 09:22:29 +02:00
|
|
|
// --- cPoller ---------------------------------------------------------------
|
|
|
|
|
|
|
|
cPoller::cPoller(int FileHandle, bool Out)
|
|
|
|
{
|
|
|
|
numFileHandles = 0;
|
|
|
|
Add(FileHandle, Out);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cPoller::Add(int FileHandle, bool Out)
|
|
|
|
{
|
|
|
|
if (FileHandle >= 0) {
|
|
|
|
for (int i = 0; i < numFileHandles; i++) {
|
2004-11-21 14:37:36 +01:00
|
|
|
if (pfd[i].fd == FileHandle && pfd[i].events == (Out ? POLLOUT : POLLIN))
|
2002-08-16 09:22:29 +02:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
if (numFileHandles < MaxPollFiles) {
|
|
|
|
pfd[numFileHandles].fd = FileHandle;
|
|
|
|
pfd[numFileHandles].events = Out ? POLLOUT : POLLIN;
|
2004-11-21 14:37:36 +01:00
|
|
|
pfd[numFileHandles].revents = 0;
|
2002-08-16 09:22:29 +02:00
|
|
|
numFileHandles++;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
esyslog("ERROR: too many file handles in cPoller");
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cPoller::Poll(int TimeoutMs)
|
|
|
|
{
|
|
|
|
if (numFileHandles) {
|
|
|
|
if (poll(pfd, numFileHandles, TimeoutMs) != 0)
|
|
|
|
return true; // returns true even in case of an error, to let the caller
|
|
|
|
// access the file and thus see the error code
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2004-12-19 16:33:34 +01:00
|
|
|
// --- cReadDir --------------------------------------------------------------
|
|
|
|
|
|
|
|
cReadDir::cReadDir(const char *Directory)
|
|
|
|
{
|
|
|
|
directory = opendir(Directory);
|
|
|
|
}
|
|
|
|
|
|
|
|
cReadDir::~cReadDir()
|
|
|
|
{
|
|
|
|
if (directory)
|
|
|
|
closedir(directory);
|
|
|
|
}
|
|
|
|
|
|
|
|
struct dirent *cReadDir::Next(void)
|
|
|
|
{
|
|
|
|
return directory && readdir_r(directory, &u.d, &result) == 0 ? result : NULL;
|
|
|
|
}
|
|
|
|
|
2000-09-17 08:23:46 +02:00
|
|
|
// --- cFile -----------------------------------------------------------------
|
|
|
|
|
|
|
|
bool cFile::files[FD_SETSIZE] = { false };
|
|
|
|
int cFile::maxFiles = 0;
|
|
|
|
|
|
|
|
cFile::cFile(void)
|
|
|
|
{
|
|
|
|
f = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
cFile::~cFile()
|
|
|
|
{
|
|
|
|
Close();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cFile::Open(const char *FileName, int Flags, mode_t Mode)
|
|
|
|
{
|
|
|
|
if (!IsOpen())
|
|
|
|
return Open(open(FileName, Flags, Mode));
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: attempt to re-open %s", FileName);
|
2000-09-17 08:23:46 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cFile::Open(int FileDes)
|
|
|
|
{
|
|
|
|
if (FileDes >= 0) {
|
|
|
|
if (!IsOpen()) {
|
|
|
|
f = FileDes;
|
|
|
|
if (f >= 0) {
|
|
|
|
if (f < FD_SETSIZE) {
|
|
|
|
if (f >= maxFiles)
|
|
|
|
maxFiles = f + 1;
|
|
|
|
if (!files[f])
|
|
|
|
files[f] = true;
|
|
|
|
else
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: file descriptor %d already in files[]", f);
|
2000-09-17 08:23:46 +02:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: file descriptor %d is larger than FD_SETSIZE (%d)", f, FD_SETSIZE);
|
2000-09-17 08:23:46 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: attempt to re-open file descriptor %d", FileDes);
|
2000-09-17 08:23:46 +02:00
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
void cFile::Close(void)
|
|
|
|
{
|
|
|
|
if (f >= 0) {
|
|
|
|
close(f);
|
|
|
|
files[f] = false;
|
|
|
|
f = -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cFile::Ready(bool Wait)
|
|
|
|
{
|
|
|
|
return f >= 0 && AnyFileReady(f, Wait ? 1000 : 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cFile::AnyFileReady(int FileDes, int TimeoutMs)
|
|
|
|
{
|
|
|
|
fd_set set;
|
|
|
|
FD_ZERO(&set);
|
|
|
|
for (int i = 0; i < maxFiles; i++) {
|
|
|
|
if (files[i])
|
|
|
|
FD_SET(i, &set);
|
|
|
|
}
|
|
|
|
if (0 <= FileDes && FileDes < FD_SETSIZE && !files[FileDes])
|
|
|
|
FD_SET(FileDes, &set); // in case we come in with an arbitrary descriptor
|
2000-09-17 15:36:33 +02:00
|
|
|
if (TimeoutMs == 0)
|
|
|
|
TimeoutMs = 10; // load gets too heavy with 0
|
2000-09-17 08:23:46 +02:00
|
|
|
struct timeval timeout;
|
|
|
|
timeout.tv_sec = TimeoutMs / 1000;
|
|
|
|
timeout.tv_usec = (TimeoutMs % 1000) * 1000;
|
|
|
|
return select(FD_SETSIZE, &set, NULL, NULL, &timeout) > 0 && (FileDes < 0 || FD_ISSET(FileDes, &set));
|
|
|
|
}
|
|
|
|
|
2000-10-08 09:25:20 +02:00
|
|
|
bool cFile::FileReady(int FileDes, int TimeoutMs)
|
|
|
|
{
|
|
|
|
fd_set set;
|
|
|
|
struct timeval timeout;
|
|
|
|
FD_ZERO(&set);
|
|
|
|
FD_SET(FileDes, &set);
|
2003-04-06 15:43:41 +02:00
|
|
|
if (TimeoutMs >= 0) {
|
|
|
|
if (TimeoutMs < 100)
|
|
|
|
TimeoutMs = 100;
|
|
|
|
timeout.tv_sec = TimeoutMs / 1000;
|
|
|
|
timeout.tv_usec = (TimeoutMs % 1000) * 1000;
|
|
|
|
}
|
|
|
|
return select(FD_SETSIZE, &set, NULL, NULL, (TimeoutMs >= 0) ? &timeout : NULL) > 0 && FD_ISSET(FileDes, &set);
|
2000-10-08 09:25:20 +02:00
|
|
|
}
|
|
|
|
|
2001-06-02 10:47:40 +02:00
|
|
|
bool cFile::FileReadyForWriting(int FileDes, int TimeoutMs)
|
|
|
|
{
|
|
|
|
fd_set set;
|
|
|
|
struct timeval timeout;
|
|
|
|
FD_ZERO(&set);
|
|
|
|
FD_SET(FileDes, &set);
|
|
|
|
if (TimeoutMs < 100)
|
|
|
|
TimeoutMs = 100;
|
|
|
|
timeout.tv_sec = 0;
|
|
|
|
timeout.tv_usec = TimeoutMs * 1000;
|
|
|
|
return select(FD_SETSIZE, NULL, &set, NULL, &timeout) > 0 && FD_ISSET(FileDes, &set);
|
|
|
|
}
|
|
|
|
|
2001-01-13 15:36:31 +01:00
|
|
|
// --- cSafeFile -------------------------------------------------------------
|
|
|
|
|
|
|
|
cSafeFile::cSafeFile(const char *FileName)
|
|
|
|
{
|
|
|
|
f = NULL;
|
|
|
|
fileName = ReadLink(FileName);
|
2002-08-11 13:32:23 +02:00
|
|
|
tempName = fileName ? MALLOC(char, strlen(fileName) + 5) : NULL;
|
2001-01-13 15:36:31 +01:00
|
|
|
if (tempName)
|
|
|
|
strcat(strcpy(tempName, fileName), ".$$$");
|
|
|
|
}
|
|
|
|
|
|
|
|
cSafeFile::~cSafeFile()
|
|
|
|
{
|
|
|
|
if (f)
|
|
|
|
fclose(f);
|
|
|
|
unlink(tempName);
|
2002-08-11 13:32:23 +02:00
|
|
|
free(fileName);
|
|
|
|
free(tempName);
|
2001-01-13 15:36:31 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
bool cSafeFile::Open(void)
|
|
|
|
{
|
|
|
|
if (!f && fileName && tempName) {
|
|
|
|
f = fopen(tempName, "w");
|
|
|
|
if (!f)
|
|
|
|
LOG_ERROR_STR(tempName);
|
|
|
|
}
|
|
|
|
return f != NULL;
|
|
|
|
}
|
|
|
|
|
2001-09-16 08:57:58 +02:00
|
|
|
bool cSafeFile::Close(void)
|
2001-01-13 15:36:31 +01:00
|
|
|
{
|
2001-09-16 08:57:58 +02:00
|
|
|
bool result = true;
|
2001-01-13 15:36:31 +01:00
|
|
|
if (f) {
|
2001-09-16 08:57:58 +02:00
|
|
|
if (ferror(f) != 0) {
|
|
|
|
LOG_ERROR_STR(tempName);
|
|
|
|
result = false;
|
|
|
|
}
|
|
|
|
if (fclose(f) < 0) {
|
|
|
|
LOG_ERROR_STR(tempName);
|
|
|
|
result = false;
|
|
|
|
}
|
2001-01-13 15:36:31 +01:00
|
|
|
f = NULL;
|
2001-09-16 08:57:58 +02:00
|
|
|
if (result && rename(tempName, fileName) < 0) {
|
|
|
|
LOG_ERROR_STR(fileName);
|
|
|
|
result = false;
|
|
|
|
}
|
2001-01-13 15:36:31 +01:00
|
|
|
}
|
2001-09-16 08:57:58 +02:00
|
|
|
else
|
|
|
|
result = false;
|
|
|
|
return result;
|
2001-01-13 15:36:31 +01:00
|
|
|
}
|
|
|
|
|
2005-10-31 13:14:26 +01:00
|
|
|
// --- cUnbufferedFile -------------------------------------------------------
|
|
|
|
|
2006-01-05 11:32:32 +01:00
|
|
|
//#define USE_FADVISE
|
|
|
|
|
2005-10-31 13:14:26 +01:00
|
|
|
#define READ_AHEAD MEGABYTE(2)
|
|
|
|
#define WRITE_BUFFER MEGABYTE(10)
|
|
|
|
|
|
|
|
cUnbufferedFile::cUnbufferedFile(void)
|
|
|
|
{
|
|
|
|
fd = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
cUnbufferedFile::~cUnbufferedFile()
|
|
|
|
{
|
|
|
|
Close();
|
|
|
|
}
|
|
|
|
|
|
|
|
int cUnbufferedFile::Open(const char *FileName, int Flags, mode_t Mode)
|
|
|
|
{
|
|
|
|
Close();
|
|
|
|
fd = open(FileName, Flags, Mode);
|
|
|
|
begin = end = ahead = -1;
|
|
|
|
written = 0;
|
|
|
|
return fd;
|
|
|
|
}
|
|
|
|
|
|
|
|
int cUnbufferedFile::Close(void)
|
|
|
|
{
|
2006-01-05 11:32:32 +01:00
|
|
|
#ifdef USE_FADVISE
|
2005-10-31 13:14:26 +01:00
|
|
|
if (fd >= 0) {
|
|
|
|
if (ahead > end)
|
|
|
|
end = ahead;
|
|
|
|
if (begin >= 0 && end > begin) {
|
|
|
|
//dsyslog("close buffer: %d (flush: %d bytes, %ld-%ld)", fd, written, begin, end);
|
|
|
|
if (written)
|
|
|
|
fdatasync(fd);
|
|
|
|
posix_fadvise(fd, begin, end - begin, POSIX_FADV_DONTNEED);
|
|
|
|
}
|
|
|
|
begin = end = ahead = -1;
|
|
|
|
written = 0;
|
|
|
|
}
|
2006-01-05 11:32:32 +01:00
|
|
|
#endif
|
2005-10-31 13:14:26 +01:00
|
|
|
int OldFd = fd;
|
|
|
|
fd = -1;
|
|
|
|
return close(OldFd);
|
|
|
|
}
|
|
|
|
|
|
|
|
off_t cUnbufferedFile::Seek(off_t Offset, int Whence)
|
|
|
|
{
|
|
|
|
if (fd >= 0)
|
|
|
|
return lseek(fd, Offset, Whence);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t cUnbufferedFile::Read(void *Data, size_t Size)
|
|
|
|
{
|
|
|
|
if (fd >= 0) {
|
2006-01-05 11:32:32 +01:00
|
|
|
#ifdef USE_FADVISE
|
2005-10-31 13:14:26 +01:00
|
|
|
off_t pos = lseek(fd, 0, SEEK_CUR);
|
|
|
|
// jump forward - adjust end position
|
|
|
|
if (pos > end)
|
|
|
|
end = pos;
|
|
|
|
// after adjusting end - don't clear more than previously requested
|
|
|
|
if (end > ahead)
|
|
|
|
end = ahead;
|
|
|
|
// jump backward - drop read ahead of previous run
|
|
|
|
if (pos < begin)
|
|
|
|
end = ahead;
|
|
|
|
if (begin >= 0 && end > begin)
|
|
|
|
posix_fadvise(fd, begin - KILOBYTE(200), end - begin + KILOBYTE(200), POSIX_FADV_DONTNEED);//XXX macros/parameters???
|
|
|
|
begin = pos;
|
2006-01-05 11:32:32 +01:00
|
|
|
#endif
|
2005-10-31 13:14:26 +01:00
|
|
|
ssize_t bytesRead = safe_read(fd, Data, Size);
|
2006-01-05 11:32:32 +01:00
|
|
|
#ifdef USE_FADVISE
|
2005-10-31 13:14:26 +01:00
|
|
|
if (bytesRead > 0) {
|
|
|
|
pos += bytesRead;
|
|
|
|
end = pos;
|
|
|
|
// this seems to trigger a non blocking read - this
|
|
|
|
// may or may not have been finished when we will be called next time.
|
|
|
|
// If it is not finished we can't release the not yet filled buffers.
|
|
|
|
// So this is commented out till we find a better solution.
|
|
|
|
//posix_fadvise(fd, pos, READ_AHEAD, POSIX_FADV_WILLNEED);
|
|
|
|
ahead = pos + READ_AHEAD;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
end = pos;
|
2006-01-05 11:32:32 +01:00
|
|
|
#endif
|
2005-10-31 13:14:26 +01:00
|
|
|
return bytesRead;
|
|
|
|
}
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t cUnbufferedFile::Write(const void *Data, size_t Size)
|
|
|
|
{
|
|
|
|
if (fd >=0) {
|
2006-01-05 11:32:32 +01:00
|
|
|
#ifdef USE_FADVISE
|
2005-10-31 13:14:26 +01:00
|
|
|
off_t pos = lseek(fd, 0, SEEK_CUR);
|
2006-01-05 11:32:32 +01:00
|
|
|
#endif
|
2005-10-31 13:14:26 +01:00
|
|
|
ssize_t bytesWritten = safe_write(fd, Data, Size);
|
2006-01-05 11:32:32 +01:00
|
|
|
#ifdef USE_FADVISE
|
2005-10-31 13:14:26 +01:00
|
|
|
if (bytesWritten >= 0) {
|
|
|
|
written += bytesWritten;
|
|
|
|
if (begin >= 0) {
|
|
|
|
if (pos < begin)
|
|
|
|
begin = pos;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
begin = pos;
|
|
|
|
if (pos + bytesWritten > end)
|
|
|
|
end = pos + bytesWritten;
|
|
|
|
if (written > WRITE_BUFFER) {
|
|
|
|
//dsyslog("flush buffer: %d (%d bytes, %ld-%ld)", fd, written, begin, end);
|
|
|
|
fdatasync(fd);
|
|
|
|
if (begin >= 0 && end > begin)
|
|
|
|
posix_fadvise(fd, begin, end - begin, POSIX_FADV_DONTNEED);
|
|
|
|
begin = end = -1;
|
|
|
|
written = 0;
|
|
|
|
}
|
|
|
|
}
|
2006-01-05 11:32:32 +01:00
|
|
|
#endif
|
2005-10-31 13:14:26 +01:00
|
|
|
return bytesWritten;
|
|
|
|
}
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
cUnbufferedFile *cUnbufferedFile::Create(const char *FileName, int Flags, mode_t Mode)
|
|
|
|
{
|
|
|
|
cUnbufferedFile *File = new cUnbufferedFile;
|
|
|
|
if (File->Open(FileName, Flags, Mode) < 0) {
|
|
|
|
delete File;
|
|
|
|
File = NULL;
|
|
|
|
}
|
|
|
|
return File;
|
|
|
|
}
|
|
|
|
|
2001-09-30 10:38:06 +02:00
|
|
|
// --- cLockFile -------------------------------------------------------------
|
|
|
|
|
|
|
|
#define LOCKFILENAME ".lock-vdr"
|
|
|
|
#define LOCKFILESTALETIME 600 // seconds before considering a lock file "stale"
|
|
|
|
|
|
|
|
cLockFile::cLockFile(const char *Directory)
|
|
|
|
{
|
|
|
|
fileName = NULL;
|
|
|
|
f = -1;
|
|
|
|
if (DirectoryOk(Directory))
|
|
|
|
asprintf(&fileName, "%s/%s", Directory, LOCKFILENAME);
|
|
|
|
}
|
|
|
|
|
|
|
|
cLockFile::~cLockFile()
|
|
|
|
{
|
|
|
|
Unlock();
|
2002-08-11 13:32:23 +02:00
|
|
|
free(fileName);
|
2001-09-30 10:38:06 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool cLockFile::Lock(int WaitSeconds)
|
|
|
|
{
|
|
|
|
if (f < 0 && fileName) {
|
|
|
|
time_t Timeout = time(NULL) + WaitSeconds;
|
|
|
|
do {
|
2005-08-06 09:56:08 +02:00
|
|
|
f = open(fileName, O_WRONLY | O_CREAT | O_EXCL, DEFFILEMODE);
|
2001-09-30 10:38:06 +02:00
|
|
|
if (f < 0) {
|
|
|
|
if (errno == EEXIST) {
|
|
|
|
struct stat fs;
|
|
|
|
if (stat(fileName, &fs) == 0) {
|
2002-11-24 20:18:55 +01:00
|
|
|
if (abs(time(NULL) - fs.st_mtime) > LOCKFILESTALETIME) {
|
2002-05-13 16:35:49 +02:00
|
|
|
esyslog("ERROR: removing stale lock file '%s'", fileName);
|
2001-09-30 10:38:06 +02:00
|
|
|
if (remove(fileName) < 0) {
|
|
|
|
LOG_ERROR_STR(fileName);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (errno != ENOENT) {
|
|
|
|
LOG_ERROR_STR(fileName);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
LOG_ERROR_STR(fileName);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (WaitSeconds)
|
|
|
|
sleep(1);
|
|
|
|
}
|
|
|
|
} while (f < 0 && time(NULL) < Timeout);
|
|
|
|
}
|
|
|
|
return f >= 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void cLockFile::Unlock(void)
|
|
|
|
{
|
|
|
|
if (f >= 0) {
|
|
|
|
close(f);
|
|
|
|
remove(fileName);
|
|
|
|
f = -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2000-02-19 13:36:48 +01:00
|
|
|
// --- cListObject -----------------------------------------------------------
|
|
|
|
|
|
|
|
cListObject::cListObject(void)
|
|
|
|
{
|
|
|
|
prev = next = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
cListObject::~cListObject()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void cListObject::Append(cListObject *Object)
|
|
|
|
{
|
|
|
|
next = Object;
|
|
|
|
Object->prev = this;
|
|
|
|
}
|
|
|
|
|
2002-05-12 14:46:46 +02:00
|
|
|
void cListObject::Insert(cListObject *Object)
|
|
|
|
{
|
|
|
|
prev = Object;
|
|
|
|
Object->next = this;
|
|
|
|
}
|
|
|
|
|
2000-02-19 13:36:48 +01:00
|
|
|
void cListObject::Unlink(void)
|
|
|
|
{
|
|
|
|
if (next)
|
|
|
|
next->prev = prev;
|
|
|
|
if (prev)
|
|
|
|
prev->next = next;
|
2000-03-11 11:22:37 +01:00
|
|
|
next = prev = NULL;
|
2000-02-19 13:36:48 +01:00
|
|
|
}
|
|
|
|
|
2005-03-20 15:15:42 +01:00
|
|
|
int cListObject::Index(void) const
|
2000-02-19 13:36:48 +01:00
|
|
|
{
|
|
|
|
cListObject *p = prev;
|
|
|
|
int i = 0;
|
|
|
|
|
|
|
|
while (p) {
|
|
|
|
i++;
|
|
|
|
p = p->prev;
|
|
|
|
}
|
|
|
|
return i;
|
|
|
|
}
|
|
|
|
|
|
|
|
// --- cListBase -------------------------------------------------------------
|
|
|
|
|
|
|
|
cListBase::cListBase(void)
|
2006-01-08 11:44:37 +01:00
|
|
|
{
|
2000-02-19 13:36:48 +01:00
|
|
|
objects = lastObject = NULL;
|
2005-05-26 11:41:33 +02:00
|
|
|
count = 0;
|
2000-02-19 13:36:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
cListBase::~cListBase()
|
|
|
|
{
|
|
|
|
Clear();
|
|
|
|
}
|
|
|
|
|
2002-05-12 14:46:46 +02:00
|
|
|
void cListBase::Add(cListObject *Object, cListObject *After)
|
2006-01-08 11:44:37 +01:00
|
|
|
{
|
2002-05-12 14:46:46 +02:00
|
|
|
if (After && After != lastObject) {
|
|
|
|
After->Next()->Insert(Object);
|
|
|
|
After->Append(Object);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
if (lastObject)
|
|
|
|
lastObject->Append(Object);
|
|
|
|
else
|
|
|
|
objects = Object;
|
|
|
|
lastObject = Object;
|
|
|
|
}
|
2005-05-26 11:41:33 +02:00
|
|
|
count++;
|
2002-05-12 14:46:46 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void cListBase::Ins(cListObject *Object, cListObject *Before)
|
2006-01-08 11:44:37 +01:00
|
|
|
{
|
2002-05-12 14:46:46 +02:00
|
|
|
if (Before && Before != objects) {
|
|
|
|
Before->Prev()->Append(Object);
|
|
|
|
Before->Insert(Object);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
if (objects)
|
|
|
|
objects->Insert(Object);
|
|
|
|
else
|
|
|
|
lastObject = Object;
|
2002-05-13 17:56:17 +02:00
|
|
|
objects = Object;
|
2002-05-12 14:46:46 +02:00
|
|
|
}
|
2005-05-26 11:41:33 +02:00
|
|
|
count++;
|
2000-02-19 13:36:48 +01:00
|
|
|
}
|
|
|
|
|
2002-05-19 15:50:11 +02:00
|
|
|
void cListBase::Del(cListObject *Object, bool DeleteObject)
|
2000-02-19 13:36:48 +01:00
|
|
|
{
|
|
|
|
if (Object == objects)
|
|
|
|
objects = Object->Next();
|
|
|
|
if (Object == lastObject)
|
|
|
|
lastObject = Object->Prev();
|
|
|
|
Object->Unlink();
|
2002-05-19 15:50:11 +02:00
|
|
|
if (DeleteObject)
|
|
|
|
delete Object;
|
2005-05-26 11:41:33 +02:00
|
|
|
count--;
|
2000-02-19 13:36:48 +01:00
|
|
|
}
|
|
|
|
|
2000-03-11 11:22:37 +01:00
|
|
|
void cListBase::Move(int From, int To)
|
|
|
|
{
|
|
|
|
Move(Get(From), Get(To));
|
|
|
|
}
|
|
|
|
|
|
|
|
void cListBase::Move(cListObject *From, cListObject *To)
|
|
|
|
{
|
|
|
|
if (From && To) {
|
|
|
|
if (From->Index() < To->Index())
|
|
|
|
To = To->Next();
|
|
|
|
if (From == objects)
|
|
|
|
objects = From->Next();
|
|
|
|
if (From == lastObject)
|
|
|
|
lastObject = From->Prev();
|
|
|
|
From->Unlink();
|
|
|
|
if (To) {
|
|
|
|
if (To->Prev())
|
|
|
|
To->Prev()->Append(From);
|
|
|
|
From->Append(To);
|
|
|
|
}
|
2002-02-05 18:28:14 +01:00
|
|
|
else {
|
2000-03-11 11:22:37 +01:00
|
|
|
lastObject->Append(From);
|
2002-02-05 18:28:14 +01:00
|
|
|
lastObject = From;
|
|
|
|
}
|
2000-03-11 11:22:37 +01:00
|
|
|
if (!From->Prev())
|
|
|
|
objects = From;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2000-02-19 13:36:48 +01:00
|
|
|
void cListBase::Clear(void)
|
|
|
|
{
|
|
|
|
while (objects) {
|
|
|
|
cListObject *object = objects->Next();
|
|
|
|
delete objects;
|
|
|
|
objects = object;
|
|
|
|
}
|
|
|
|
objects = lastObject = NULL;
|
2005-05-26 11:41:33 +02:00
|
|
|
count = 0;
|
2000-02-19 13:36:48 +01:00
|
|
|
}
|
|
|
|
|
2000-10-29 13:17:22 +01:00
|
|
|
cListObject *cListBase::Get(int Index) const
|
2000-02-19 13:36:48 +01:00
|
|
|
{
|
2000-09-09 14:57:43 +02:00
|
|
|
if (Index < 0)
|
|
|
|
return NULL;
|
2000-02-19 13:36:48 +01:00
|
|
|
cListObject *object = objects;
|
|
|
|
while (object && Index-- > 0)
|
|
|
|
object = object->Next();
|
|
|
|
return object;
|
|
|
|
}
|
|
|
|
|
2004-11-01 10:40:38 +01:00
|
|
|
static int CompareListObjects(const void *a, const void *b)
|
|
|
|
{
|
|
|
|
const cListObject *la = *(const cListObject **)a;
|
|
|
|
const cListObject *lb = *(const cListObject **)b;
|
|
|
|
return la->Compare(*lb);
|
|
|
|
}
|
|
|
|
|
2001-08-26 14:17:20 +02:00
|
|
|
void cListBase::Sort(void)
|
|
|
|
{
|
2004-11-01 10:40:38 +01:00
|
|
|
int n = Count();
|
|
|
|
cListObject *a[n];
|
|
|
|
cListObject *object = objects;
|
|
|
|
int i = 0;
|
|
|
|
while (object && i < n) {
|
|
|
|
a[i++] = object;
|
|
|
|
object = object->Next();
|
|
|
|
}
|
|
|
|
qsort(a, n, sizeof(cListObject *), CompareListObjects);
|
|
|
|
objects = lastObject = NULL;
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
a[i]->Unlink();
|
2005-05-26 11:41:33 +02:00
|
|
|
count--;
|
2004-11-01 10:40:38 +01:00
|
|
|
Add(a[i]);
|
|
|
|
}
|
2001-08-26 14:17:20 +02:00
|
|
|
}
|
|
|
|
|
2005-05-28 13:17:20 +02:00
|
|
|
// --- cHashBase -------------------------------------------------------------
|
|
|
|
|
|
|
|
cHashBase::cHashBase(int Size)
|
|
|
|
{
|
|
|
|
size = Size;
|
|
|
|
hashTable = (cList<cHashObject>**)calloc(size, sizeof(cList<cHashObject>*));
|
|
|
|
}
|
|
|
|
|
|
|
|
cHashBase::~cHashBase(void)
|
|
|
|
{
|
2005-09-11 13:23:49 +02:00
|
|
|
Clear();
|
2005-05-28 13:17:20 +02:00
|
|
|
free(hashTable);
|
|
|
|
}
|
|
|
|
|
2005-05-29 10:24:54 +02:00
|
|
|
void cHashBase::Add(cListObject *Object, unsigned int Id)
|
2005-05-28 13:17:20 +02:00
|
|
|
{
|
2005-05-29 10:24:54 +02:00
|
|
|
unsigned int hash = hashfn(Id);
|
2005-05-28 13:17:20 +02:00
|
|
|
if (!hashTable[hash])
|
|
|
|
hashTable[hash] = new cList<cHashObject>;
|
|
|
|
hashTable[hash]->Add(new cHashObject(Object, Id));
|
|
|
|
}
|
|
|
|
|
2005-05-29 10:24:54 +02:00
|
|
|
void cHashBase::Del(cListObject *Object, unsigned int Id)
|
2005-05-28 13:17:20 +02:00
|
|
|
{
|
|
|
|
cList<cHashObject> *list = hashTable[hashfn(Id)];
|
|
|
|
if (list) {
|
|
|
|
for (cHashObject *hob = list->First(); hob; hob = list->Next(hob)) {
|
|
|
|
if (hob->object == Object) {
|
|
|
|
list->Del(hob);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2005-09-11 13:23:49 +02:00
|
|
|
void cHashBase::Clear(void)
|
|
|
|
{
|
|
|
|
for (int i = 0; i < size; i++) {
|
|
|
|
delete hashTable[i];
|
|
|
|
hashTable[i] = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2005-05-29 10:24:54 +02:00
|
|
|
cListObject *cHashBase::Get(unsigned int Id) const
|
2005-05-28 13:17:20 +02:00
|
|
|
{
|
|
|
|
cList<cHashObject> *list = hashTable[hashfn(Id)];
|
|
|
|
if (list) {
|
|
|
|
for (cHashObject *hob = list->First(); hob; hob = list->Next(hob)) {
|
|
|
|
if (hob->id == Id)
|
|
|
|
return hob->object;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2005-05-29 10:24:54 +02:00
|
|
|
cList<cHashObject> *cHashBase::GetList(unsigned int Id) const
|
2005-05-28 13:17:20 +02:00
|
|
|
{
|
|
|
|
return hashTable[hashfn(Id)];
|
|
|
|
}
|