aria2c-file-parser/read_a2.py
2017-11-19 17:32:31 +08:00

74 lines
2.8 KiB
Python

import struct
import binascii
# ================================================================
# 0 1 2 3
# 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
# +---+-------+-------+-------------------------------------------+
# |VER| EXT |INFO |INFO HASH ... |
# |(2)| (4) |HASH | (INFO HASH LENGTH) |
# | | |LENGTH | |
# | | | (4) | |
# +---+---+---+-------+---+---------------+-------+---------------+
# |PIECE |TOTAL LENGTH |UPLOAD LENGTH |BIT- |BITFIELD ... |
# |LENGTH | (8) | (8) |FIELD | (BITFIELD |
# | (4) | | |LENGTH | LENGTH) |
# | | | | (4) | |
# +-------+-------+-------+-------+-------+-------+---------------+
# |NUM |INDEX |LENGTH |PIECE |PIECE BITFIELD ... |
# |IN- | (4) | (4) |BIT- | (PIECE BITFIELD LENGTH) |
# |FLIGHT | | |FIELD | |
# |PIECE | | |LENGTH | |
# | (4) | | | (4) | |
# +-------+-------+-------+-------+-------------------------------+
#
# ^ ^
# | |
# +-------------------------------------------------------+
# Repeated in (NUM IN-FLIGHT) PIECE times
# more detail
# https://aria2.github.io/manual/en/html/technical-notes.html
# ================================================================
def parse_aria_control_file(file_name):
with open(file_name, "rb") as f:
try:
f.seek(0) # Go to beginning, read VER
version = f.read(2)
i = int(version.encode('hex'), 16)
print "version is " + str(i)
# skip EXT, find info hash_binary length
f.seek(6)
length = f.read(4)
hash_length = int(length.encode('hex'), 16)
print "hash length is " + str(hash_length)
# read next hash_length
f.seek(10) # Go to info hash
hash_binary = f.read(hash_length)
info_hash = ""
for ch in hash_binary:
hex_word = hex(ord(ch))[2:]
info_hash += hex_word.upper()
magnet_link = "magnet:?xt=urn:btih:" + info_hash
print magnet_link
except:
pass
finally:
f.close()
if __name__ == '__main__':
parse_aria_control_file("dahufa.aria2")
# version is 1
# hash length is 20
# magnet:?xt=urn:btih:959E2ECEB954313D3869EFF7924CA7CD8DE739