156 lines
4 KiB
C++
156 lines
4 KiB
C++
/* Tarlz - Archiver with multimember lzip compression
|
|
Copyright (C) 2013-2020 Antonio Diaz Diaz.
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#define _FILE_OFFSET_BITS 64
|
|
|
|
#include <cctype>
|
|
#include <cerrno>
|
|
#include <climits>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <pthread.h>
|
|
#include <stdint.h>
|
|
#include <unistd.h>
|
|
|
|
#include "arg_parser.h"
|
|
#include "tarlz.h"
|
|
|
|
|
|
void xinit_mutex( pthread_mutex_t * const mutex )
|
|
{
|
|
const int errcode = pthread_mutex_init( mutex, 0 );
|
|
if( errcode )
|
|
{ show_error( "pthread_mutex_init", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
void xinit_cond( pthread_cond_t * const cond )
|
|
{
|
|
const int errcode = pthread_cond_init( cond, 0 );
|
|
if( errcode )
|
|
{ show_error( "pthread_cond_init", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
|
|
void xdestroy_mutex( pthread_mutex_t * const mutex )
|
|
{
|
|
const int errcode = pthread_mutex_destroy( mutex );
|
|
if( errcode )
|
|
{ show_error( "pthread_mutex_destroy", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
void xdestroy_cond( pthread_cond_t * const cond )
|
|
{
|
|
const int errcode = pthread_cond_destroy( cond );
|
|
if( errcode )
|
|
{ show_error( "pthread_cond_destroy", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
|
|
void xlock( pthread_mutex_t * const mutex )
|
|
{
|
|
const int errcode = pthread_mutex_lock( mutex );
|
|
if( errcode )
|
|
{ show_error( "pthread_mutex_lock", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
|
|
void xunlock( pthread_mutex_t * const mutex )
|
|
{
|
|
const int errcode = pthread_mutex_unlock( mutex );
|
|
if( errcode )
|
|
{ show_error( "pthread_mutex_unlock", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
|
|
void xwait( pthread_cond_t * const cond, pthread_mutex_t * const mutex )
|
|
{
|
|
const int errcode = pthread_cond_wait( cond, mutex );
|
|
if( errcode )
|
|
{ show_error( "pthread_cond_wait", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
|
|
void xsignal( pthread_cond_t * const cond )
|
|
{
|
|
const int errcode = pthread_cond_signal( cond );
|
|
if( errcode )
|
|
{ show_error( "pthread_cond_signal", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
|
|
void xbroadcast( pthread_cond_t * const cond )
|
|
{
|
|
const int errcode = pthread_cond_broadcast( cond );
|
|
if( errcode )
|
|
{ show_error( "pthread_cond_broadcast", errcode ); cleanup_and_fail(); }
|
|
}
|
|
|
|
|
|
unsigned long long parse_octal( const uint8_t * const ptr, const int size )
|
|
{
|
|
unsigned long long result = 0;
|
|
int i = 0;
|
|
while( i < size && std::isspace( ptr[i] ) ) ++i;
|
|
for( ; i < size && ptr[i] >= '0' && ptr[i] <= '7'; ++i )
|
|
{ result <<= 3; result += ptr[i] - '0'; }
|
|
return result;
|
|
}
|
|
|
|
|
|
/* Returns the number of bytes really read.
|
|
If (returned value < size) and (errno == 0), means EOF was reached.
|
|
*/
|
|
int readblock( const int fd, uint8_t * const buf, const int size )
|
|
{
|
|
int sz = 0;
|
|
errno = 0;
|
|
while( sz < size )
|
|
{
|
|
const int n = read( fd, buf + sz, size - sz );
|
|
if( n > 0 ) sz += n;
|
|
else if( n == 0 ) break; // EOF
|
|
else if( errno != EINTR ) break;
|
|
errno = 0;
|
|
}
|
|
return sz;
|
|
}
|
|
|
|
|
|
/* Returns the number of bytes really written.
|
|
If (returned value < size), it is always an error.
|
|
*/
|
|
int writeblock( const int fd, const uint8_t * const buf, const int size )
|
|
{
|
|
int sz = 0;
|
|
errno = 0;
|
|
while( sz < size )
|
|
{
|
|
const int n = write( fd, buf + sz, size - sz );
|
|
if( n > 0 ) sz += n;
|
|
else if( n < 0 && errno != EINTR ) break;
|
|
errno = 0;
|
|
}
|
|
return sz;
|
|
}
|
|
|
|
|
|
bool nonempty_arg( const Arg_parser & parser, const int i )
|
|
{
|
|
return ( parser.code( i ) == 0 && !parser.argument( i ).empty() );
|
|
}
|