182 lines
6.1 KiB
Perl
Executable file
182 lines
6.1 KiB
Perl
Executable file
#!/usr/bin/perl -w
|
|
|
|
use Proc::Daemon;
|
|
use DBI;
|
|
use Digest::SHA;
|
|
use File::Copy;
|
|
|
|
$database = 'gnutube';
|
|
$dbhost = 'localhost';
|
|
$dbuser = 'root';
|
|
$dbpass = '';
|
|
$gnutube_root = '/var/www/gnutube';
|
|
|
|
#TODO: deamonize by uncommenting this line
|
|
#Proc::Daemon::Init;
|
|
|
|
$LOG = "/var/www/gnutube/daemon.log";
|
|
|
|
|
|
sub appendlog
|
|
{
|
|
if (open(FILE, ">>$LOG"))
|
|
{
|
|
print FILE scalar(localtime)." ".$$." ".join(" ",@_)."\n";
|
|
close FILE;
|
|
}
|
|
}
|
|
|
|
sub interrupt
|
|
{
|
|
appendlog(@_);
|
|
die;
|
|
}
|
|
|
|
$dbh = DBI->connect("DBI:mysql:$database:$dbhost", $dbuser, $dbpass) or interrupt "could not connect to db";
|
|
|
|
#video status:
|
|
# 0 - new entry - nothing done yet
|
|
# 1 - valid public video
|
|
# 2 - error: invalid audio and/or video stream
|
|
# 3 - error: file not found
|
|
# 4 - error: file is not a video
|
|
# 5 - error: video is a duplicate
|
|
|
|
while(1)
|
|
{
|
|
#get fresh video id from db
|
|
my $sth = $dbh->prepare(qq{select id, title, description, userid, timestamp from uploaded where status = 0 limit 1}) or interrupt $dbh->errstr;
|
|
$sth->execute() or interrupt $dbh->errstr;
|
|
my ($id, $title, $description, $userid, $timestamp) = $sth->fetchrow_array();
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
|
|
if($id)
|
|
{
|
|
$info = `ffplay -stats -an -vn -nodisp $gnutube_root/tmp/$id 2>&1`;
|
|
|
|
if($info =~ /ignoring/)
|
|
{
|
|
appendlog $id, "invalid stream";
|
|
|
|
#write status 2 to uploaded table
|
|
$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
|
|
$sth->execute(2, $id) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
unlink "$gnutube_root/tmp/$id";
|
|
}
|
|
elsif ($info =~ /I\/O error occured/)
|
|
{
|
|
appendlog $id, "file not found";
|
|
|
|
#write status 3 to uploaded table
|
|
$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
|
|
$sth->execute(3, $id) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
unlink "$gnutube_root/tmp/$id";
|
|
}
|
|
elsif ($info =~ /Unknown format/ or $info =~ /could not find codec parameters/)
|
|
{
|
|
appendlog $id, "file is no video";
|
|
|
|
#write status 4 to uploaded table
|
|
$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
|
|
$sth->execute(4, $id) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
unlink "$gnutube_root/tmp/$id";
|
|
}
|
|
else
|
|
{
|
|
$sha = new Digest::SHA(256);
|
|
$sha->addfile("$gnutube_root/tmp/$id");
|
|
$sha = $sha->hexdigest;
|
|
|
|
#check if this hash is already in database
|
|
my $sth = $dbh->prepare(qq{select id from videos where hash = ? limit 1}) or interrupt $dbh->errstr;
|
|
$sth->execute($sha) or interrupt $dbh->errstr;
|
|
my ($resultid) = $sth->fetchrow_array();
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
|
|
#if so, then video is a duplicate
|
|
if($resultid)
|
|
{
|
|
appendlog "$id, video already uploaded: $resultid";
|
|
|
|
#write status 5 to uploaded table
|
|
$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
|
|
$sth->execute(5, $id) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
unlink "$gnutube_root/tmp/$id";
|
|
}
|
|
else
|
|
{
|
|
($container, $duration) = $info =~ /Input \#0, (\w+),.+?\n.+?Duration: (\d{2}:\d{2}:\d{2}\.\d)/;
|
|
|
|
#these two regexes have to be applied seperately because nobody knows which stream (audio or video) comes first
|
|
($audio) = $info =~ /Audio: (\w+)/;
|
|
($video, $width, $height, $fps) = $info =~ /Video: (\w+),.+?(\d+)x(\d+),.+?(\d+\.\d+) fps/;
|
|
|
|
if(!$audio or !$video or !$duration)
|
|
{
|
|
appendlog $id, "a stream is missing or video is corrupt";
|
|
|
|
#write status 2 to uploaded table
|
|
$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
|
|
$sth->execute(2, $id) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
unlink "$gnutube_root/tmp/$id";
|
|
}
|
|
else
|
|
{
|
|
#TODO: maybe delete entry from uploaded table after successful upload?
|
|
$filesize = -s "$gnutube_root/tmp/$id";
|
|
|
|
#check if the upload already is in the right format
|
|
if ($container eq 'ogg' and $video eq 'theora' and $audio eq 'vorbis')
|
|
{
|
|
appendlog $id, "file already is ogg-theora/vorbis";
|
|
|
|
#write status 1 to uploaded table
|
|
$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
|
|
$sth->execute(1, $id) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
|
|
#add video to videos table
|
|
$sth = $dbh->prepare(qq{insert into videos (id, title, description, userid, timestamp,
|
|
hash, filesize, duration, width, height, fps)
|
|
values (?, ?, ?, ?, ?, ?, ?, time_to_sec( ? ), ?, ?, ?)}) or interrupt $dbh->errstr;
|
|
$sth->execute($id, $title, $description, $userid, $timestamp, $sha, $filesize, $duration, $width, $height, $fps) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
|
|
#move video
|
|
move "$gnutube_root/tmp/$id", "$gnutube_root/videos/$id";
|
|
}
|
|
else #encode video
|
|
{
|
|
#FIXME: remove endtime - this is for testing only
|
|
system "ffmpeg2theora --optimize --videobitrate 1000 --audiobitrate 64 --sharpness 0 --endtime 10 --output $gnutube_root/videos/$id $gnutube_root/tmp/$id 2>&1";
|
|
appendlog $id, $audio, $video, $width, $height, $fps, $duration, $sha;
|
|
|
|
#write status 1 to uploaded table
|
|
$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
|
|
$sth->execute(1, $id) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
|
|
#add video to videos table
|
|
$sth = $dbh->prepare(qq{insert into videos (id, title, description, userid, timestamp,
|
|
hash, filesize, duration, width, height, fps)
|
|
values (?, ?, ?, ?, ?, ?, ?, time_to_sec( ? ), ?, ?, ?)}) or interrupt $dbh->errstr;
|
|
$sth->execute($id, $title, $description, $userid, $timestamp, $sha, $filesize, $duration, $width, $height, $fps) or interrupt $dbh->errstr;
|
|
$sth->finish() or interrupt $dbh->errstr;
|
|
|
|
#delete temp file
|
|
unlink "$gnutube_root/tmp/$id";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
sleep 10;
|
|
}
|
|
}
|