/********************************************************************** Copyright 2019 Andrey V.Kosteltsev Licensed under the Radix cross Linux License, Version 1.0 . you may not use this file except in compliance with the License. You may obtain a copy of the License at https://radix-linux.su/licenses/LICENSE-1.0-en_US.txt Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. **********************************************************************/ #include <config.h> #include <stdlib.h> #include <stdio.h> #include <sys/types.h> #include <stdint.h> #include <dirent.h> #include <sys/stat.h> /* chmod(2) */ #include <sys/file.h> /* flock(2) */ #include <fcntl.h> #include <linux/limits.h> #include <alloca.h> /* alloca(3) */ #include <string.h> /* strdup(3) */ #include <strings.h> /* index(3) */ #include <libgen.h> /* basename(3) */ #include <ctype.h> /* tolower(3) */ #include <errno.h> #include <time.h> #include <sys/time.h> #include <pwd.h> #include <grp.h> #include <stdarg.h> #include <unistd.h> #include <sys/resource.h> #include <signal.h> #if !defined SIGCHLD && defined SIGCLD # define SIGCHLD SIGCLD #endif #define _GNU_SOURCE #include <getopt.h> #include <msglog.h> #include <wrapper.h> #include <system.h> #define PROGRAM_NAME "chrefs" #include <defs.h> char *program = PROGRAM_NAME; char *destination = NULL, *operation = NULL, *pkglog_fname = NULL, *tmpdir = NULL, *requires_fname = NULL; int exit_status = EXIT_SUCCESS; /* errors counter */ char *selfdir = NULL; char *pkgname = NULL, *pkgver = NULL, *arch = NULL, *distroname = NULL, *distrover = NULL, *group = NULL; /******************************************** *requires[] declarations: */ struct package { char *name; char *version; char *arch; char *distro_name; char *distro_version; }; static struct package *create_package( char *name, char *version, char *arch, char *distro_name, char *distro_version ); static void free_package( struct package *pkg ); static struct package **create_requires( size_t size ); static void free_requires( struct package **requires ); /* End of *requires[] declarations. ********************************************/ struct package **requires = NULL; /******************************************** LOCK FILE declarations: */ static int __lock_file( FILE *fp ); static void __unlock_file( int fd ); /* End of LOCK FILE declarations. ********************************************/ #define FREE_PKGINFO_VARIABLES() \ if( pkgname ) { free( pkgname ); } pkgname = NULL; \ if( pkgver ) { free( pkgver ); } pkgver = NULL; \ if( arch ) { free( arch ); } arch = NULL; \ if( distroname ) { free( distroname ); } distroname = NULL; \ if( distrover ) { free( distrover ); } distrover = NULL; \ if( group ) { free( group ); } group = NULL; \ if( requires ) { free_requires( requires ); } requires = NULL void free_resources() { if( selfdir ) { free( selfdir ); selfdir = NULL; } if( destination ) { free( destination ); destination = NULL; } if( operation ) { free( operation ); operation = NULL; } if( pkglog_fname ) { free( pkglog_fname ); pkglog_fname = NULL; } if( requires_fname ) { free( requires_fname ); requires_fname = NULL; } FREE_PKGINFO_VARIABLES(); } void usage() { free_resources(); fprintf( stdout, "\n" ); fprintf( stdout, "Usage: %s [options] <pkglog>\n", program ); fprintf( stdout, "\n" ); fprintf( stdout, "Increment or Decrement reference counters in required PKGLOG files\n" ); fprintf( stdout, "in the Setup Database packages directory, which is determined by\n" ); fprintf( stdout, "option --destination OR by the directory where the input <pkglog>\n" ); fprintf( stdout, "file is located. If destination is defined then <pkglog> file name\n" ); fprintf( stdout, "should be defined relative to destination.\n" ); fprintf( stdout, "\n" ); fprintf( stdout, "Options:\n" ); fprintf( stdout, " -h,--help Display this information.\n" ); fprintf( stdout, " -v,--version Display the version of %s utility.\n", program ); fprintf( stdout, " -d,--destination=<DIR> Setup Database packages directory.\n" ); fprintf( stdout, " -o,--operation=<inc|dec> Operation: Increment or Decrement.\n" ); fprintf( stdout, "\n" ); fprintf( stdout, "Parameter:\n" ); fprintf( stdout, " <pkglog> Input PKGLOG file (TEXT format).\n" ); fprintf( stdout, "\n" ); exit( EXIT_FAILURE ); } void to_lowercase( char *s ) { char *p = s; while( p && *p ) { int c = *p; *p = tolower( c ); ++p; } } void to_uppercase( char *s ) { char *p = s; while( p && *p ) { int c = *p; *p = toupper( c ); ++p; } } void version() { char *upper = NULL; upper = (char *)alloca( strlen( program ) + 1 ); strcpy( (char *)upper, (const char *)program ); to_uppercase( upper ); fprintf( stdout, "%s (%s) %s\n", program, upper, PROGRAM_VERSION ); fprintf( stdout, "Copyright (C) 2019 Andrey V.Kosteltsev.\n" ); fprintf( stdout, "This is free software. There is NO warranty; not even\n" ); fprintf( stdout, "for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n" ); fprintf( stdout, "\n" ); free_resources(); exit( EXIT_SUCCESS ); } static void remove_trailing_slash( char *dir ) { char *s; if( !dir || dir[0] == '\0' ) return; s = dir + strlen( dir ) - 1; while( *s == '/' ) { *s = '\0'; --s; } } static int _mkdir_p( const char *dir, const mode_t mode ) { char *buf; char *p = NULL; struct stat sb; if( !dir ) return -1; buf = (char *)alloca( strlen( dir ) + 1 ); strcpy( buf, dir ); remove_trailing_slash( buf ); /* check if path exists and is a directory */ if( stat( buf, &sb ) == 0 ) { if( S_ISDIR(sb.st_mode) ) { return 0; } } /* mkdir -p */ for( p = buf + 1; *p; ++p ) { if( *p == '/' ) { *p = 0; /* test path */ if( stat( buf, &sb ) != 0 ) { /* path does not exist - create directory */ if( mkdir( buf, mode ) < 0 ) { return -1; } } else if( !S_ISDIR(sb.st_mode) ) { /* not a directory */ return -1; } *p = '/'; } } /* test path */ if( stat( buf, &sb ) != 0 ) { /* path does not exist - create directory */ if( mkdir( buf, mode ) < 0 ) { return -1; } } else if( !S_ISDIR(sb.st_mode) ) { /* not a directory */ return -1; } return 0; } static void _rm_tmpdir( const char *dirpath ) { DIR *dir; char *path; size_t len; struct stat path_sb, entry_sb; struct dirent *entry; if( stat( dirpath, &path_sb ) == -1 ) { return; /* stat returns error code; errno is set */ } if( S_ISDIR(path_sb.st_mode) == 0 ) { return; /* dirpath is not a directory */ } if( (dir = opendir(dirpath) ) == NULL ) { return; /* Cannot open direcroty; errno is set */ } len = strlen( dirpath ); while( (entry = readdir( dir )) != NULL) { /* skip entries '.' and '..' */ if( ! strcmp( entry->d_name, "." ) || ! strcmp( entry->d_name, ".." ) ) continue; /* determinate a full name of an entry */ path = alloca( len + strlen( entry->d_name ) + 2 ); strcpy( path, dirpath ); strcat( path, "/" ); strcat( path, entry->d_name ); if( stat( path, &entry_sb ) == 0 ) { if( S_ISDIR(entry_sb.st_mode) ) { /* recursively remove a nested directory */ _rm_tmpdir( path ); } else { /* remove a file object */ (void)unlink( path ); } } /* else { stat() returns error code; errno is set; and we have to continue the loop } */ } /* remove the devastated directory and close the object of this directory */ (void)rmdir( dirpath ); closedir( dir ); } static char *_mk_tmpdir( void ) { char *buf = NULL, *p, *tmp = "/tmp"; size_t len = 0, size = 0; (void)umask( S_IWGRP | S_IWOTH ); /* octal 022 */ /* Get preferred directory for tmp files */ if( (p = getenv( "TMP" )) != NULL ) { tmp = p; } else if( (p = getenv( "TEMP" )) != NULL ) { tmp = p; } size = strlen( tmp ) + strlen( DISTRO_NAME ) + strlen( program ) + 12; buf = (char *)malloc( size ); if( !buf ) return NULL; len = snprintf( buf, size, (const char *)"%s/%s/%s-%.7u", tmp, DISTRO_NAME, program, getpid() ); if( len == 0 || len == size - 1 ) { free( buf ); return NULL; } _rm_tmpdir( (const char *)&buf[0] ); if( _mkdir_p( buf, S_IRWXU | S_IRWXG | S_IRWXO ) == 0 ) { return buf; } free( buf ); return NULL; } /******************************************** *requires[] functions: */ static struct package *create_package( char *name, char *version, char *arch, char *distro_name, char *distro_version ) { struct package *pkg = (struct package *)0; if( !name || *name == '\0' ) return pkg; if( !version || *version == '\0' ) return pkg; if( !arch || *arch == '\0' ) return pkg; if( !distro_name || *distro_name == '\0' ) return pkg; if( !distro_version || *distro_version == '\0' ) return pkg; pkg = (struct package *)malloc( sizeof(struct package) ); if( pkg ) { pkg->name = xstrdup( (const char *)name ); pkg->version = xstrdup( (const char *)version ); pkg->arch = xstrdup( (const char *)arch ); pkg->distro_name = xstrdup( (const char *)distro_name ); pkg->distro_version = xstrdup( (const char *)distro_version ); } return pkg; } static void free_package( struct package *pkg ) { if( pkg ) { if( pkg->name ) free( pkg->name ); if( pkg->version ) free( pkg->version ); if( pkg->arch ) free( pkg->arch ); if( pkg->distro_name ) free( pkg->distro_name ); if( pkg->distro_version ) free( pkg->distro_version ); free( pkg ); } } static struct package **create_requires( size_t size ) { struct package **requires = (struct package **)0; if( size > 0 ) { requires = (struct package **)malloc( size * sizeof(struct package *) ); bzero( (void *)requires, size * sizeof(struct package *) ); } return( requires ); } static void free_requires( struct package **requires ) { if( requires ) { struct package **ptr = requires; while( *ptr ) { if( *ptr ) free_package( *ptr ); ptr++; } free( requires ); } } /* End of *requires[] functions. ********************************************/ /******************************************** LOCK FILE functions: */ static int __lock_file( FILE *fp ) { int fd = fileno( fp ); if( flock( fd, LOCK_EX ) == -1 ) { return -1; /* Мы не проверяем errno == EWOULDBLOCK, так какданная ошибка говорит о том что файл заблокирован другим процессом с флагом LOCK_NB, а мы не собираемся циклически проверять блокировку. У нас все просто: процесс просто ждет освобождения дескриптора и не пытается во время ожидания выполнять другие задачи. */ } return fd; } static void __unlock_file( int fd ) { if( fd != -1 ) flock( fd, LOCK_UN ); /* Здесь, в случае ошибки, мы не будем выводить никаких сообщений. Наш процесс выполняет простую атомарную задачу и, наверное, завершится в скором времени, освободив все дескрипторы. */ } /* End of LOCK FILE functions. ********************************************/ void fatal_error_actions( void ) { logmsg( errlog, MSG_NOTICE, "Free resources on FATAL error..." ); if( tmpdir ) { _rm_tmpdir( (const char *)tmpdir ); free( tmpdir ); } free_resources(); } void sigint( int signum ) { (void)signum; if( tmpdir ) { _rm_tmpdir( (const char *)tmpdir ); free( tmpdir ); } free_resources(); } static void set_signal_handlers() { struct sigaction sa; sigset_t set; memset( &sa, 0, sizeof( sa ) ); sa.sa_handler = sigint; /* TERM, INT */ sa.sa_flags = SA_RESTART; sigemptyset( &set ); sigaddset( &set, SIGTERM ); sigaddset( &set, SIGINT ); sa.sa_mask = set; sigaction( SIGTERM, &sa, NULL ); sigaction( SIGINT, &sa, NULL ); memset( &sa, 0, sizeof( sa ) ); /* ignore SIGPIPE */ sa.sa_handler = SIG_IGN; sa.sa_flags = 0; sigaction( SIGPIPE, &sa, NULL ); /* System V fork+wait does not work if SIGCHLD is ignored */ signal( SIGCHLD, SIG_DFL ); } enum _pkglog_type { PKGLOG_TEXT = 0, PKGLOG_GZ, PKGLOG_BZ2, PKGLOG_XZ, PKGLOG_TAR, PKGLOG_UNKNOWN }; static enum _pkglog_type pkglog_type = PKGLOG_UNKNOWN; static char uncompress[2] = { 0, 0 }; static enum _pkglog_type check_pkglog_file( const char *fname ) { struct stat st; size_t pkglog_size = 0; unsigned char buf[8]; int rc, fd; /* SIGNATURES: https://www.garykessler.net/library/file_sigs.html */ uncompress[0] = '\0'; if( stat( fname, &st ) == -1 ) { FATAL_ERROR( "Cannot access %s file: %s", basename( (char *)fname ), strerror( errno ) ); } pkglog_size = st.st_size; if( (fd = open( fname, O_RDONLY )) == -1 ) { FATAL_ERROR( "Cannot open %s file: %s", basename( (char *)fname ), strerror( errno ) ); } rc = (int)read( fd, (void *)&buf[0], 7 ); if( rc != 7 ) { FATAL_ERROR( "Unknown type of input file %s", basename( (char *)fname ) ); } buf[7] = '\0'; /* TEXT */ if( !strncmp( (const char *)&buf[0], "PACKAGE", 7 ) ) { close( fd ); return PKGLOG_TEXT; } /* GZ */ if( buf[0] == 0x1F && buf[1] == 0x8B && buf[2] == 0x08 ) { uncompress[0] = 'x'; close( fd ); return PKGLOG_GZ; } /* BZ2 */ if( buf[0] == 0x42 && buf[1] == 0x5A && buf[2] == 0x68 ) { uncompress[0] = 'j'; close( fd ); return PKGLOG_BZ2; } /* XZ */ if( buf[0] == 0xFD && buf[1] == 0x37 && buf[2] == 0x7A && buf[3] == 0x58 && buf[4] == 0x5A && buf[5] == 0x00 ) { uncompress[0] = 'J'; close( fd ); return PKGLOG_XZ; } if( pkglog_size > 262 ) { if( lseek( fd, 257, SEEK_SET ) == -1 ) { FATAL_ERROR( "Cannot check signature of %s file: %s", basename( (char *)fname ), strerror( errno ) ); } rc = (int)read( fd, &buf[0], 5 ); if( rc != 5 ) { FATAL_ERROR( "Cannot read signature of %s file", basename( (char *)fname ) ); } /* TAR */ if( buf[0] == 0x75 && buf[1] == 0x73 && buf[2] == 0x74 && buf[3] == 0x61 && buf[4] == 0x72 ) { close( fd ); return PKGLOG_TAR; } } close( fd ); return PKGLOG_UNKNOWN; } void get_args( int argc, char *argv[] ) { const char* short_options = "hvd:o:"; const struct option long_options[] = { { "help", no_argument, NULL, 'h' }, { "version", no_argument, NULL, 'v' }, { "destination", required_argument, NULL, 'd' }, { "operation", required_argument, NULL, 'o' }, { NULL, 0, NULL, 0 } }; int ret; int option_index = 0; while( (ret = getopt_long( argc, argv, short_options, long_options, &option_index )) != -1 ) { switch( ret ) { case 'h': { usage(); break; } case 'v': { version(); break; } case 'd': { if( optarg != NULL ) { destination = xstrdup( (const char *)optarg ); remove_trailing_slash( destination ); } else /* option is present but without value */ usage(); break; } case 'o': { operation = xstrdup( (const char *)optarg ); to_lowercase( operation ); if( strcmp( operation, "inc" ) != 0 && strcmp( operation, "dec" ) != 0 ) { ERROR( "Invalid '%s' operation requested", operation ); usage(); } break; } case '?': default: { usage(); break; } } } if( operation == NULL ) { usage(); } /* last command line argument is the LOGFILE */ if( optind < argc ) { char *buf = NULL; buf = (char *)malloc( (size_t)PATH_MAX ); if( !buf ) { FATAL_ERROR( "Cannot allocate memory" ); } if( destination == NULL ) { pkglog_fname = xstrdup( (const char *)argv[optind++] ); if( pkglog_fname == NULL ) { FATAL_ERROR( "Unable to set input PKGLOG file name" ); } bzero( (void *)buf, PATH_MAX ); (void)sprintf( buf, "%s", pkglog_fname ); destination = xstrdup( (const char *)dirname( buf ) ); if( destination == NULL ) { FATAL_ERROR( "Unable to set destination directory" ); } } else { bzero( (void *)buf, PATH_MAX ); (void)sprintf( buf, "%s/%s", destination, argv[optind++] ); pkglog_fname = xstrdup( (const char *)buf ); if( pkglog_fname == NULL ) { FATAL_ERROR( "Unable to set inpit PKGLOG file name" ); } } free( buf ); pkglog_type = check_pkglog_file( (const char *)pkglog_fname ); if( pkglog_type != PKGLOG_TEXT ) { ERROR( "%s: Unknown input file format", basename( pkglog_fname ) ); usage(); } } else { usage(); } } /* Especialy for pkginfo lines. Remove leading spaces and take non-space characters only: */ static char *skip_spaces( char *s ) { char *q, *p = (char *)0; if( !s || *s == '\0' ) return p; p = s; while( (*p == ' ' || *p == '\t') && *p != '\0' ) { ++p; } q = p; while( *q != ' ' && *q != '\t' && *q != '\0' ) { ++q; } *q = '\0'; if( *p == '\0' ) return (char *)0; return( xstrdup( (const char *)p ) ); } /* remove spaces at end of line: */ static void skip_eol_spaces( char *s ) { char *p = (char *)0; if( !s || *s == '\0' ) return; p = s + strlen( s ) - 1; while( isspace( *p ) ) { *p-- = '\0'; } } int get_pkginfo() { int ret = -1; FILE *log = NULL; if( pkglog_fname != NULL ) { log = fopen( (const char *)pkglog_fname, "r" ); if( !log ) { FATAL_ERROR( "Cannot open %s file", pkglog_fname ); } } if( log != NULL ) { char *ln = NULL; char *line = NULL; char *pkgname_pattern = "PACKAGE NAME:", *pkgver_pattern = "PACKAGE VERSION:", *group_pattern = "GROUP:", *arch_pattern = "ARCH:", *distroname_pattern = "DISTRO:", *distrover_pattern = "DISTRO VERSION:"; int last = 10; /* read first 10 lines only */ line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; if( last ) { int n = 1; while( (ln = fgets( line, PATH_MAX, log )) ) { char *match = NULL; ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ if( (match = strstr( ln, pkgname_pattern )) && match == ln ) /* at start of line only */ { pkgname = skip_spaces( ln + strlen( pkgname_pattern ) ); } if( (match = strstr( ln, pkgver_pattern )) && match == ln ) { pkgver = skip_spaces( ln + strlen( pkgver_pattern ) ); } if( (match = strstr( ln, arch_pattern )) && match == ln ) { arch = skip_spaces( ln + strlen( arch_pattern ) ); } if( (match = strstr( ln, distroname_pattern )) && match == ln ) { distroname = skip_spaces( ln + strlen( distroname_pattern ) ); } if( (match = strstr( ln, distrover_pattern )) && match == ln ) { distrover = skip_spaces( ln + strlen( distrover_pattern ) ); } if( (match = strstr( ln, group_pattern )) && match == ln ) { group = skip_spaces( ln + strlen( group_pattern ) ); } if( n < last ) ++n; else break; } /* End of while() */ } free( line ); if( pkgname == NULL ) ++ret; if( pkgver == NULL ) ++ret; if( arch == NULL ) ++ret; if( distroname == NULL ) ++ret; if( distrover == NULL ) ++ret; /* group can be equal to NULL */ fclose( log ); } return( ret ); } int get_references_section( int *start, int *stop, unsigned int *cnt, FILE *log ) { int ret = -1, found = 0; if( !start || !stop || !cnt ) return ret; if( log != NULL ) { char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, log )) ) { char *match = NULL; if( (match = strstr( ln, "REFERENCE COUNTER:" )) && match == ln ) /* at start of line only */ { *start = ret + 1; ++found; /* Get reference counter */ { unsigned int count; int rc; ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ rc = sscanf( ln, "REFERENCE COUNTER: %u", &count ); if( rc == 1 && cnt != NULL ) { *cnt = count; } } } if( (match = strstr( ln, "REQUIRES:" )) && match == ln ) { *stop = ret + 1; ++found; } ++ret; } free( line ); ret = ( found == 2 ) ? 0 : 1; /* 0 - success; 1 - not found. */ fseek( log, 0, SEEK_SET ); } return( ret ); } int get_requires_section( int *start, int *stop, const char *log_fname ) { int ret = -1, found = 0; FILE *log = NULL; if( !start || !stop ) return ret; if( log_fname != NULL ) { log = fopen( (const char *)log_fname, "r" ); if( !log ) { return ret; } } if( log != NULL ) { char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, log )) ) { char *match = NULL; if( (match = strstr( ln, "REQUIRES:" )) && match == ln ) /* at start of line only */ { *start = ret + 1; ++found; } if( (match = strstr( ln, "PACKAGE DESCRIPTION:" )) && match == ln ) { *stop = ret + 1; ++found; } ++ret; } free( line ); ret = ( found == 2 ) ? 0 : 1; /* 0 - success; 1 - not found. */ fclose( log ); } return( ret ); } int get_description_section( int *start, int *stop, const char *log_fname ) { int ret = -1, found = 0; FILE *log = NULL; if( !start || !stop ) return ret; if( log_fname != NULL ) { log = fopen( (const char *)log_fname, "r" ); if( !log ) { return ret; } } if( log != NULL ) { char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, log )) ) { char *match = NULL; if( (match = strstr( ln, "PACKAGE DESCRIPTION:" )) && match == ln ) /* at start of line only */ { *start = ret + 1; ++found; } if( (match = strstr( ln, "RESTORE LINKS:" )) && match == ln ) { *stop = ret + 1; ++found; } ++ret; } free( line ); ret = ( found == 2 ) ? 0 : 1; /* 0 - success; 1 - not found. */ fclose( log ); } return( ret ); } int get_restore_links_section( int *start, int *stop, const char *log_fname ) { int ret = -1, found = 0; FILE *log = NULL; if( !start || !stop ) return ret; if( log_fname != NULL ) { log = fopen( (const char *)log_fname, "r" ); if( !log ) { return ret; } } if( log != NULL ) { char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, log )) ) { char *match = NULL; if( (match = strstr( ln, "RESTORE LINKS:" )) && match == ln ) /* at start of line only */ { *start = ret + 1; ++found; } if( (match = strstr( ln, "INSTALL SCRIPT:" )) && match == ln ) { *stop = ret + 1; ++found; } ++ret; } free( line ); ret = ( found == 2 ) ? 0 : 1; /* 0 - success; 1 - not found. */ fclose( log ); } return( ret ); } int get_install_script_section( int *start, int *stop, const char *log_fname ) { int ret = -1, found = 0; FILE *log = NULL; if( !start || !stop ) return ret; if( log_fname != NULL ) { log = fopen( (const char *)log_fname, "r" ); if( !log ) { return ret; } } if( log != NULL ) { char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, log )) ) { char *match = NULL; if( (match = strstr( ln, "INSTALL SCRIPT:" )) && match == ln ) /* at start of line only */ { *start = ret + 1; ++found; } if( (match = strstr( ln, "FILE LIST:" )) && match == ln ) { *stop = ret + 1; ++found; } ++ret; } free( line ); ret = ( found == 2 ) ? 0 : 1; /* 0 - success; 1 - not found. */ fclose( log ); } return( ret ); } int get_file_list_section( int *start, int *stop, const char *log_fname ) { int ret = -1, found = 0; FILE *log = NULL; if( !start || !stop ) return ret; if( log_fname != NULL ) { log = fopen( (const char *)log_fname, "r" ); if( !log ) { return ret; } } if( log != NULL ) { char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; *stop = 0; while( (ln = fgets( line, PATH_MAX, log )) ) { char *match = NULL; if( (match = strstr( ln, "FILE LIST:" )) && match == ln ) /* at start of line only */ { *start = ret + 1; ++found; } ++ret; } free( line ); ret = ( found == 1 ) ? 0 : 1; /* 0 - success; 1 - not found. */ fclose( log ); } return( ret ); } /*********************************************************** get_requires(): -------------- if( success ) - returns number of requires if( error ) - returns -1 */ int get_requires( const char *requires, const char *log_fname ) { int ret = -1; FILE *req = NULL, *log = NULL; int start = 0, stop = 0; if( get_requires_section( &start, &stop, log_fname ) != 0 ) { return ret; } if( log_fname != NULL ) { log = fopen( (const char *)log_fname, "r" ); if( ! log ) { return ret; } } if( requires != NULL ) { req = fopen( (const char *)requires, "w" ); if( ! req ) { return ret; } } /* get PKGLOG sections */ { char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; if( start && start < stop ) { int n = 1, lines = 0; while( (ln = fgets( line, PATH_MAX, log )) ) { ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ if( (n > start) && (n < stop) ) { fprintf( req, "%s\n", ln ); ++lines; } ++n; } ret = lines; /* number of lines in the LIST */ } free( line ); fclose( log ); fflush( req ); fclose( req ); } return( ret ); } struct package **read_requires( const char *fname, int n ) { struct package **requires = (struct package **)0; struct package **ptr = (struct package **)0; FILE *file; if( ! fname ) return NULL; requires = create_requires( n + 1 ); if( requires ) { int i; char *ln = NULL; char *line = NULL; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } struct package *package; file = fopen( fname, "r" ); if( !file ) { free( line ); free_requires( requires ); return NULL; } ptr = requires; for( i = 0; i < n; ++i ) { if( (ln = fgets( line, PATH_MAX, file )) ) { char *d, *name, *version; ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ /* сut 'name=version' by delimiter '=': */ d = index( ln, '=' ); *d = '\0'; version = ++d; name = ln; package = create_package( name, version, arch, DISTRO_NAME, DISTRO_VERSION ); if( package ) { *ptr = package; ptr++; } } } *ptr = (struct package *)0; free( line ); } return requires; } int find_requires( char *pname, const char *grp ) { char *name = NULL; char *buf = NULL; struct package **ptr = requires; buf = (char *)malloc( (size_t)PATH_MAX ); if( !buf ) { FATAL_ERROR( "Cannot allocate memory" ); } while( *ptr ) { if( grp && *grp != '\0' ) { char *p = NULL; size_t len = strlen( (*ptr)->name ); if( (p = strstr( (*ptr)->name, grp )) && (p == (*ptr)->name) ) { /* if group is equal to required package's group then remove group from the name */ name = alloca( len + 1 ); strcpy( name, (const char *)(*ptr)->name ); name = index( name, '/' ) + 1; } else { /* if group is not equal to required package's group then add group to the name */ name = alloca( len + strlen( grp ) + 2 ); (void)sprintf( name, "%s/%s", grp, (const char *)(*ptr)->name ); } } else { name = (*ptr)->name; } (void)sprintf( buf, "%s-%s-%s-%s-%s", name, (*ptr)->version, (*ptr)->arch, (*ptr)->distro_name, (*ptr)->distro_version ); if( ! strcmp( buf, pname ) ) { free( buf ); return 1; } ptr++; } free( buf ); return 0; } int save_tmp_head( FILE *log, int stop, const char *fname ) { FILE *fp; int ret = -1; char *ln = NULL; char *line = NULL; int n = 1, lines = 0; if( !stop || !log || !fname || *fname == '\0' ) return ret; fp = fopen( fname, "w" ); if( !fp ) { return ret; } line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, log )) ) { ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ if( n < stop ) { fprintf( fp, "%s\n", ln ); ++n; ++lines; } else break; } ret = lines; /* number of lines in the HEAD */ free( line ); fseek( log, 0, SEEK_SET ); fclose( fp ); return ret; } int save_tmp_tail( FILE *log, int start, const char *fname ) { FILE *fp; int ret = -1; char *ln = NULL; char *line = NULL; int n = 1, lines = 0; if( !start || !log || !fname || *fname == '\0' ) return ret; fp = fopen( fname, "w" ); if( !fp ) { return ret; } line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, log )) && (n < start) ) ++n; while( (ln = fgets( line, PATH_MAX, log )) ) { ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ fprintf( fp, "%s\n", ln ); ++lines; } ret = lines; /* number of lines in the TAIL */ free( line ); fseek( log, 0, SEEK_SET ); fclose( fp ); return ret; } int write_tmp_part( FILE *log, const char *fname ) { FILE *fp; int ret = -1; char *ln = NULL; char *line = NULL; int lines = 0; if( !log || !fname || *fname == '\0' ) return ret; fp = fopen( fname, "r" ); if( !fp ) { return ret; } line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } ++ret; while( (ln = fgets( line, PATH_MAX, fp )) ) { ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ fprintf( log, "%s\n", ln ); ++lines; } ret = lines; /* number of written lines */ free( line ); fclose( fp ); return ret; } static char **create_references( size_t size ) { char **references = (char **)0; if( size > 0 ) { references = (char **)malloc( size * sizeof(char *) ); bzero( (void *)references, size * sizeof(char *) ); } return( references ); } static void free_references( char **references ) { if( references ) { char **ptr = references; while( *ptr ) { if( *ptr ) free( *ptr ); ptr++; } free( references ); } } static char **get_references( FILE *log, int start, unsigned int *cnt, char *grp, char *name, char *version ) { char **refs = (char **)0; char **ptr; char *ln = NULL; char *line = NULL; int n = 1; size_t len = 0; unsigned int counter, pkgs; char *pkg = NULL; if( !log || !cnt || *cnt == 0 || !name || !version ) return refs; line = (char *)malloc( (size_t)PATH_MAX ); if( !line ) { FATAL_ERROR( "Cannot allocate memory" ); } pkg = (char *)malloc( (size_t)PATH_MAX ); if( !pkg ) { FATAL_ERROR( "Cannot allocate memory" ); } counter = *cnt; if( grp && *grp != '\0' ) { (void)sprintf( pkg, "%s/%s=", grp, name ); } else { (void)sprintf( pkg, "%s=", name ); } len = strlen( pkg ); refs = ptr = create_references( counter + 1 ); /* null terminated char *references[] */ while( (ln = fgets( line, PATH_MAX, log )) && (n < start) ) ++n; n = 0; pkgs = 0; while( (ln = fgets( line, PATH_MAX, log )) ) { ln[strlen(ln) - 1] = '\0'; /* replace new-line symbol */ skip_eol_spaces( ln ); /* remove spaces at end-of-line */ if( strstr( ln, "REQUIRES:" ) ) break; /* if cnt greater than real number of references */ if( n < counter ) { if( strncmp( ln, pkg, len ) ) /* always remove 'name=version' from list */ { if( refs ) { *ptr = xstrdup( (const char *)ln ); ++ptr; *ptr = (char *)0; ++pkgs; } } ++n; } else break; } free( line ); free( pkg ); fseek( log, 0, SEEK_SET ); if( pkgs == 0 ) { free_references( refs ); refs = (char **)0; } *cnt = pkgs; return refs; } static void _change_references( char *grp, char *name, char *version, const char *log_fname ) { int fd; FILE *log; char *head_fname = NULL, *tail_fname = NULL; int head_lines, tail_lines; int rc, start, stop; unsigned int counter; int inc = ( strcmp( operation, "inc" ) == 0 ) ? 1 : 0; char **references = NULL; if( !name || !version || log_fname == NULL ) return; if( check_pkglog_file( log_fname ) != PKGLOG_TEXT ) return; log = fopen( (const char *)log_fname, "r+" ); if( !log ) { ERROR( "Cannot access %s file: %s", log_fname, strerror( errno ) ); return; } fd = __lock_file( log ); rc = get_references_section( &start, &stop, &counter, log ); if( rc != 0 ) { ERROR( "%s: PKGLOG doesn't contains REFERENCE COUNTER section", log_fname ); __unlock_file( fd ); fclose( log ); return; } head_fname = (char *)alloca( strlen( tmpdir ) + 7 ); (void)sprintf( head_fname, "%s/.HEAD", tmpdir ); tail_fname = (char *)alloca( strlen( tmpdir ) + 7 ); (void)sprintf( tail_fname, "%s/.TAIL", tmpdir ); head_lines = save_tmp_head( log, start, (const char *)head_fname ); tail_lines = save_tmp_tail( log, stop - 1, (const char *)tail_fname ); if( head_lines < 10 && tail_lines < 12 ) { ERROR( "%s: Invalid PKGLOG file", log_fname ); __unlock_file( fd ); fclose( log ); return; } references = get_references( log, start, &counter, grp, name, version ); if( ftruncate( fd, 0 ) != 0 ) { ERROR( "Cannot change REFERENCE COUNTER in the %s file: %s", log_fname, strerror( errno ) ); free_references( references ); __unlock_file( fd ); fclose( log ); return; } head_lines = write_tmp_part( log, (const char *)head_fname ); if( inc ) ++counter; fprintf( log, "REFERENCE COUNTER: %u\n", counter ); if( inc ) { if( grp && *grp != '\0' ) { fprintf( log, "%s/%s=%s\n", grp, name, version ); } else { fprintf( log, "%s=%s\n", name, version ); } } if( references ) { char **ptr = references; while( *ptr ) { if( *ptr ) fprintf( log, "%s\n", *ptr ); ptr++; } free_references( references ); } tail_lines = write_tmp_part( log, (const char *)tail_fname ); __unlock_file( fd ); fclose( log ); } static void _search_required_packages( const char *dirpath, const char *grp ) { DIR *dir; char *path; size_t len; struct stat path_sb, entry_sb; struct dirent *entry; if( stat( dirpath, &path_sb ) == -1 ) { FATAL_ERROR( "%s: Cannot stat Setup Database or destination directory", dirpath ); } if( S_ISDIR(path_sb.st_mode) == 0 ) { FATAL_ERROR( "%s: Setup Database or destination is not a directory", dirpath ); } if( (dir = opendir(dirpath) ) == NULL ) { FATAL_ERROR( "Canot access %s directory: %s", dirpath, strerror( errno ) ); } len = strlen( dirpath ); while( (entry = readdir( dir )) != NULL) { /* skip entries '.' and '..' */ if( ! strcmp( entry->d_name, "." ) || ! strcmp( entry->d_name, ".." ) ) continue; /* determinate a full name of an entry */ path = alloca( len + strlen( entry->d_name ) + 2 ); strcpy( path, dirpath ); strcat( path, "/" ); strcat( path, entry->d_name ); if( stat( path, &entry_sb ) == 0 ) { if( S_ISREG(entry_sb.st_mode) ) { if( find_requires( entry->d_name, grp ) ) { _change_references( group, pkgname, pkgver, (const char *)path ); } } if( S_ISDIR(entry_sb.st_mode) && grp == NULL ) { _search_required_packages( (const char *)path, (const char *)entry->d_name ); } } /* else { stat() returns error code; errno is set; and we have to continue the loop } */ } closedir( dir ); } /********************************************* Get directory where this program is placed: */ char *get_selfdir( void ) { char *buf = NULL; ssize_t len; buf = (char *)malloc( PATH_MAX ); if( !buf ) { FATAL_ERROR( "Cannot allocate memory" ); } bzero( (void *)buf, PATH_MAX ); len = readlink( "/proc/self/exe", buf, (size_t)PATH_MAX ); if( len > 0 && len < PATH_MAX ) { char *p = xstrdup( (const char *)dirname( buf ) ); free( buf ); return p; } FATAL_ERROR( "Cannot determine self directory. Please mount /proc filesystem" ); } void set_stack_size( void ) { const rlim_t stack_size = 16 * 1024 * 1024; /* min stack size = 16 MB */ struct rlimit rl; int ret; ret = getrlimit( RLIMIT_STACK, &rl ); if( ret == 0 ) { if( rl.rlim_cur < stack_size ) { rl.rlim_cur = stack_size; ret = setrlimit( RLIMIT_STACK, &rl ); if( ret != 0 ) { fprintf(stderr, "setrlimit returned result = %d\n", ret); FATAL_ERROR( "Cannot set stack size" ); } } } } int main( int argc, char *argv[] ) { gid_t gid; int ret; set_signal_handlers(); gid = getgid(); setgroups( 1, &gid ); fatal_error_hook = fatal_error_actions; selfdir = get_selfdir(); errlog = stderr; program = basename( argv[0] ); get_args( argc, argv ); /* set_stack_size(); */ exit_status = get_pkginfo(); if( exit_status != 0 ) { FATAL_ERROR( "%s: Invalid input PKGLOG file", basename( pkglog_fname ) ); } tmpdir = _mk_tmpdir(); if( !tmpdir ) { FATAL_ERROR( "Cannot create temporary directory" ); } else { char *buf = NULL; buf = (char *)malloc( (size_t)strlen( tmpdir ) + 11 ); if( !buf ) { FATAL_ERROR( "Cannot allocate memory" ); } (void)sprintf( (char *)&buf[0], "%s/.REQUIRES", tmpdir ); requires_fname = xstrdup( (const char *)&buf[0] ); free( buf ); } /******************* Getting REQUIRES: */ if( (ret = get_requires( (const char *)requires_fname, (const char *)pkglog_fname )) > 0 ) { /* We have non-empty list of REQUIRES in the 'requires_fname' file */ requires = read_requires( (const char *)requires_fname, ret ); _search_required_packages( (const char *)destination, NULL ); } if( tmpdir ) { _rm_tmpdir( (const char *)tmpdir ); free( tmpdir ); } free_resources(); exit( exit_status ); }
Radix cross Linux package tools
Package Tools – is a set of utilities to create, install, and update RcL packages
4 Commits
0 Branches
2 Tags
| Donate