|
Packit |
8f70b4 |
/*
|
|
Packit |
8f70b4 |
* lftp - file transfer program
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* Copyright (c) 1996-2017 by Alexander V. Lukyanov (lav@yars.free.net)
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* This program is free software; you can redistribute it and/or modify
|
|
Packit |
8f70b4 |
* it under the terms of the GNU General Public License as published by
|
|
Packit |
8f70b4 |
* the Free Software Foundation; either version 3 of the License, or
|
|
Packit |
8f70b4 |
* (at your option) any later version.
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* This program is distributed in the hope that it will be useful,
|
|
Packit |
8f70b4 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
8f70b4 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
Packit |
8f70b4 |
* GNU General Public License for more details.
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* You should have received a copy of the GNU General Public License
|
|
Packit |
8f70b4 |
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
Packit |
8f70b4 |
*/
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
#include <config.h>
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
#include <stdarg.h>
|
|
Packit |
8f70b4 |
#include <fcntl.h>
|
|
Packit |
8f70b4 |
#include <sys/types.h>
|
|
Packit |
8f70b4 |
#include <sys/stat.h>
|
|
Packit |
8f70b4 |
#include "xstring.h"
|
|
Packit |
8f70b4 |
#include "log.h"
|
|
Packit |
8f70b4 |
#include "SMTask.h"
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
Ref<Log> Log::global;
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
static ResType log_vars[] = {
|
|
Packit |
8f70b4 |
{"log:enabled", "no", ResMgr::BoolValidate},
|
|
Packit |
8f70b4 |
{"log:level", "9", ResMgr::UNumberValidate},
|
|
Packit |
8f70b4 |
{"log:show-time", "no", ResMgr::BoolValidate},
|
|
Packit |
8f70b4 |
{"log:show-pid", "no", ResMgr::BoolValidate},
|
|
Packit |
8f70b4 |
{"log:show-ctx", "no", ResMgr::BoolValidate},
|
|
Packit |
8f70b4 |
{"log:file", "", ResMgr::FileCreatable},
|
|
Packit |
8f70b4 |
{"log:max-size", "1M", ResMgr::UNumberValidate},
|
|
Packit |
8f70b4 |
{"log:prefix-recv","<--- "},
|
|
Packit |
8f70b4 |
{"log:prefix-send","---> "},
|
|
Packit |
8f70b4 |
{"log:prefix-note","---- "},
|
|
Packit |
8f70b4 |
{"log:prefix-error","**** "},
|
|
Packit |
8f70b4 |
{0}
|
|
Packit |
8f70b4 |
};
|
|
Packit |
8f70b4 |
static ResDecls log_vars_register(log_vars);
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
Log::Log(const char *name)
|
|
Packit |
8f70b4 |
: name(name)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
output=-1;
|
|
Packit |
8f70b4 |
need_close_output=false;
|
|
Packit |
8f70b4 |
tty_cb=0;
|
|
Packit |
8f70b4 |
enabled=false;
|
|
Packit |
8f70b4 |
level=0;
|
|
Packit |
8f70b4 |
tty=false;
|
|
Packit |
8f70b4 |
show_pid=true;
|
|
Packit |
8f70b4 |
show_time=true;
|
|
Packit |
8f70b4 |
show_context=true;
|
|
Packit |
8f70b4 |
at_line_start=true;
|
|
Packit |
8f70b4 |
Reconfig(0);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
bool Log::WillOutput(int l)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
if(!enabled || l>level || output==-1)
|
|
Packit |
8f70b4 |
return false;
|
|
Packit |
8f70b4 |
if(tty)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
pid_t pg=tcgetpgrp(output);
|
|
Packit |
8f70b4 |
if(pg!=(pid_t)-1 && pg!=getpgrp())
|
|
Packit |
8f70b4 |
return false;
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
return true;
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
void Log::Write(int l,const char *s,int len)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
if(!WillOutput(l))
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
DoWrite(s,len);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
void Log::DoWrite(const char *s,int len)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
if(len==0)
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
if(buf.length()==0 || buf.last_char()=='\n') {
|
|
Packit |
8f70b4 |
if(show_pid)
|
|
Packit |
8f70b4 |
buf.appendf("[%ld] ",(long)getpid());
|
|
Packit |
8f70b4 |
if(show_time)
|
|
Packit |
8f70b4 |
buf.append(SMTask::now.IsoDateTime()).append(' ');
|
|
Packit |
8f70b4 |
if(show_context)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
const char *ctx=SMTask::GetCurrentLogContext();
|
|
Packit |
8f70b4 |
if(ctx)
|
|
Packit |
8f70b4 |
buf.append(ctx).append(' ');
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
buf.append(s,len);
|
|
Packit |
8f70b4 |
if(buf.last_char()!='\n')
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
if(tty_cb && tty)
|
|
Packit |
8f70b4 |
tty_cb();
|
|
Packit |
8f70b4 |
int res=write(output,buf,buf.length());
|
|
Packit |
8f70b4 |
if(res==-1) {
|
|
Packit |
8f70b4 |
if(E_RETRY(errno))
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
ResMgr::Set("log:enabled",name,"no");
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
if(res==(int)buf.length())
|
|
Packit |
8f70b4 |
buf.truncate();
|
|
Packit |
8f70b4 |
else
|
|
Packit |
8f70b4 |
buf.set_substr(0,res,"",0);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
void Log::Format(int l,const char *f,...)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
if(!WillOutput(l))
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
va_list v;
|
|
Packit |
8f70b4 |
va_start(v,f);
|
|
Packit |
8f70b4 |
DoWrite(xstring::vformat(f,v));
|
|
Packit |
8f70b4 |
va_end(v);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
void Log::vFormat(int l,const char *f,va_list v)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
if(!WillOutput(l))
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
DoWrite(xstring::vformat(f,v));
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
void Log::Cleanup()
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
global=0;
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
Log::~Log()
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
CloseOutput();
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
void Log::SetOutput(int o,bool need_close)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
CloseOutput();
|
|
Packit |
8f70b4 |
output=o;
|
|
Packit |
8f70b4 |
need_close_output=need_close;
|
|
Packit |
8f70b4 |
if(output!=-1)
|
|
Packit |
8f70b4 |
tty=isatty(output);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
void Log::Reconfig(const char *n) {
|
|
Packit |
8f70b4 |
enabled=QueryBool("log:enabled");
|
|
Packit |
8f70b4 |
level=Query("log:level");
|
|
Packit |
8f70b4 |
show_time=QueryBool("log:show-time");
|
|
Packit |
8f70b4 |
show_pid=QueryBool("log:show-pid");
|
|
Packit |
8f70b4 |
show_context=QueryBool("log:show-ctx");
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
if(!n || !strcmp(n,"log:file")) {
|
|
Packit |
8f70b4 |
const char *file=Query("log:file");
|
|
Packit |
8f70b4 |
int fd=2;
|
|
Packit |
8f70b4 |
bool need_close_fd=false;
|
|
Packit |
8f70b4 |
if(file && *file) {
|
|
Packit |
8f70b4 |
struct stat st;
|
|
Packit |
8f70b4 |
if(stat(file,&st)!=-1) {
|
|
Packit |
8f70b4 |
if(st.st_size > long(Query("log:max-size"))) {
|
|
Packit |
8f70b4 |
debug((9,"rotating log %s\n",file));
|
|
Packit |
8f70b4 |
if(rename(file,xstring::cat(file,".old",NULL))==-1)
|
|
Packit |
8f70b4 |
debug((1,"rename(%s): %s\n",file,strerror(errno)));
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
fd=open(file,O_WRONLY|O_CREAT|O_APPEND|O_NONBLOCK,0600);
|
|
Packit |
8f70b4 |
if(fd==-1) {
|
|
Packit |
8f70b4 |
perror(file);
|
|
Packit |
8f70b4 |
fd=2;
|
|
Packit |
8f70b4 |
} else {
|
|
Packit |
8f70b4 |
need_close_fd=true;
|
|
Packit |
8f70b4 |
fcntl(fd,F_SETFD,FD_CLOEXEC);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
if(fd!=output)
|
|
Packit |
8f70b4 |
SetOutput(fd,need_close_fd);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
}
|