Ejemplo n.º 1
0
def integrity_check ( filename , item , skip_check=SKIP_NONE , bsize=128 ) :
    """Checks integrity of the given file with values supplied on a dictionary.
Will return False on failure of any check, and True if no verification fails.
If all checks are skipped or no checksum can be verified, string "OK" is
returned, which still evaluates to boolean true, but serves to detect partial
verification if required."""

    keysout = ( "name" , "href" , "filename" , "arch" , "size" ,
                "group" , "provides" , "requires" )

    checksums = filter( lambda x : x not in keysout ,
                        map( lambda x : x.lower() , item.keys() )
                        )

    name = os.path.basename(filename)

    if skip_check == SKIP_ALL :
        logger.warning( "No check selected for '%s'" % name )
        return "OK"

    res = True

    if not ( skip_check & SKIP_SIZE ) :
        if os.stat( filename ).st_size != int( item['size'] ) :
            logger.warning( "Bad size on file '%s'" % name )
            return False

    # Policy is to verify all the available checksums
    if not ( skip_check & SKIP_CKSUM ) :
        res = "OK"
        for cktype in cksum_handles.keys() :
            if item.has_key( cktype ) :
                if cksum_handles[cktype]( filename , bsize ) == item[cktype] :
                    res = True
                else :
                    logger.warning( "Bad %s checksum '%s'" % ( cktype , name ) )
                    return False

        if res == "OK" :
          if not checksums :
            logger.warning( "No checksum defined for %s" % name )
          else :
            logger.warning( "Unknown checksum types available for %s : %s" % ( name , checksums ) )

    return res
Ejemplo n.º 2
0
def integrity_check ( filename , item , skip_check=0 , bsize=128 ) :
    """Checks integrity of the given file with values supplied on a dictionary, skipping some if requested.
Returns true for successfull checks, and false when some check fails. If no checksums were actually performed, returns None.

NOTE : None could be returned if size is verified but no known checksum type exists. This counter-intuitive result can
be avoided specifying SKIP_CKSUM for those cases where the absence of valid checksums is known in advance, but cannot be
globally fixed.
"""

    checksums = map( lambda x : x.lower() , item.keys() )

    # Remove file name, unrelated to checksum types
    name = False
    for key in ( "href" , "name" ) :
        if item.has_key( key ) :
            name = item[key]
            checksums.remove( key )
            break
    else :
        name = os.path.basename(filename)

    if skip_check == ( SKIP_SIZE | SKIP_CKSUM ) :
        logger.warning( "No check selected for '%s'" % name )
        # FIXME : This return should n't be required if setting res=None in advance
        return None

    if not ( skip_check & SKIP_SIZE ) :
        if os.stat( filename ).st_size != int( item['size'] ) :
            logger.warning( "Bad size on file '%s'" % name )
            return False
        res = True
        checksums.remove( "size" )

    # Policy is to verify all the available checksums
    if not ( skip_check & SKIP_CKSUM ) :
        res = None
        for type in cksum_handles.keys() :
            if item.has_key( type ) :
                if cksum_handles[type]( filename , bsize ) == item[type] :
                    res = True
                else :
                    logger.warning( "Bad %s checksum '%s'" % ( type , name ) )
                    return False

        if res is None :
            logger.warning( "Unknonw checksum types available for file : %s" % checksums )

    return res