lsyncd/core/core.c

1568 lines
28 KiB
C
Raw Normal View History

/*
2018-03-30 13:15:49 +00:00
| core.c from Lsyncd -- the Live (Mirror) Syncing Demon
2018-03-27 06:54:14 +00:00
|
|
|
| License: GPLv2 (see COPYING) or any later version
| Authors: Axel Kittenberger <axkibe@gmail.com>
*/
2010-11-22 21:17:08 +00:00
2010-11-22 14:54:50 +00:00
#include "lsyncd.h"
2010-10-16 10:26:48 +00:00
#define SYSLOG_NAMES 1
2011-03-30 14:08:01 +00:00
#include <sys/select.h>
2010-10-16 18:21:01 +00:00
#include <sys/stat.h>
2010-10-19 10:12:11 +00:00
#include <sys/times.h>
2010-10-18 12:23:46 +00:00
#include <sys/types.h>
#include <sys/wait.h>
2010-10-16 18:21:01 +00:00
#include <dirent.h>
2010-10-16 10:26:48 +00:00
#include <errno.h>
2010-10-28 17:56:33 +00:00
#include <fcntl.h>
2010-10-16 18:21:01 +00:00
#include <limits.h>
#include <signal.h>
#include <stdbool.h>
#include <stddef.h>
2010-10-16 18:21:01 +00:00
#include <stdlib.h>
2010-10-14 13:52:01 +00:00
#include <stdio.h>
2010-10-16 10:26:48 +00:00
#include <string.h>
#include <strings.h>
2010-10-20 15:34:01 +00:00
#include <syslog.h>
2010-11-04 13:43:57 +00:00
#include <math.h>
2010-10-20 15:34:01 +00:00
#include <time.h>
2010-10-16 10:26:48 +00:00
#include <unistd.h>
2011-02-25 07:57:11 +00:00
#define LUA_USE_APICHECK 1
2010-10-14 13:52:01 +00:00
#include <lua.h>
#include <lualib.h>
#include <lauxlib.h>
2018-03-23 08:34:33 +00:00
#include "log.h"
2018-03-27 07:09:07 +00:00
#include "mem.h"
#include "util.h"
2018-03-26 17:05:00 +00:00
#include "pipe.h"
2018-03-28 07:17:49 +00:00
#include "observe.h"
2018-03-30 07:40:09 +00:00
#include "time.h"
2018-03-22 08:14:13 +00:00
2018-03-27 06:54:14 +00:00
#ifdef WITH_INOTIFY
#include "inotify.h"
#endif
/*
| The Lua part of Lsyncd
*/
2018-03-23 16:36:51 +00:00
extern const char mantle_out[];
extern size_t mantle_size;
/*
| The Lua coded default sync implementations
*/
extern const char default_out[];
extern size_t default_size;
/*
| Makes sure there is one file system monitor.
*/
2014-04-29 14:11:27 +00:00
#ifndef WITH_INOTIFY
# error "needing at least one notification system. please rerun cmake"
2010-11-28 08:13:05 +00:00
#endif
/*
| All monitors supported by this Lsyncd.
*/
2010-11-28 08:13:05 +00:00
static char *monitors[] = {
2014-04-29 14:11:27 +00:00
#ifdef WITH_INOTIFY
2010-11-28 08:13:05 +00:00
"inotify",
#endif
2010-11-28 08:13:05 +00:00
NULL,
};
2016-05-03 08:45:22 +00:00
2010-10-27 19:34:56 +00:00
/**
| Configuration parameters that matter to the core
*/
struct settings settings = {
.log_file = NULL,
.log_syslog = false,
.log_ident = NULL,
.log_facility = LOG_USER,
.log_level = LOG_NOTICE
2010-10-27 19:34:56 +00:00
};
2010-10-20 15:34:01 +00:00
2016-05-03 08:45:22 +00:00
/*
| True if stdout and stderr are detected to
| be directed to /dev/null.
*/
2018-03-23 08:34:33 +00:00
bool no_output = false;
2016-05-03 08:45:22 +00:00
/*
| The config file loaded by Lsyncd.
*/
char * lsyncd_config_file = NULL;
2010-10-27 19:34:56 +00:00
2016-05-03 08:45:22 +00:00
/*
| False after first time Lsyncd started up.
|
| Configuration error messages are thus written to
| stdout/stderr only on first start.
*/
2018-03-23 08:34:33 +00:00
bool first_time = true;
2010-10-27 19:34:56 +00:00
2016-05-03 08:45:22 +00:00
/*
| Set by TERM or HUP signal handler
| telling Lsyncd should end or reset ASAP.
*/
volatile sig_atomic_t hup = 0;
volatile sig_atomic_t term = 0;
volatile sig_atomic_t sigcode = 0;
2010-10-16 10:26:48 +00:00
2016-05-03 08:45:22 +00:00
/*
| The kernel's clock ticks per second.
*/
2018-03-30 07:36:26 +00:00
extern long clocks_per_sec;
2010-10-19 10:12:11 +00:00
2016-05-03 08:45:22 +00:00
/*
| signal handler
*/
void sig_child( int sig ) { /* nothing */ }
2010-11-14 09:11:09 +00:00
2016-05-03 08:45:22 +00:00
/*
| signal handler
*/
2010-11-14 09:11:09 +00:00
void
sig_handler( int sig )
2010-11-14 09:11:09 +00:00
{
switch( sig )
{
case SIGTERM:
case SIGINT:
term = 1;
sigcode = sig;
return;
case SIGHUP:
hup = 1;
return;
2010-11-14 09:11:09 +00:00
}
}
2016-05-03 08:45:22 +00:00
/*:::::::::::::::::::.
:: Helper Routines
'::::::::::::::::::::*/
/*
| Variable which address is used as
| the cores index in the lua registry to
| index the mantle-core-interface
|
| Its value is used to determined if the
2018-03-14 08:57:49 +00:00
| mantle has registered itself already.
*/
static int mci = 0;
2010-12-01 12:19:17 +00:00
2016-05-03 08:45:22 +00:00
/*
| Dummy variable which address is used as
| the cores index n the lua registry to
| the lua runners error handler.
*/
2010-12-01 12:19:17 +00:00
static int callError;
/*
| A user observance became read-ready.
*/
2010-12-01 12:19:17 +00:00
static void
user_obs_ready(
lua_State * L,
2018-03-28 07:17:49 +00:00
int fd,
void * extra
)
2010-12-01 12:19:17 +00:00
{
2012-02-15 11:15:32 +00:00
// pushes the ready table on table
lua_pushlightuserdata( L, ( void * ) user_obs_ready );
lua_gettable( L, LUA_REGISTRYINDEX );
2012-02-15 11:15:32 +00:00
// pushes the error handler
lua_pushlightuserdata( L, (void *) &callError );
lua_gettable( L, LUA_REGISTRYINDEX );
2010-12-01 12:19:17 +00:00
// pushes the user func
lua_pushnumber( L, fd );
lua_gettable( L, -3 );
2010-12-01 12:19:17 +00:00
2012-02-15 11:15:32 +00:00
// gives the ufunc the fd
lua_pushnumber( L, fd );
2010-12-01 12:19:17 +00:00
2012-02-15 11:15:32 +00:00
// calls the user function
if( lua_pcall( L, 1, 0, -3 ) ) exit( -1 );
lua_pop( L, 2 );
2010-12-01 12:19:17 +00:00
}
/*
| A user observance became write-ready
*/
2010-12-01 12:19:17 +00:00
static void
user_obs_writey(
lua_State * L,
2018-03-28 07:17:49 +00:00
int fd,
void * extra
)
2010-12-01 12:19:17 +00:00
{
2012-02-15 11:15:32 +00:00
// pushes the writey table on table
lua_pushlightuserdata( L, (void *) user_obs_writey );
lua_gettable( L, LUA_REGISTRYINDEX );
2010-12-01 12:19:17 +00:00
2012-02-15 11:15:32 +00:00
// pushes the error handler
2010-12-01 12:19:17 +00:00
lua_pushlightuserdata(L, (void *) &callError);
lua_gettable( L, LUA_REGISTRYINDEX );
2010-12-01 12:19:17 +00:00
2012-02-15 11:15:32 +00:00
// pushes the user func
lua_pushnumber( L, fd );
lua_gettable( L, -3 );
2010-12-01 12:19:17 +00:00
2012-02-15 11:15:32 +00:00
// gives the user func the fd
lua_pushnumber( L, fd );
2010-12-01 12:19:17 +00:00
2012-02-15 11:15:32 +00:00
// calls the user function
if( lua_pcall( L, 1, 0, -3 ) ) exit(-1);
lua_pop( L, 2 );
2010-12-01 12:19:17 +00:00
}
/*
| Tidies up a user observance
| FIXME - give the user a chance to do something in that case!
*/
2010-12-01 12:19:17 +00:00
static void
2018-03-28 07:17:49 +00:00
user_obs_tidy(
int fd,
void * extra
)
2010-12-01 12:19:17 +00:00
{
2018-03-28 07:17:49 +00:00
close( fd );
2010-12-01 12:19:17 +00:00
}
2010-11-12 15:39:43 +00:00
/*:::::::::::::::::::::::::::::::.
2018-03-14 08:57:49 +00:00
:: Library calls for the mantle
'::::::::::::::::::::::::::::::::*/
2010-10-17 15:24:55 +00:00
2010-11-04 13:43:57 +00:00
int l_stackdump( lua_State* L );
/*
| Logs a message.
|
| Params on Lua stack:
|
| 1: loglevel of massage
| 2: the string to log
*/
2012-02-15 11:15:32 +00:00
static int
l_log( lua_State *L )
2010-10-20 15:34:01 +00:00
{
2012-02-15 11:15:32 +00:00
const char * cat; // log category
const char * message; // log message
int priority; // log priority
2010-11-03 14:54:33 +00:00
cat = luaL_checkstring( L, 1 );
priority = check_logcat( cat );
// skips filtered messages
2018-03-27 15:43:26 +00:00
if( priority > settings.log_level ) return 0;
2010-11-03 14:54:33 +00:00
// replaces non string values
2010-11-04 13:43:57 +00:00
{
int i;
int top = lua_gettop(L);
for( i = 1; i <= top; i++ )
{
int t = lua_type( L, i );
switch( t )
{
case LUA_TTABLE :
lua_pushfstring(
L,
"(Table: %p)",
lua_topointer( L, i )
);
lua_replace( L, i );
break;
case LUA_TBOOLEAN :
if( lua_toboolean( L, i ) )
{ lua_pushstring( L, "(true)" ); }
else
{ lua_pushstring( L, "(false)" ); }
lua_replace(L, i);
break;
case LUA_TUSERDATA:
{
2018-03-30 07:36:26 +00:00
clock_t *c = ( clock_t * ) luaL_checkudata( L, i, "Lsyncd.jiffies" );
double d = *c;
d /= clocks_per_sec;
lua_pushfstring( L, "(Timestamp: %f)", d );
lua_replace( L, i );
}
break;
case LUA_TNIL:
lua_pushstring( L, "(nil)" );
lua_replace( L, i );
break;
2010-11-04 13:43:57 +00:00
}
}
}
2012-02-15 11:15:32 +00:00
// concates if there is more than one string parameter
lua_concat( L, lua_gettop( L ) - 1 );
message = luaL_checkstring( L, 2 );
logstring0( priority, cat, message );
2010-11-02 17:07:42 +00:00
2010-10-20 15:34:01 +00:00
return 0;
}
/*
| Executes a subprocess. Does not wait for it to return.
|
| Params on Lua stack:
|
| 1: Path to binary to call
| 2: List of string as arguments
| or "<" in which case the next argument is a string
| that will be piped on stdin.
| The arguments will follow that one.
|
| Returns (Lua stack) the pid on success, 0 on failure.
*/
2010-10-17 17:13:53 +00:00
static int
l_exec( lua_State *L )
2010-10-17 17:13:53 +00:00
{
// the binary to call
const char *binary = luaL_checkstring(L, 1);
// number of arguments
int argc = lua_gettop( L ) - 1;
// the pid spawned
pid_t pid;
// the arguments position in the lua arguments
int li = 1;
// the pipe to text
char const * pipe_text = NULL;
// the pipes length
size_t pipe_len = 0;
// the arguments
char const ** argv;
// pipe file descriptors
int pipefd[ 2 ];
2012-02-15 11:15:32 +00:00
int i;
// expands tables
// and removes nils
for( i = 1; i <= lua_gettop( L ); i++ )
{
if( lua_isnil( L, i ) )
{
lua_remove( L, i );
i--;
argc--;
continue;
}
if( lua_istable( L, i ) )
{
2012-02-15 11:15:32 +00:00
int tlen;
int it;
lua_checkstack( L, lua_gettop( L ) + lua_objlen( L, i ) + 1 );
// moves table to top of stack
lua_pushvalue( L, i );
lua_remove( L, i );
2012-02-15 11:15:32 +00:00
argc--;
tlen = lua_objlen( L, -1 );
for( it = 1; it <= tlen; it++ )
{
lua_pushinteger( L, it );
lua_gettable( L, -2 );
lua_insert( L, i );
2012-02-15 11:15:32 +00:00
i++;
argc++;
}
2012-02-15 11:15:32 +00:00
i--;
lua_pop( L, 1 );
}
}
2012-02-15 11:15:32 +00:00
// writes a log message (if needed).
if( check_logcat( "Exec" ) <= settings.log_level )
{
lua_checkstack( L, lua_gettop( L ) + argc * 3 + 2 );
lua_pushvalue( L, 1 );
for( i = 1; i <= argc; i++ )
{
lua_pushstring( L, " [" );
lua_pushvalue( L, i + 1 );
lua_pushstring( L, "]" );
2010-11-05 18:04:29 +00:00
}
lua_concat( L, 3 * argc + 1 );
// replaces midfile 0 chars by linefeed
size_t len = 0;
const char * cs = lua_tolstring( L, -1, &len );
char * s = s_calloc( len + 1, sizeof( char ) );
for( i = 0; i < len; i++ )
{
s[ i ] = cs[ i ] ? cs[ i ] : '\n';
}
2018-03-26 16:09:22 +00:00
logstring0( LOG_DEBUG, "Exec", s );
free( s );
lua_pop( L, 1 );
2010-11-05 18:04:29 +00:00
}
if( argc >= 2 && !strcmp( luaL_checkstring( L, 2 ), "<" ) )
{
2012-02-15 12:47:58 +00:00
// pipes something into stdin
if( !lua_isstring( L, 3 ) )
{
2018-03-26 16:09:22 +00:00
logstring( "Error", "in spawn(), expected a string after pipe '<'" );
exit( -1 );
}
pipe_text = lua_tolstring( L, 3, &pipe_len );
if( strlen( pipe_text ) > 0 )
{
2018-03-26 17:05:00 +00:00
pipe_create( pipefd );
}
else
{
2011-01-20 18:55:46 +00:00
pipe_text = NULL;
2010-11-12 14:05:10 +00:00
}
2018-03-26 17:05:00 +00:00
2010-11-12 14:05:10 +00:00
argc -= 2;
li += 2;
}
2012-02-15 11:15:32 +00:00
// prepares the arguments
argv = s_calloc( argc + 2, sizeof( char * ) );
argv[ 0 ] = binary;
for( i = 1; i <= argc; i++ )
{
argv[i] = luaL_checkstring( L, i + li );
}
argv[ i ] = NULL;
// the fork!
pid = fork( );
2010-10-17 20:26:37 +00:00
if( pid == 0 )
{
2012-02-15 11:15:32 +00:00
// replaces stdin for pipes
2018-03-26 17:05:00 +00:00
if( pipe_text ) dup2( pipefd[ 0 ], STDIN_FILENO );
2012-02-15 11:15:32 +00:00
// if lsyncd runs as a daemon and has a logfile it will redirect
// stdout/stderr of child processes to the logfile.
if( settings.log_file )
{
if( !freopen( settings.log_file, "a", stdout ) )
{
printlogf( L, "Error", "cannot redirect stdout to '%s'.", settings.log_file );
2010-10-28 17:56:33 +00:00
}
if( !freopen( settings.log_file, "a", stderr ) )
{
printlogf( L, "Error", "cannot redirect stderr to '%s'.", settings.log_file );
2010-10-28 17:56:33 +00:00
}
}
execv( binary, ( char ** ) argv );
2012-02-15 11:15:32 +00:00
// in a sane world execv does not return!
printlogf( L, "Error", "Failed executing [ %s ]!", binary );
exit( -1 );
2010-10-17 17:13:53 +00:00
}
if( pipe_text )
{
2018-03-26 17:05:00 +00:00
// closes read-end of pipe, this is for child process only
close( pipefd[ 0 ] );
2018-03-26 17:05:00 +00:00
pipe_write( pipefd, pipe_text, pipe_len );
2010-11-12 14:05:10 +00:00
}
free( argv );
lua_pushnumber( L, pid );
2010-10-18 09:02:51 +00:00
return 1;
2010-10-17 17:13:53 +00:00
}
/*
| Registers the mantle core interface with the core.
|
| Params on Lua stack:
2018-03-14 08:57:49 +00:00
| 1: The luacode mantle.
|
| Returns on Lua stack:
| nothing
*/
static int
l_mci( lua_State *L )
{
if( mci )
{
logstring( "Error", "Luacode interface already registered!" );
exit( -1 );
}
mci = 1;
lua_pushlightuserdata( L, (void *) &mci );
// switches the passed mantle interface as parameter and the key &mci
lua_insert( L, 1 );
// saves the table of the mci in the lua registry
lua_settable( L, LUA_REGISTRYINDEX );
// saves the error function extras
2018-03-14 08:57:49 +00:00
lua_pushlightuserdata( L, (void *) &callError );
lua_pushlightuserdata( L, (void *) &mci );
2018-03-14 08:57:49 +00:00
lua_gettable( L, LUA_REGISTRYINDEX );
lua_pushstring( L, "callError" );
lua_gettable( L, -2 );
lua_remove( L, -2 );
lua_settable( L, LUA_REGISTRYINDEX );
if( lua_gettop( L ) )
{
logstring( "Error", "internal, stack is dirty." );
l_stackdump( L );
exit( -1 );
}
}
/*
| Converts a relative directory path to an absolute.
|
| Params on Lua stack:
| 1: a relative path to directory
|
| Returns on Lua stack:
| The absolute path of directory
*/
2010-10-17 15:24:55 +00:00
static int
l_realdir( lua_State *L )
2010-10-17 15:24:55 +00:00
{
2010-10-16 18:21:01 +00:00
luaL_Buffer b;
const char *rdir = luaL_checkstring(L, 1);
char *adir = get_realpath(rdir);
2012-02-15 11:15:32 +00:00
2016-05-03 08:45:22 +00:00
if( !adir )
{
printlogf(
L, "Error",
"failure getting absolute path of [%s]",
rdir
);
2010-10-16 18:21:01 +00:00
return 0;
}
2012-02-15 12:47:58 +00:00
2010-10-16 18:21:01 +00:00
{
2012-02-15 11:15:32 +00:00
// makes sure its a directory
2010-10-16 18:21:01 +00:00
struct stat st;
2016-05-03 08:45:22 +00:00
if( stat( adir, &st ) )
{
printlogf(
L, "Error",
"cannot get absolute path of dir '%s': %s",
rdir,
strerror( errno )
);
free( adir );
2010-10-28 17:56:33 +00:00
return 0;
}
2016-05-03 08:45:22 +00:00
if( !S_ISDIR( st.st_mode ) )
{
printlogf(
L, "Error",
"cannot get absolute path of dir '%s': is not a directory",
rdir
);
free( adir );
2010-10-16 18:21:01 +00:00
return 0;
}
}
2012-02-15 11:15:32 +00:00
// returns absolute path with a concated '/'
2016-05-03 08:45:22 +00:00
luaL_buffinit( L, &b );
luaL_addstring( &b, adir );
luaL_addchar( &b, '/' );
luaL_pushresult( &b );
free( adir );
2010-10-16 18:21:01 +00:00
return 1;
}
/*
| Dumps the Lua stack.
| For debugging purposes.
*/
int
l_stackdump( lua_State * L )
{
int i;
int top = lua_gettop( L );
2018-03-14 08:57:49 +00:00
printlogf( L, "Debug", "total on stack %d", top );
for( i = 1; i <= top; i++ )
{
int t = lua_type( L, i );
switch( t )
{
case LUA_TSTRING:
2016-05-03 08:45:22 +00:00
printlogf(
L, "Debug",
"%d string: '%s'", i, lua_tostring( L, i )
);
2016-05-03 08:45:22 +00:00
break;
case LUA_TBOOLEAN:
2016-05-03 08:45:22 +00:00
2018-03-26 16:09:22 +00:00
printlogf( L, "Debug",
"%d boolean %s", i, lua_toboolean( L, i ) ? "true" : "false"
);
2016-05-03 08:45:22 +00:00
break;
case LUA_TNUMBER:
2016-05-03 08:45:22 +00:00
2018-03-26 16:09:22 +00:00
printlogf( L, "Debug", "%d number: %g", i, lua_tonumber( L, i ) );
2016-05-03 08:45:22 +00:00
break;
default:
2016-05-03 08:45:22 +00:00
2018-03-26 16:09:22 +00:00
printlogf( L, "Debug", "%d %s", i, lua_typename( L, t ) );
2016-05-03 08:45:22 +00:00
break;
2010-10-17 15:24:55 +00:00
}
2010-10-17 20:26:37 +00:00
}
2010-10-17 15:24:55 +00:00
return 0;
}
/*
| Reads the directories entries.
|
| Params on Lua stack:
| 1: absolute path to directory
|
| Returns on Lua stack:
| a table of directory names.
| names are keys
| values are boolean true on dirs.
*/
2010-10-17 15:24:55 +00:00
static int
2016-12-05 14:11:00 +00:00
l_readdir( lua_State *L )
2010-10-17 15:24:55 +00:00
{
const char * dirname = luaL_checkstring( L, 1 );
2016-12-05 14:11:00 +00:00
2010-10-16 18:21:01 +00:00
DIR *d;
d = opendir( dirname );
2016-05-03 08:45:22 +00:00
if( d == NULL )
{
2018-03-26 16:09:22 +00:00
printlogf( L, "Error", "cannot open dir [%s].", dirname );
2010-10-17 20:26:37 +00:00
return 0;
2010-10-16 18:21:01 +00:00
}
2012-02-15 11:15:32 +00:00
lua_newtable( L );
while( !hup && !term )
{
struct dirent *de = readdir( d );
2010-10-16 18:21:01 +00:00
bool isdir;
2018-03-26 16:09:22 +00:00
// finished?
if( de == NULL ) break;
2010-11-17 18:52:55 +00:00
2012-02-15 11:15:32 +00:00
// ignores . and ..
2018-03-26 16:09:22 +00:00
if( !strcmp( de->d_name, "." ) || !strcmp( de->d_name, ".." ) ) continue;
2010-11-17 18:52:55 +00:00
if( de->d_type == DT_UNKNOWN )
{
2012-02-15 11:15:32 +00:00
// must call stat on some systems :-/
// ( e.g. ReiserFS )
2018-03-26 16:09:22 +00:00
char *entry = s_malloc( strlen( dirname ) + strlen( de->d_name ) + 2 );
2016-05-03 08:45:22 +00:00
2010-10-16 18:21:01 +00:00
struct stat st;
strcpy( entry, dirname );
strcat( entry, "/" );
strcat( entry, de->d_name );
lstat( entry, &st );
isdir = S_ISDIR( st.st_mode );
free( entry );
}
else
{
// otherwise readdir can be trusted
2010-10-16 18:21:01 +00:00
isdir = de->d_type == DT_DIR;
}
2012-02-15 11:15:32 +00:00
// adds this entry to the Lua table
lua_pushstring( L, de->d_name );
lua_pushboolean( L, isdir );
lua_settable( L, -3 );
2010-10-16 18:21:01 +00:00
}
closedir( d );
2016-05-03 08:45:22 +00:00
2010-10-16 18:21:01 +00:00
return 1;
}
2010-10-16 10:26:48 +00:00
/*
| Immediately terminates Lsyncd.
|
| Params on Lua stack:
| 1: exitcode of Lsyncd.
|
| Does not return.
|
*/
2012-02-15 11:15:32 +00:00
int
2016-12-05 14:11:00 +00:00
l_terminate( lua_State *L )
2010-10-18 12:23:46 +00:00
{
int exitcode = luaL_checkinteger( L, 1 );
exit( exitcode );
2010-10-18 12:23:46 +00:00
return 0;
}
/*
| Configures core parameters.
|
| Params on Lua stack:
| 1: a string, configure option
| 2: depends on Param 1
*/
2012-02-15 11:15:32 +00:00
static int
l_configure( lua_State *L )
2010-10-27 19:34:56 +00:00
{
const char * command = luaL_checkstring( L, 1 );
if( !strcmp( command, "running" ) )
{
2018-03-14 08:57:49 +00:00
// set by mantle after first initialize
2012-02-15 11:15:32 +00:00
// from this on log to configurated log end instead of
// stdout/stderr
2011-08-29 09:21:40 +00:00
first_time = false;
if( !settings.log_file )
{
settings.log_syslog = true;
const char * log_ident = settings.log_ident ? settings.log_ident : "lsyncd";
openlog( log_ident, 0, settings.log_facility );
}
logstring( "Normal", "--- Startup ---" );
}
else if( !strcmp( command, "logfile" ) )
{
const char * file = luaL_checkstring( L, 2 );
if( settings.log_file )
{
free( settings.log_file );
}
settings.log_file =
s_strdup( file );
}
else if( !strcmp( command, "logfacility" ) )
{
if( lua_isstring( L, 2 ) )
{
const char * fname = luaL_checkstring( L, 2 );
int i;
for( i = 0; facilitynames[ i ].c_name; i++ )
{
if( !strcasecmp( fname, facilitynames[ i ].c_name ) )
{ break; }
2012-02-15 11:15:32 +00:00
}
if( !facilitynames[ i ].c_name )
{
printlogf(
L, "Error",
"Logging facility '%s' unknown.",
fname
);
exit( -1 );
}
settings.log_facility = facilitynames[ i ].c_val;
}
else if (lua_isnumber(L, 2))
{
settings.log_facility = luaL_checknumber(L, 2);
}
else
{
2018-03-26 16:09:22 +00:00
printlogf( L, "Error", "Logging facility must be a number or string" );
exit( -1 );
}
}
else if( !strcmp( command, "logident" ) )
{
const char * ident = luaL_checkstring( L, 2 );
2018-03-26 16:09:22 +00:00
if( settings.log_ident ) free( settings.log_ident );
settings.log_ident = s_strdup( ident );
}
else
{
printlogf(
L, "Error",
"Internal error, unknown parameter in l_configure( %s )",
command
);
exit( -1 );
2010-10-27 19:34:56 +00:00
}
2010-10-27 19:34:56 +00:00
return 0;
}
/*
| Allows user scripts to observe filedescriptors
|
| Params on Lua stack:
| 1: file descriptor
| 2: function to call when read becomes ready
| 3: function to call when write becomes ready
*/
2012-02-15 11:15:32 +00:00
static int
l_observe_fd( lua_State *L )
2010-12-01 12:19:17 +00:00
{
int fd = luaL_checknumber( L, 1 );
2010-12-01 12:19:17 +00:00
bool ready = false;
bool writey = false;
2012-02-15 11:15:32 +00:00
// Stores the user function in the lua registry.
2018-03-27 15:43:26 +00:00
// It uses the address of the cores ready / writey functions
2012-02-15 11:15:32 +00:00
// for the user as key
if( !lua_isnoneornil( L, 2 ) )
{
lua_pushlightuserdata( L, (void *) user_obs_ready );
lua_gettable( L, LUA_REGISTRYINDEX );
if( lua_isnil( L, -1 ) )
{
2018-03-15 12:43:02 +00:00
lua_pop( L, 1 );
lua_newtable( L );
lua_pushlightuserdata( L, (void *) user_obs_ready );
lua_pushvalue( L, -2 );
lua_settable( L, LUA_REGISTRYINDEX );
2010-12-01 12:19:17 +00:00
}
2018-03-15 12:43:02 +00:00
lua_pushnumber( L, fd );
lua_pushvalue( L, 2 );
lua_settable( L, -3 );
lua_pop( L, 1 );
2010-12-01 12:19:17 +00:00
ready = true;
}
if( !lua_isnoneornil( L, 3 ) )
{
lua_pushlightuserdata( L, (void *) user_obs_writey );
lua_gettable (L, LUA_REGISTRYINDEX );
if( lua_isnil(L, -1) )
{
lua_pop ( L, 1 );
lua_newtable ( L );
lua_pushlightuserdata ( L, (void *) user_obs_writey );
lua_pushvalue ( L, -2 );
lua_settable ( L, LUA_REGISTRYINDEX );
2010-12-01 12:19:17 +00:00
}
lua_pushnumber ( L, fd );
lua_pushvalue ( L, 3 );
lua_settable ( L, -3 );
lua_pop ( L, 1 );
2010-12-01 12:19:17 +00:00
writey = true;
}
// tells the core to watch the fd
observe_fd(
fd,
ready ? user_obs_ready : NULL,
writey ? user_obs_writey : NULL,
user_obs_tidy,
NULL
);
2010-12-01 12:19:17 +00:00
return 0;
}
/*
| Removes a user observance
|
| Params on Lua stack:
| 1: exitcode of Lsyncd.
*/
2010-12-01 12:19:17 +00:00
extern int
l_nonobserve_fd( lua_State *L )
2010-12-01 12:19:17 +00:00
{
int fd = luaL_checknumber( L, 1 );
2010-12-01 12:19:17 +00:00
2012-02-15 11:15:32 +00:00
// removes the read function
lua_pushlightuserdata( L, (void *) user_obs_ready );
lua_gettable( L, LUA_REGISTRYINDEX );
if( !lua_isnil( L, -1 ) )
{
lua_pushnumber ( L, fd );
lua_pushnil ( L );
lua_settable ( L, -2 );
2010-12-01 12:19:17 +00:00
}
lua_pop( L, 1 );
2012-02-15 11:15:32 +00:00
lua_pushlightuserdata( L, (void *) user_obs_writey );
lua_gettable( L, LUA_REGISTRYINDEX );
if ( !lua_isnil( L, -1 ) )
{
lua_pushnumber ( L, fd );
lua_pushnil ( L );
lua_settable ( L, -2 );
2010-12-01 12:19:17 +00:00
}
lua_pop( L, 1 );
2010-12-01 12:19:17 +00:00
nonobserve_fd( fd );
2010-12-01 12:19:17 +00:00
return 0;
}
2016-05-03 08:45:22 +00:00
/*
| The Lsnycd's core library.
*/
static const luaL_Reg corelib[] =
{
{ "configure", l_configure },
{ "exec", l_exec },
{ "log", l_log },
{ "mci", l_mci },
{ "now", l_now },
{ "nonobserve_fd", l_nonobserve_fd },
{ "observe_fd", l_observe_fd },
{ "readdir", l_readdir },
{ "realdir", l_realdir },
{ "stackdump", l_stackdump },
{ "terminate", l_terminate },
{ NULL, NULL }
2010-10-19 20:14:55 +00:00
};
2016-05-03 08:45:22 +00:00
/*
| Registers the Lsyncd's core library.
*/
void
register_core( lua_State *L )
{
2018-03-13 11:29:43 +00:00
lua_newtable( L );
luaL_setfuncs( L, corelib, 0 );
lua_setglobal( L, LSYNCD_CORE_LIBNAME );
2018-03-30 07:36:26 +00:00
register_jiffies( L );
2012-02-15 11:15:32 +00:00
2014-04-29 14:11:27 +00:00
#ifdef WITH_INOTIFY
lua_getglobal( L, LSYNCD_CORE_LIBNAME );
register_inotify( L );
2018-03-13 11:29:43 +00:00
lua_setfield( L, -2, LSYNCD_INOTIFY_LIBNAME );
lua_pop( L, 1 );
#endif
if( lua_gettop( L ) )
{
logstring( "Error", "internal, stack not empty in lsyncd_register( )" );
exit( -1 );
}
}
/*:::::::::::::::.
:: Lsyncd Core
'::::::::::::::::*/
2010-10-19 20:14:55 +00:00
/*
2018-03-14 08:57:49 +00:00
| Pushes a function from the mantle on the stack.
| As well as the callError handler.
*/
extern void
load_mci(
lua_State * L,
const char * name
)
2010-11-10 15:57:37 +00:00
{
printlogf( L, "Call", "%s( )", name );
2012-02-15 11:15:32 +00:00
// pushes the error handler
lua_pushlightuserdata( L, (void *) &callError );
lua_gettable( L, LUA_REGISTRYINDEX );
2012-02-15 11:15:32 +00:00
// pushes the function
lua_pushlightuserdata( L, (void *) &mci );
lua_gettable( L, LUA_REGISTRYINDEX );
lua_pushstring( L, name );
lua_gettable( L, -2 );
lua_remove( L, -2 );
2010-11-10 15:57:37 +00:00
}
2010-10-20 15:34:01 +00:00
2016-05-03 08:45:22 +00:00
/*
| Normal operation happens in here.
*/
2010-11-10 15:57:37 +00:00
static void
2010-10-19 10:12:11 +00:00
masterloop(lua_State *L)
{
while( true )
{
2010-10-24 16:41:58 +00:00
bool have_alarm;
2012-02-15 11:15:32 +00:00
bool force_alarm = false;
2018-03-30 07:36:26 +00:00
clock_t cnow = now( );
2012-02-15 11:15:32 +00:00
clock_t alarm_time = 0;
2010-10-19 10:12:11 +00:00
2012-03-16 15:05:16 +00:00
// memory usage debugging
// lua_gc( L, LUA_GCCOLLECT, 0 );
// printf(
// "gccount: %d\n",
// lua_gc( L, LUA_GCCOUNT, 0 ) * 1024 + lua_gc( L, LUA_GCCOUNTB, 0 ) );
//
2018-03-14 08:57:49 +00:00
// queries the mantle about the soonest alarm
//
load_mci( L, "getAlarm" );
2014-02-28 09:15:48 +00:00
if( lua_pcall( L, 0, 1, -2 ) ) exit( -1 );
if( lua_type( L, -1 ) == LUA_TBOOLEAN)
{
2014-02-28 09:15:48 +00:00
have_alarm = false;
2018-03-27 15:43:26 +00:00
force_alarm = lua_toboolean( L, -1 );
}
else
{
2010-11-08 12:14:10 +00:00
have_alarm = true;
2018-03-27 15:43:26 +00:00
alarm_time = *( ( clock_t * ) luaL_checkudata( L, -1, "Lsyncd.jiffies" ) );
2010-11-05 18:20:33 +00:00
}
2014-02-28 09:15:48 +00:00
lua_pop( L, 2 );
2010-10-19 10:12:11 +00:00
if(
force_alarm ||
2018-03-30 07:36:26 +00:00
( have_alarm && time_before_eq( alarm_time, cnow ) )
)
{
2012-02-15 11:15:32 +00:00
// there is a delay that wants to be handled already thus instead
// of reading/writing from observances it jumps directly to
// handling
2018-03-26 16:09:22 +00:00
// TODO: Actually it might be smarter to handle observances
// anyway. since event queues might overflow.
logstring( "Masterloop", "immediately handling delays." );
}
else
{
// uses select( ) to determine what happens next:
// a) a new event on an observance
// b) an alarm on timeout
// c) the return of a child process
2010-10-25 08:42:27 +00:00
struct timespec tv;
2010-10-19 10:12:11 +00:00
if( have_alarm )
{
2018-03-26 16:09:22 +00:00
// TODO use trunc instead of long conversions
2018-03-30 07:36:26 +00:00
double d = ( ( double )( alarm_time - cnow ) ) / clocks_per_sec;
2010-11-04 13:43:57 +00:00
tv.tv_sec = d;
2018-03-27 15:43:26 +00:00
tv.tv_nsec = ( ( d - ( long ) d ) ) * 1000000000.0;
printlogf(
L, "Masterloop",
"going into select ( timeout %f seconds )",
d
);
}
else
{
logstring( "Masterloop", "going into select ( no timeout )" );
2010-10-19 10:12:11 +00:00
}
2018-03-28 06:59:10 +00:00
observe_select( L, have_alarm ? &tv : NULL );
2012-02-15 11:15:32 +00:00
}
2018-03-28 06:59:10 +00:00
// collects possibly zombified child processes
while( 1 )
{
2010-10-23 19:19:46 +00:00
int status;
pid_t pid = waitpid( 0, &status, WNOHANG );
// no more zombies
if( pid <= 0 ) break;
2018-03-14 08:57:49 +00:00
// calls the mantle to handle the collection
load_mci( L, "collectProcess" );
lua_pushinteger( L, pid );
lua_pushinteger( L, WEXITSTATUS( status ) );
if( lua_pcall( L, 2, 0, -4 ) ) exit(-1);
lua_pop( L, 1 );
2012-02-15 11:15:32 +00:00
}
2010-10-23 19:19:46 +00:00
// reacts on HUP signals
if( hup )
{
load_mci( L, "hup" );
if( lua_pcall( L, 0, 0, -2 ) ) exit( -1 );
lua_pop( L, 1 );
2010-11-14 09:11:09 +00:00
hup = 0;
}
// reacts on TERM and INT signals
if( term == 1 )
{
load_mci( L, "term" );
lua_pushnumber( L, sigcode );
if( lua_pcall( L, 1, 0, -3 ) ) exit( -1 );
lua_pop( L, 1 );
2010-11-14 09:11:09 +00:00
term = 2;
}
2018-03-14 08:57:49 +00:00
// lets the mantle do stuff every cycle,
2012-02-15 11:15:32 +00:00
// like starting new processes, writing the statusfile etc.
load_mci( L, "cycle" );
2018-03-30 07:36:26 +00:00
l_now( L );
2018-03-30 07:36:26 +00:00
if( lua_pcall( L, 1, 1, -3 ) ) exit( -1 );
if( !lua_toboolean( L, -1 ) )
{
2012-02-15 11:15:32 +00:00
// cycle told core to break mainloop
lua_pop( L, 2 );
2010-11-14 09:11:09 +00:00
return;
}
lua_pop( L, 2 );
if( lua_gettop( L ) )
{
logstring( "Error", "internal, stack is dirty." );
l_stackdump( L );
exit( -1 );
2010-12-01 13:17:04 +00:00
}
2010-10-19 10:12:11 +00:00
}
}
2016-05-03 08:45:22 +00:00
/*
| The effective main for one run.
2014-02-28 09:15:48 +00:00
|
| HUP signals may cause several runs of the one main.
*/
2010-10-17 15:24:55 +00:00
int
main1( int argc, char *argv[] )
2010-10-14 13:52:01 +00:00
{
2012-02-15 11:15:32 +00:00
// the Lua interpreter
lua_State * L;
2010-10-27 19:34:56 +00:00
2010-11-03 21:02:14 +00:00
int argp = 1;
2012-02-15 11:15:32 +00:00
// load Lua
L = luaL_newstate( );
luaL_openlibs( L );
2018-03-14 08:57:49 +00:00
2010-11-11 19:52:20 +00:00
{
2012-02-15 11:15:32 +00:00
// checks the lua version
const char * version;
2010-11-11 19:52:20 +00:00
int major, minor;
lua_getglobal( L, "_VERSION" );
version = luaL_checkstring( L, -1 );
if( sscanf( version, "Lua %d.%d", &major, &minor ) != 2 )
{
fprintf( stderr, "cannot parse lua library version!\n" );
exit (-1 );
2010-11-11 19:52:20 +00:00
}
if( major < 5 || ( major == 5 && minor < 2 ) )
{
fprintf( stderr, "Lua library is too old. Needs 5.2 at least" );
exit( -1 );
2010-11-11 19:52:20 +00:00
}
lua_pop( L, 1 );
2010-11-11 19:52:20 +00:00
}
2010-11-03 21:02:14 +00:00
2010-11-03 16:04:11 +00:00
{
// logging is prepared quite early
2010-11-03 21:02:14 +00:00
int i = 1;
add_logcat( "Normal", LOG_NOTICE );
add_logcat( "Warn", LOG_WARNING );
add_logcat( "Error", LOG_ERR );
while( i < argc )
{
if( strcmp( argv[ i ], "-log" ) && strcmp( argv[ i ], "--log" ) )
{
// arg is neither -log or --log
i++;
continue;
2010-11-03 16:04:11 +00:00
}
2012-02-15 12:47:58 +00:00
// -(-)log was last argument
if( ++i >= argc ) break;
2012-02-15 12:47:58 +00:00
if( !add_logcat( argv[ i ], LOG_NOTICE ) )
{
printlogf(
L, "Error",
"'%s' is not a valid logging category",
argv[ i ]
);
exit( -1 );
2010-11-03 16:04:11 +00:00
}
}
}
2010-11-03 14:54:33 +00:00
// registers Lsycnd's core library
register_core( L );
if( check_logcat( "Debug" ) <= settings.log_level )
{
2012-02-15 12:47:58 +00:00
// printlogf doesnt support %ld :-(
2018-03-12 16:03:56 +00:00
printf( "kernels clocks_per_sec=%ld\n", clocks_per_sec );
2010-11-04 13:43:57 +00:00
}
2018-03-23 16:36:51 +00:00
// loads the lsyncd mantle
if( luaL_loadbuffer( L, mantle_out, mantle_size, "mantle" ) )
{
2018-03-23 16:36:51 +00:00
printlogf( L, "Error", "loading mantle: %s", lua_tostring( L, -1 ) );
exit( -1 );
2010-11-03 21:02:14 +00:00
}
2018-03-12 16:03:56 +00:00
// prepares the luacode executing the script
if( lua_pcall( L, 0, 0, 0 ) )
2010-11-10 15:57:37 +00:00
{
2018-03-23 16:36:51 +00:00
printlogf( L, "Error", "preparing mantle: %s", lua_tostring( L, -1 ) );
exit( -1 );
2010-10-17 17:13:53 +00:00
}
2010-10-18 17:09:59 +00:00
{
2018-03-15 12:43:02 +00:00
// asserts the Lsyncd's version matches
// double checks the if mantle version is the same as core version
2010-10-18 17:09:59 +00:00
const char *lversion;
2018-03-15 12:43:02 +00:00
lua_getglobal( L, "lsyncd_version" );
lversion = luaL_checkstring( L, -1 );
if( strcmp( lversion, PACKAGE_VERSION ) )
{
printlogf(
L, "Error",
2018-03-12 16:03:56 +00:00
"Version mismatch luacode is '%s', but core is '%s'",
lversion, PACKAGE_VERSION
);
exit( -1 );
2010-10-18 17:09:59 +00:00
}
lua_pop( L, 1 );
2010-10-18 17:09:59 +00:00
}
2018-03-23 16:36:51 +00:00
// loads the default sync implementations
if( luaL_loadbuffer( L, default_out, default_size, "default" ) )
{
printlogf( L, "Error",
"loading default sync implementations: %s", lua_tostring( L, -1 ) );
exit( -1 );
}
// loads the user enivornment
// the default sync implementations are actually not priviledged in any way
lua_getglobal( L, "userENV" );
lua_setupvalue( L, -2, 1 );
// prepares the default sync implementations
if( lua_pcall( L, 0, 0, 0 ) )
{
printlogf( L, "Error",
"preparing default sync implementations: %s", lua_tostring( L, -1 ) );
exit( -1 );
}
2018-03-15 12:43:02 +00:00
// checks if there is a "-help" or "--help"
2010-10-27 09:06:13 +00:00
{
int i;
for( i = argp; i < argc; i++ )
{
if ( !strcmp( argv[ i ], "-help" ) || !strcmp( argv[ i ], "--help" ) )
{
load_mci( L, "help" );
if( lua_pcall( L, 0, 0, -2 ) ) exit( -1 );
lua_pop( L, 1 );
exit( 0 );
2010-10-27 09:06:13 +00:00
}
}
}
// starts the option parser in Lua script
2010-11-03 21:02:14 +00:00
{
int idx = 1;
const char *s;
2012-02-15 11:15:32 +00:00
// creates a table with all remaining argv option arguments
load_mci( L, "configure" );
lua_newtable( L );
while( argp < argc )
{
lua_pushnumber( L, idx++ );
lua_pushstring( L, argv[ argp++ ] );
lua_settable( L, -3 );
2010-11-03 21:02:14 +00:00
}
2012-02-15 11:15:32 +00:00
// creates a table with the cores event monitor interfaces
2010-11-28 09:37:43 +00:00
idx = 0;
lua_newtable( L );
while( monitors[ idx ] )
{
lua_pushnumber( L, idx + 1 );
lua_pushstring( L, monitors[ idx++ ] );
lua_settable( L, -3 );
2010-11-04 13:52:34 +00:00
}
if( lua_pcall( L, 2, 1, -4 ) ) exit( -1 );
if( first_time )
{
// If not first time, simply retains the config file given
s = lua_tostring(L, -1);
if( s ) lsyncd_config_file = s_strdup( s );
2010-11-03 21:02:14 +00:00
}
lua_pop( L, 2 );
2010-11-03 21:02:14 +00:00
}
// checks existence of the config file
if( lsyncd_config_file )
{
2010-10-27 09:06:13 +00:00
struct stat st;
// gets the absolute path to the config file
// so in case of HUPing the daemon, it finds it again
char * apath = get_realpath( lsyncd_config_file );
if( !apath )
{
printlogf( L, "Error", "Cannot find config file at '%s'.", lsyncd_config_file );
exit( -1 );
}
free( lsyncd_config_file );
lsyncd_config_file = apath;
if( stat( lsyncd_config_file, &st ) )
{
printlogf( L, "Error", "Cannot find config file at '%s'.", lsyncd_config_file );
exit( -1 );
2010-10-27 09:06:13 +00:00
}
2012-02-15 11:15:32 +00:00
// loads and executes the config file
if( luaL_loadfile( L, lsyncd_config_file ) )
{
printlogf(
L, "Error",
"error loading %s: %s", lsyncd_config_file, lua_tostring( L, -1 )
);
exit( -1 );
2010-11-03 21:02:14 +00:00
}
2018-03-15 12:43:02 +00:00
// loads the user enivornment
2018-03-15 16:11:45 +00:00
lua_getglobal( L, "userENV" );
2018-03-15 12:43:02 +00:00
lua_setupvalue( L, -2, 1 );
if( lua_pcall( L, 0, LUA_MULTRET, 0) )
{
printlogf(
L, "Error",
"error preparing %s: %s", lsyncd_config_file, lua_tostring( L, -1 )
);
exit( -1 );
2010-11-03 21:02:14 +00:00
}
2010-10-17 17:13:53 +00:00
}
2010-10-16 10:26:48 +00:00
2014-04-29 14:11:27 +00:00
#ifdef WITH_INOTIFY
open_inotify( L );
#endif
// adds signal handlers
// listens to SIGCHLD, but blocks it until pselect( )
// opens the signal handler up
2010-10-25 08:42:27 +00:00
{
sigset_t set;
sigemptyset( &set );
sigaddset( &set, SIGCHLD );
signal( SIGCHLD, sig_child );
sigprocmask( SIG_BLOCK, &set, NULL );
2012-02-15 11:15:32 +00:00
signal( SIGHUP, sig_handler );
signal( SIGTERM, sig_handler );
signal( SIGINT, sig_handler );
2010-10-25 08:42:27 +00:00
}
2010-10-24 21:35:29 +00:00
2018-03-14 08:57:49 +00:00
// runs initializations from mantle
// it will set the configuration and add watches
2010-10-25 21:04:28 +00:00
{
load_mci( L, "initialize" );
lua_pushboolean( L, first_time );
if( lua_pcall( L, 1, 0, -3 ) ) exit( -1 );
lua_pop( L, 1 );
2010-10-25 21:04:28 +00:00
}
2010-10-18 17:09:59 +00:00
//
// enters the master loop
//
masterloop( L );
2010-10-18 17:09:59 +00:00
2012-02-15 11:15:32 +00:00
// cleanup
2018-03-28 06:59:10 +00:00
observe_tidy_all( );
// frees logging categories
2018-03-23 08:34:33 +00:00
log_free( );
2010-11-14 09:11:09 +00:00
lua_close( L );
2010-10-14 13:52:01 +00:00
return 0;
}
2010-11-14 09:11:09 +00:00
2016-05-03 08:45:22 +00:00
/*
| Main
*/
2010-11-14 09:11:09 +00:00
int
main( int argc, char * argv[ ] )
2010-11-14 09:11:09 +00:00
{
2012-02-15 11:15:32 +00:00
// gets a kernel parameter
clocks_per_sec = sysconf( _SC_CLK_TCK );
2010-11-14 09:11:09 +00:00
2018-03-09 09:03:08 +00:00
setlinebuf( stdout );
setlinebuf( stderr );
while( !term )
{
main1( argc, argv );
}
// exits with error code responding to the signal it died for
// FIXME this no longer holds true to systemd recommendations
return 128 + sigcode;
2010-11-14 09:11:09 +00:00
}