Ajouter des octets de la chaîne?

J'ai un fichier que j'ai ouvert en Python que je suis en train d'écrire à un autre fichier de sortie. J'ai besoin qu'elle soit un multiple de 16 ans, en taille, c'est ce que j'ai imaginé:

 with open(input_file, 'rb') as infile:
     with open(output_file, 'wb') as outfile:
         while True:
             input_chunk = infile.read(64)

             if len(input_chunk) == 0:
                 break
             elif len(input_chunk) % 16 > 0:
                 input_chunk.extend([0 for i in range(len(input_chunk) % 16)])

             output_file.write(input_chunk)

Malheureusement, il ne réussit pas à ajouter des zéros avec:

AttributeError: 'str' object has no attribute 'extend'

Tout d'abord, pourquoi faire j'ai une chaîne de caractères ici, plutôt qu'un tableau d'octets? Je suis en train de lire le fichier binaire en mode binaire.

Deuxième, si je fais face à une chaîne de caractères, comment puis-je écrire un nombre d'octets avec la valeur 0 à la fin de cette chaîne?

les chaînes de caractères sont des objets immuables.
Peut-être input_chunk = input_chunk + '\0' * (len(input_chunk) % 16)
Est-il possible que je peux lire un fichier sous la forme de tableaux d'octets?
Je pense que le problème est sur la prolonger l'appel.
Êtes-vous à l'aide de Python 2 ou 3?

OriginalL'auteur Naftuli Kay | 2013-02-19

  1. 5

    Tout d'abord, pourquoi faire j'ai une chaîne de caractères ici, plutôt qu'un tableau d'octets?

    Parce que c'est ce file.read est de retour ...

    Deuxième, si je fais face à une chaîne de caractères, comment puis-je écrire un nombre d'octets avec la valeur 0 à la fin de cette chaîne?

    Vous ne pouvez pas écrire sur cette chaîne, car les chaînes sont immuables. Cependant, vous pouvez l'écrire sur une autre chaîne et de créer de nouvelles chaînes par concaténation:

    >>> import struct
>>> input_chunk = 'foo bar baz'
>>> input_chunk + struct.pack('16B',*([0]*16))
'foo bar baz\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
>>>

    Noter que si vous savez que vous avez 64 octets et que vous voulez une chaîne de caractères qui est de 80 octets rembourré avec des valeurs null, struct.pack sera pad automatiquement avec les valeurs null:

    struct.pack('80s',string_of_64_bytes)

    Pour le 's' de caractères de format, le nombre est interprété comme la taille de la chaîne, pas un compteur de répétition, comme pour les autres format de caractères; par exemple, '10' signifie un seul de 10 octets de la chaîne, tandis que les " 10c " signifie 10 caractères. Si un nombre n'est pas spécifié, la valeur par défaut est 1. Pour l'emballage, la chaîne est tronquée ou collier avec des octets nuls comme approprié pour le faire rentrer. Pour le déballage, la chaîne résultante a toujours exactement le nombre spécifié d'octets. Comme cas particulier, '0' signifie une seule chaîne vide (alors que '0c' signifie 0 caractères).

    Il sonne comme l'OP est en train de faire du chiffrement AES. Si c'est le cas, alors le rembourrage ne doit pas être tous nuls. Il devrait être suivant de Remplissage PKCS#7 conformément à la RFC 5652 et être remplis avec de l'octet qui est la valeur de chaque octet.

    OriginalL'auteur mgilson

  2. 2

    En Python 2.x, un str objet est un "tableau d'octets".

    Si vous avez besoin d'un mutables, tableau, utilisez un objet bytearray sur les versions 2.6+:

    >>> a = bytearray('my_string')
>>> a.extend(' hello')
>>> str(a)
'my_string hello'

    Autrement:

    >>> import array
>>> a = array.array('c', 'my_string')
    Vous avez également bytearray en 2.7.
    En fait, bytearray est disponible depuis la 2.6 bytearray
    Que fait bytearray un meilleur choix ensuite.

    OriginalL'auteur JBernardo

  3. 1

    Suffit d'utiliser bytearray et vous devriez être bon en tant que-est:

               input_chunk = bytearray(infile.read(64))

    Mais je voudrais juste utiliser quelque chose comme [0] * (16 - len(input_chunk)%16) au lieu d'une liste de la compréhension pour le rembourrage.

    OriginalL'auteur Mark Reed

  4. 0

    Vous pouvez remplir votre input_chunk comme ce

    input_chunk += '\x00'*(-len(input_chunk)%16)
    Comment puis-je convertir un int en un octet comme ça? le \x00 devra être dynamique, comme je vais changer ça pour être le reste de len(input_chunk)%16.

    OriginalL'auteur John La Rooy