用Python進行ASN.1的編解碼

記錄一下如何用Python來進行ASN.1的編解碼。

首先定義一個ASN.1的Schema，并儲存為visioncontrol.asn檔案

VISIONUPLOAD DEFINITIONS AUTOMATIC TAGS::= BEGIN 

-- Message from the platform
ITSMessageVisionUpload ::= SEQUENCE {
   imageType        ImageType,  
   imageData        OCTET STRING
}

ImageType ::= ENUMERATED {
   jpg                              (0),  
   png                              (1)
}

END
           

有兩種方法來進行編解碼

方法一：

安裝asn1ate， git clone https://github.com/kimgr/asn1ate.git， python setup.py build, python setup.py install

執行指令asn1ate visioncontrol.asn > visioncontrol.py， 來生成對應的Python類

安裝pyasn1，git clone http://www.github.com/etingof/pyasn1

之後就可以進行編解碼了，非常簡便易用。

import visioncontrol
from hexdump import hexdump
from pyasn1.codec.der.encoder import encode

vc = visioncontrol.ITSMessageVisionUpload()
vc['imageType']='jpg'
vc['imageData']=b'10100000'
dercode = encode(vc)

hexdump(dercode) 
           

這個方法有個缺點，就是pyasn1支援的編碼格式比較少，隻有BER，DER幾種。如果我們想編碼其他的格式，例如UPER的話就不行，是以我又找到了方法二。

方法二：

安裝asn1tools, pip install asn1tools

import asn1tools

visionupload = asn1tools.compile_files('cvc-vds-its-visionupload.asn', 'uper')

#Read the image as bytesarray
pic = open('test2.jpg', 'rb')
picdata = pic.read()
pic.close()

encoded = visionupload.encode('ITSMessageVisionUpload', {'imageType':'jpg', 'imageData':picdata})

vision_decoded = visionupload.decode('ITSMessageVisionUpload', encoded)
           

這個方法支援的編碼格式多，而且也不需要預先把編碼格式轉換為Python類，使用更加簡單。