123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037 |
- #
- # This file is part of pyasn1 software.
- #
- # Copyright (c) 2005-2020, Ilya Etingof <etingof@gmail.com>
- # License: https://pyasn1.readthedocs.io/en/latest/license.html
- #
- import gzip
- import io
- import os
- import sys
- import tempfile
- import unittest
- import zipfile
- from __tests__.base import BaseTestCase
- from pyasn1.type import tag
- from pyasn1.type import namedtype
- from pyasn1.type import opentype
- from pyasn1.type import univ
- from pyasn1.type import char
- from pyasn1.codec import streaming
- from pyasn1.codec.ber import decoder
- from pyasn1.codec.ber import eoo
- from pyasn1 import error
- class LargeTagDecoderTestCase(BaseTestCase):
- def testLargeTag(self):
- assert decoder.decode(bytes((127, 141, 245, 182, 253, 47, 3, 2, 1, 1))) == (1, b'')
- def testLongTag(self):
- assert decoder.decode(bytes((0x1f, 2, 1, 0)))[0].tagSet == univ.Integer.tagSet
- def testTagsEquivalence(self):
- integer = univ.Integer(2).subtype(implicitTag=tag.Tag(tag.tagClassContext, 0, 0))
- assert decoder.decode(bytes((0x9f, 0x80, 0x00, 0x02, 0x01, 0x02)), asn1Spec=integer) == decoder.decode(
- bytes((0x9f, 0x00, 0x02, 0x01, 0x02)), asn1Spec=integer)
- class DecoderCacheTestCase(BaseTestCase):
- def testCache(self):
- assert decoder.decode(bytes((0x1f, 2, 1, 0))) == decoder.decode(bytes((0x1f, 2, 1, 0)))
- class IntegerDecoderTestCase(BaseTestCase):
- def testPosInt(self):
- assert decoder.decode(bytes((2, 1, 12))) == (12, b'')
- def testNegInt(self):
- assert decoder.decode(bytes((2, 1, 244))) == (-12, b'')
- def testZero(self):
- assert decoder.decode(bytes((2, 0))) == (0, b'')
- def testZeroLong(self):
- assert decoder.decode(bytes((2, 1, 0))) == (0, b'')
- def testMinusOne(self):
- assert decoder.decode(bytes((2, 1, 255))) == (-1, b'')
- def testPosLong(self):
- assert decoder.decode(
- bytes((2, 9, 0, 255, 255, 255, 255, 255, 255, 255, 255))
- ) == (0xffffffffffffffff, b'')
- def testNegLong(self):
- assert decoder.decode(
- bytes((2, 9, 255, 0, 0, 0, 0, 0, 0, 0, 1))
- ) == (-0xffffffffffffffff, b'')
- def testSpec(self):
- try:
- decoder.decode(
- bytes((2, 1, 12)), asn1Spec=univ.Null()
- ) == (12, b'')
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong asn1Spec worked out'
- assert decoder.decode(
- bytes((2, 1, 12)), asn1Spec=univ.Integer()
- ) == (12, b'')
- def testTagFormat(self):
- try:
- decoder.decode(bytes((34, 1, 12)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- class BooleanDecoderTestCase(BaseTestCase):
- def testTrue(self):
- assert decoder.decode(bytes((1, 1, 1))) == (1, b'')
- def testTrueNeg(self):
- assert decoder.decode(bytes((1, 1, 255))) == (1, b'')
- def testExtraTrue(self):
- assert decoder.decode(bytes((1, 1, 1, 0, 120, 50, 50))) == (1, bytes((0, 120, 50, 50)))
- def testFalse(self):
- assert decoder.decode(bytes((1, 1, 0))) == (0, b'')
- def testTagFormat(self):
- try:
- decoder.decode(bytes((33, 1, 1)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- class BitStringDecoderTestCase(BaseTestCase):
- def testDefMode(self):
- assert decoder.decode(
- bytes((3, 3, 1, 169, 138))
- ) == ((1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1), b'')
- def testIndefMode(self):
- assert decoder.decode(
- bytes((3, 3, 1, 169, 138))
- ) == ((1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1), b'')
- def testDefModeChunked(self):
- assert decoder.decode(
- bytes((35, 8, 3, 2, 0, 169, 3, 2, 1, 138))
- ) == ((1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1), b'')
- def testIndefModeChunked(self):
- assert decoder.decode(
- bytes((35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0))
- ) == ((1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1), b'')
- def testDefModeChunkedSubst(self):
- assert decoder.decode(
- bytes((35, 8, 3, 2, 0, 169, 3, 2, 1, 138)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((3, 2, 0, 169, 3, 2, 1, 138)), b'')
- def testDefModeChunkedSubstV04(self):
- assert decoder.decode(
- bytes((35, 8, 3, 2, 0, 169, 3, 2, 1, 138)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((3, 2, 0, 169, 3, 2, 1, 138)), b'')
- def testIndefModeChunkedSubst(self):
- assert decoder.decode(
- bytes((35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((3, 2, 0, 169, 3, 2, 1, 138, 0, 0)), b'')
- def testIndefModeChunkedSubstV04(self):
- assert decoder.decode(
- bytes((35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((3, 2, 0, 169, 3, 2, 1, 138, 0, 0)), b'')
- def testTypeChecking(self):
- try:
- decoder.decode(bytes((35, 4, 2, 2, 42, 42)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'accepted mis-encoded bit-string constructed out of an integer'
- class OctetStringDecoderTestCase(BaseTestCase):
- def testDefMode(self):
- assert decoder.decode(
- bytes((4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120))
- ) == (b'Quick brown fox', b'')
- def testIndefMode(self):
- assert decoder.decode(
- bytes((36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0, 0))
- ) == (b'Quick brown fox', b'')
- def testDefModeChunked(self):
- assert decoder.decode(
- bytes(
- (36, 23, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120))
- ) == (b'Quick brown fox', b'')
- def testIndefModeChunked(self):
- assert decoder.decode(
- bytes((36, 128, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120, 0, 0))
- ) == (b'Quick brown fox', b'')
- def testDefModeChunkedSubst(self):
- assert decoder.decode(
- bytes(
- (36, 23, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120)), b'')
- def testDefModeChunkedSubstV04(self):
- assert decoder.decode(
- bytes(
- (36, 23, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120)), b'')
- def testIndefModeChunkedSubst(self):
- assert decoder.decode(
- bytes((36, 128, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111,
- 120, 0, 0)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes(
- (4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120, 0, 0)), b'')
- def testIndefModeChunkedSubstV04(self):
- assert decoder.decode(
- bytes((36, 128, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111,
- 120, 0, 0)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes(
- (4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120, 0, 0)), b'')
- class ExpTaggedOctetStringDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.o = univ.OctetString(
- 'Quick brown fox',
- tagSet=univ.OctetString.tagSet.tagExplicitly(
- tag.Tag(tag.tagClassApplication, tag.tagFormatSimple, 5)
- ))
- def testDefMode(self):
- o, r = decoder.decode(
- bytes((101, 17, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120))
- )
- assert not r
- assert self.o == o
- assert self.o.tagSet == o.tagSet
- assert self.o.isSameTypeWith(o)
- def testIndefMode(self):
- o, r = decoder.decode(
- bytes((101, 128, 36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0, 0, 0, 0))
- )
- assert not r
- assert self.o == o
- assert self.o.tagSet == o.tagSet
- assert self.o.isSameTypeWith(o)
- def testDefModeChunked(self):
- o, r = decoder.decode(
- bytes((101, 25, 36, 23, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120))
- )
- assert not r
- assert self.o == o
- assert self.o.tagSet == o.tagSet
- assert self.o.isSameTypeWith(o)
- def testIndefModeChunked(self):
- o, r = decoder.decode(
- bytes((101, 128, 36, 128, 4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120, 0, 0, 0, 0))
- )
- assert not r
- assert self.o == o
- assert self.o.tagSet == o.tagSet
- assert self.o.isSameTypeWith(o)
- def testDefModeSubst(self):
- assert decoder.decode(
- bytes((101, 17, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120)), b'')
- def testDefModeSubstV04(self):
- assert decoder.decode(
- bytes((101, 17, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120)), b'')
- def testIndefModeSubst(self):
- assert decoder.decode(
- bytes((
- 101, 128, 36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0,
- 0, 0, 0)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes(
- (36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0, 0, 0, 0)), b'')
- def testIndefModeSubstV04(self):
- assert decoder.decode(
- bytes((
- 101, 128, 36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0,
- 0, 0, 0)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes(
- (36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0, 0, 0, 0)), b'')
- class NullDecoderTestCase(BaseTestCase):
- def testNull(self):
- assert decoder.decode(bytes((5, 0))) == (b'', b'')
- def testTagFormat(self):
- try:
- decoder.decode(bytes((37, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- # Useful analysis of OID encoding issues could be found here:
- # https://misc.daniel-marschall.de/asn.1/oid_facts.html
- class ObjectIdentifierDecoderTestCase(BaseTestCase):
- def testOne(self):
- assert decoder.decode(
- bytes((6, 6, 43, 6, 0, 191, 255, 126))
- ) == ((1, 3, 6, 0, 0xffffe), b'')
- def testEdge1(self):
- assert decoder.decode(
- bytes((6, 1, 39))
- ) == ((0, 39), b'')
- def testEdge2(self):
- assert decoder.decode(
- bytes((6, 1, 79))
- ) == ((1, 39), b'')
- def testEdge3(self):
- assert decoder.decode(
- bytes((6, 1, 120))
- ) == ((2, 40), b'')
- def testEdge4(self):
- assert decoder.decode(
- bytes((6, 5, 0x90, 0x80, 0x80, 0x80, 0x4F))
- ) == ((2, 0xffffffff), b'')
- def testEdge5(self):
- assert decoder.decode(
- bytes((6, 1, 0x7F))
- ) == ((2, 47), b'')
- def testEdge6(self):
- assert decoder.decode(
- bytes((6, 2, 0x81, 0x00))
- ) == ((2, 48), b'')
- def testEdge7(self):
- assert decoder.decode(
- bytes((6, 3, 0x81, 0x34, 0x03))
- ) == ((2, 100, 3), b'')
- def testEdge8(self):
- assert decoder.decode(
- bytes((6, 2, 133, 0))
- ) == ((2, 560), b'')
- def testEdge9(self):
- assert decoder.decode(
- bytes((6, 4, 0x88, 0x84, 0x87, 0x02))
- ) == ((2, 16843570), b'')
- def testNonLeading0x80(self):
- assert decoder.decode(
- bytes((6, 5, 85, 4, 129, 128, 0)),
- ) == ((2, 5, 4, 16384), b'')
- def testLeading0x80Case1(self):
- try:
- decoder.decode(
- bytes((6, 5, 85, 4, 128, 129, 0))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testLeading0x80Case2(self):
- try:
- decoder.decode(
- bytes((6, 7, 1, 0x80, 0x80, 0x80, 0x80, 0x80, 0x7F))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testLeading0x80Case3(self):
- try:
- decoder.decode(
- bytes((6, 2, 0x80, 1))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testLeading0x80Case4(self):
- try:
- decoder.decode(
- bytes((6, 2, 0x80, 0x7F))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testTagFormat(self):
- try:
- decoder.decode(bytes((38, 1, 239)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- def testZeroLength(self):
- try:
- decoder.decode(bytes((6, 0, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'zero length tolerated'
- def testIndefiniteLength(self):
- try:
- decoder.decode(bytes((6, 128, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'indefinite length tolerated'
- def testReservedLength(self):
- try:
- decoder.decode(bytes((6, 255, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'reserved length tolerated'
- def testLarge1(self):
- assert decoder.decode(
- bytes((0x06, 0x11, 0x83, 0xC6, 0xDF, 0xD4, 0xCC, 0xB3, 0xFF, 0xFF, 0xFE, 0xF0, 0xB8, 0xD6, 0xB8, 0xCB, 0xE2, 0xB7, 0x17))
- ) == ((2, 18446744073709551535184467440737095), b'')
- def testLarge2(self):
- assert decoder.decode(
- bytes((0x06, 0x13, 0x88, 0x37, 0x83, 0xC6, 0xDF, 0xD4, 0xCC, 0xB3, 0xFF, 0xFF, 0xFE, 0xF0, 0xB8, 0xD6, 0xB8, 0xCB, 0xE2, 0xB6, 0x47))
- ) == ((2, 999, 18446744073709551535184467440737095), b'')
- class RelativeOIDDecoderTestCase(BaseTestCase):
- def testOne(self):
- obj, rest = decoder.decode(bytes((13, 1, 39)))
- assert str(obj) == '39'
- assert rest == b''
- def testTwo(self):
- assert decoder.decode(
- bytes((13, 2, 5, 6))
- ) == ((5, 6), b'')
- def testThree(self):
- assert decoder.decode(
- bytes((13, 3, 5, 6, 7))
- ) == ((5, 6, 7), b'')
- def testNonLeading0x80(self):
- assert decoder.decode(
- bytes((13, 5, 85, 4, 129, 128, 0)),
- ) == ((85, 4, 16384), b'')
- def testLeading0x80(self):
- try:
- decoder.decode(
- bytes((13, 5, 85, 4, 128, 129, 0))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testTagFormat(self):
- try:
- decoder.decode(bytes((38, 1, 239)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- def testZeroLength(self):
- try:
- decoder.decode(bytes((13, 0, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'zero length tolerated'
- def testIndefiniteLength(self):
- try:
- decoder.decode(bytes((13, 128, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'indefinite length tolerated'
- def testReservedLength(self):
- try:
- decoder.decode(bytes((13, 255, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'reserved length tolerated'
- def testLarge(self):
- assert decoder.decode(
- bytes((0x0D, 0x13, 0x88, 0x37, 0x83, 0xC6, 0xDF, 0xD4, 0xCC, 0xB3, 0xFF, 0xFF, 0xFE, 0xF0, 0xB8, 0xD6, 0xB8, 0xCB, 0xE2, 0xB6, 0x47))
- ) == ((1079, 18446744073709551535184467440737095), b'')
- class RealDecoderTestCase(BaseTestCase):
- def testChar(self):
- assert decoder.decode(
- bytes((9, 7, 3, 49, 50, 51, 69, 49, 49))
- ) == (univ.Real((123, 10, 11)), b'')
- def testBin1(self): # check base = 2
- assert decoder.decode( # (0.5, 2, 0) encoded with base = 2
- bytes((9, 3, 128, 255, 1))
- ) == (univ.Real((1, 2, -1)), b'')
- def testBin2(self): # check base = 2 and scale factor
- assert decoder.decode( # (3.25, 2, 0) encoded with base = 8
- bytes((9, 3, 148, 255, 13))
- ) == (univ.Real((26, 2, -3)), b'')
- def testBin3(self): # check base = 16
- assert decoder.decode( # (0.00390625, 2, 0) encoded with base = 16
- bytes((9, 3, 160, 254, 1))
- ) == (univ.Real((1, 2, -8)), b'')
- def testBin4(self): # check exponent = 0
- assert decoder.decode( # (1, 2, 0) encoded with base = 2
- bytes((9, 3, 128, 0, 1))
- ) == (univ.Real((1, 2, 0)), b'')
- def testBin5(self): # case of 2 octs for exponent and negative exponent
- assert decoder.decode( # (3, 2, -1020) encoded with base = 16
- bytes((9, 4, 161, 255, 1, 3))
- ) == (univ.Real((3, 2, -1020)), b'')
- # TODO: this requires Real type comparison fix
- # def testBin6(self):
- # assert decoder.decode(
- # bytes((9, 5, 162, 0, 255, 255, 1))
- # ) == (univ.Real((1, 2, 262140)), b'')
- # def testBin7(self):
- # assert decoder.decode(
- # bytes((9, 7, 227, 4, 1, 35, 69, 103, 1))
- # ) == (univ.Real((-1, 2, 76354972)), b'')
- def testPlusInf(self):
- assert decoder.decode(
- bytes((9, 1, 64))
- ) == (univ.Real('inf'), b'')
- def testMinusInf(self):
- assert decoder.decode(
- bytes((9, 1, 65))
- ) == (univ.Real('-inf'), b'')
- def testEmpty(self):
- assert decoder.decode(
- bytes((9, 0))
- ) == (univ.Real(0.0), b'')
- def testTagFormat(self):
- try:
- decoder.decode(bytes((41, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- def testShortEncoding(self):
- try:
- decoder.decode(bytes((9, 1, 131)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'accepted too-short real'
- class UniversalStringDecoderTestCase(BaseTestCase):
- def testDecoder(self):
- assert decoder.decode(bytes((28, 12, 0, 0, 0, 97, 0, 0, 0, 98, 0, 0, 0, 99))) == (char.UniversalString('abc'), b'')
- class BMPStringDecoderTestCase(BaseTestCase):
- def testDecoder(self):
- assert decoder.decode(bytes((30, 6, 0, 97, 0, 98, 0, 99))) == (char.BMPString('abc'), b'')
- class UTF8StringDecoderTestCase(BaseTestCase):
- def testDecoder(self):
- assert decoder.decode(bytes((12, 3, 97, 98, 99))) == (char.UTF8String('abc'), b'')
- class SequenceOfDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.SequenceOf(componentType=univ.OctetString())
- self.s.setComponentByPosition(0, univ.OctetString('quick brown'))
- def testDefMode(self):
- assert decoder.decode(
- bytes((48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- ) == (self.s, b'')
- def testIndefMode(self):
- assert decoder.decode(
- bytes((48, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0))
- ) == (self.s, b'')
- def testDefModeChunked(self):
- assert decoder.decode(
- bytes((48, 19, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110))
- ) == (self.s, b'')
- def testIndefModeChunked(self):
- assert decoder.decode(
- bytes((48, 128, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 0, 0))
- ) == (self.s, b'')
- def testSchemalessDecoder(self):
- assert decoder.decode(
- bytes((48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=univ.SequenceOf()
- ) == (self.s, b'')
- class ExpTaggedSequenceOfDecoderTestCase(BaseTestCase):
- def testWithSchema(self):
- s = univ.SequenceOf().subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatConstructed, 3))
- s2, r = decoder.decode(
- bytes((163, 15, 48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=s
- )
- assert not r
- assert s2 == [b'quick brown']
- assert s.tagSet == s2.tagSet
- def testWithoutSchema(self):
- s = univ.SequenceOf().subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatConstructed, 3))
- s2, r = decoder.decode(
- bytes((163, 15, 48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- )
- assert not r
- assert s2 == [b'quick brown']
- assert s.tagSet == s2.tagSet
- class SequenceOfDecoderWithSchemaTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.SequenceOf(componentType=univ.OctetString())
- self.s.setComponentByPosition(0, univ.OctetString('quick brown'))
- def testDefMode(self):
- assert decoder.decode(
- bytes((48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefMode(self):
- assert decoder.decode(
- bytes((48, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testDefModeChunked(self):
- assert decoder.decode(
- bytes((48, 19, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefModeChunked(self):
- assert decoder.decode(
- bytes((48, 128, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- class SetOfDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.SetOf(componentType=univ.OctetString())
- self.s.setComponentByPosition(0, univ.OctetString('quick brown'))
- def testDefMode(self):
- assert decoder.decode(
- bytes((49, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- ) == (self.s, b'')
- def testIndefMode(self):
- assert decoder.decode(
- bytes((49, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0))
- ) == (self.s, b'')
- def testDefModeChunked(self):
- assert decoder.decode(
- bytes((49, 19, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110))
- ) == (self.s, b'')
- def testIndefModeChunked(self):
- assert decoder.decode(
- bytes((49, 128, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 0, 0))
- ) == (self.s, b'')
- def testSchemalessDecoder(self):
- assert decoder.decode(
- bytes((49, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=univ.SetOf()
- ) == (self.s, b'')
- class SetOfDecoderWithSchemaTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.SetOf(componentType=univ.OctetString())
- self.s.setComponentByPosition(0, univ.OctetString('quick brown'))
- def testDefMode(self):
- assert decoder.decode(
- bytes((49, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefMode(self):
- assert decoder.decode(
- bytes((49, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testDefModeChunked(self):
- assert decoder.decode(
- bytes((49, 19, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefModeChunked(self):
- assert decoder.decode(
- bytes((49, 128, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- class SequenceDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('place-holder', univ.Null(b'')),
- namedtype.NamedType('first-name', univ.OctetString(b'')),
- namedtype.NamedType('age', univ.Integer(33))
- )
- )
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testWithOptionalAndDefaultedDefMode(self):
- assert decoder.decode(
- bytes((48, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefMode(self):
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefModeChunked(self):
- assert decoder.decode(
- bytes(
- (48, 24, 5, 0, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 2, 1, 1))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefModeSubst(self):
- assert decoder.decode(
- bytes((48, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), b'')
- def testWithOptionalAndDefaultedDefModeSubstV04(self):
- assert decoder.decode(
- bytes((48, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), b'')
- def testWithOptionalAndDefaultedIndefModeSubst(self):
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), b'')
- def testWithOptionalAndDefaultedIndefModeSubstV04(self):
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), b'')
- def testTagFormat(self):
- try:
- decoder.decode(
- bytes((16, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- class SequenceDecoderWithSchemaTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('place-holder', univ.Null(b'')),
- namedtype.OptionalNamedType('first-name', univ.OctetString()),
- namedtype.DefaultedNamedType('age', univ.Integer(33)),
- )
- )
- def __init(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- def __initWithOptional(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- def __initWithDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def __initWithOptionalAndDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testDefMode(self):
- self.__init()
- assert decoder.decode(
- bytes((48, 2, 5, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefMode(self):
- self.__init()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testDefModeChunked(self):
- self.__init()
- assert decoder.decode(
- bytes((48, 2, 5, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefModeChunked(self):
- self.__init()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalDefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((48, 15, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionaIndefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 0, 0)),
- asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalDefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((48, 21, 5, 0, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110)),
- asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalIndefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0,
- 0, 0, 0)),
- asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedDefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((48, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedIndefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedDefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((48, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedIndefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((48, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)),
- asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1,
- 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes(
- (48, 24, 5, 0, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 2, 1, 1)),
- asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((48, 128, 5, 0, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0,
- 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- class SequenceDecoderWithUntaggedOpenTypesTestCase(BaseTestCase):
- def setUp(self):
- openType = opentype.OpenType(
- 'id',
- {1: univ.Integer(),
- 2: univ.OctetString()}
- )
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('id', univ.Integer()),
- namedtype.NamedType('blob', univ.Any(), openType=openType)
- )
- )
- def testDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 6, 2, 1, 1, 2, 1, 12)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 1
- assert s[1] == 12
- def testDecodeOpenTypesChoiceTwo(self):
- s, r = decoder.decode(
- bytes((48, 16, 2, 1, 2, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 2
- assert s[1] == univ.OctetString('quick brown')
- def testDecodeOpenTypesUnknownType(self):
- try:
- s, r = decoder.decode(
- bytes((48, 6, 2, 1, 2, 6, 1, 39)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- except error.PyAsn1Error:
- pass
- else:
- assert False, 'unknown open type tolerated'
- def testDecodeOpenTypesUnknownId(self):
- s, r = decoder.decode(
- bytes((48, 6, 2, 1, 3, 6, 1, 39)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 3
- assert s[1] == univ.OctetString(hexValue='060127')
- def testDontDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 6, 2, 1, 1, 2, 1, 12)), asn1Spec=self.s
- )
- assert not r
- assert s[0] == 1
- assert s[1] == bytes((2, 1, 12))
- def testDontDecodeOpenTypesChoiceTwo(self):
- s, r = decoder.decode(
- bytes((48, 16, 2, 1, 2, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- )
- assert not r
- assert s[0] == 2
- assert s[1] == bytes((4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- class SequenceDecoderWithImplicitlyTaggedOpenTypesTestCase(BaseTestCase):
- def setUp(self):
- openType = opentype.OpenType(
- 'id',
- {1: univ.Integer(),
- 2: univ.OctetString()}
- )
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('id', univ.Integer()),
- namedtype.NamedType(
- 'blob', univ.Any().subtype(implicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 3)), openType=openType
- )
- )
- )
- def testDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 8, 2, 1, 1, 131, 3, 2, 1, 12)), asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 1
- assert s[1] == 12
- def testDecodeOpenTypesUnknownId(self):
- s, r = decoder.decode(
- bytes((48, 8, 2, 1, 3, 131, 3, 2, 1, 12)), asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 3
- assert s[1] == univ.OctetString(hexValue='02010C')
- class SequenceDecoderWithExplicitlyTaggedOpenTypesTestCase(BaseTestCase):
- def setUp(self):
- openType = opentype.OpenType(
- 'id',
- {1: univ.Integer(),
- 2: univ.OctetString()}
- )
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('id', univ.Integer()),
- namedtype.NamedType(
- 'blob', univ.Any().subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 3)), openType=openType
- )
- )
- )
- def testDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 8, 2, 1, 1, 163, 3, 2, 1, 12)), asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 1
- assert s[1] == 12
- def testDecodeOpenTypesUnknownId(self):
- s, r = decoder.decode(
- bytes((48, 8, 2, 1, 3, 163, 3, 2, 1, 12)), asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 3
- assert s[1] == univ.OctetString(hexValue='02010C')
- class SequenceDecoderWithUnaggedSetOfOpenTypesTestCase(BaseTestCase):
- def setUp(self):
- openType = opentype.OpenType(
- 'id',
- {1: univ.Integer(),
- 2: univ.OctetString()}
- )
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('id', univ.Integer()),
- namedtype.NamedType('blob', univ.SetOf(componentType=univ.Any()),
- openType=openType)
- )
- )
- def testDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 8, 2, 1, 1, 49, 3, 2, 1, 12)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 1
- assert s[1][0] == 12
- def testDecodeOpenTypesChoiceTwo(self):
- s, r = decoder.decode(
- bytes((48, 18, 2, 1, 2, 49, 13, 4, 11, 113, 117, 105, 99,
- 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 2
- assert s[1][0] == univ.OctetString('quick brown')
- def testDecodeOpenTypesUnknownType(self):
- try:
- s, r = decoder.decode(
- bytes((48, 6, 2, 1, 2, 6, 1, 39)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- except error.PyAsn1Error:
- pass
- else:
- assert False, 'unknown open type tolerated'
- def testDecodeOpenTypesUnknownId(self):
- s, r = decoder.decode(
- bytes((48, 8, 2, 1, 3, 49, 3, 2, 1, 12)), asn1Spec=self.s,
- decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 3
- assert s[1][0] == univ.OctetString(hexValue='02010c')
- def testDontDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 8, 2, 1, 1, 49, 3, 2, 1, 12)), asn1Spec=self.s
- )
- assert not r
- assert s[0] == 1
- assert s[1][0] == bytes((2, 1, 12))
- def testDontDecodeOpenTypesChoiceTwo(self):
- s, r = decoder.decode(
- bytes((48, 18, 2, 1, 2, 49, 13, 4, 11, 113, 117, 105, 99,
- 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- )
- assert not r
- assert s[0] == 2
- assert s[1][0] == bytes((4, 11, 113, 117, 105, 99, 107, 32, 98, 114,
- 111, 119, 110))
- class SequenceDecoderWithImplicitlyTaggedSetOfOpenTypesTestCase(BaseTestCase):
- def setUp(self):
- openType = opentype.OpenType(
- 'id',
- {1: univ.Integer(),
- 2: univ.OctetString()}
- )
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('id', univ.Integer()),
- namedtype.NamedType(
- 'blob', univ.SetOf(
- componentType=univ.Any().subtype(
- implicitTag=tag.Tag(
- tag.tagClassContext, tag.tagFormatSimple, 3))),
- openType=openType
- )
- )
- )
- def testDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 10, 2, 1, 1, 49, 5, 131, 3, 2, 1, 12)),
- asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 1
- assert s[1][0] == 12
- def testDecodeOpenTypesUnknownId(self):
- s, r = decoder.decode(
- bytes((48, 10, 2, 1, 3, 49, 5, 131, 3, 2, 1, 12)),
- asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 3
- assert s[1][0] == univ.OctetString(hexValue='02010C')
- class SequenceDecoderWithExplicitlyTaggedSetOfOpenTypesTestCase(BaseTestCase):
- def setUp(self):
- openType = opentype.OpenType(
- 'id',
- {1: univ.Integer(),
- 2: univ.OctetString()}
- )
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('id', univ.Integer()),
- namedtype.NamedType(
- 'blob', univ.SetOf(
- componentType=univ.Any().subtype(
- explicitTag=tag.Tag(
- tag.tagClassContext, tag.tagFormatSimple, 3))),
- openType=openType
- )
- )
- )
- def testDecodeOpenTypesChoiceOne(self):
- s, r = decoder.decode(
- bytes((48, 10, 2, 1, 1, 49, 5, 131, 3, 2, 1, 12)),
- asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 1
- assert s[1][0] == 12
- def testDecodeOpenTypesUnknownId(self):
- s, r = decoder.decode(
- bytes( (48, 10, 2, 1, 3, 49, 5, 131, 3, 2, 1, 12)),
- asn1Spec=self.s, decodeOpenTypes=True
- )
- assert not r
- assert s[0] == 3
- assert s[1][0] == univ.OctetString(hexValue='02010C')
- class SetDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Set(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('place-holder', univ.Null(b'')),
- namedtype.NamedType('first-name', univ.OctetString(b'')),
- namedtype.NamedType('age', univ.Integer(33))
- )
- )
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testWithOptionalAndDefaultedDefMode(self):
- assert decoder.decode(
- bytes((49, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefMode(self):
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefModeChunked(self):
- assert decoder.decode(
- bytes(
- (49, 24, 5, 0, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 2, 1, 1))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0))
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefModeSubst(self):
- assert decoder.decode(
- bytes((49, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), b'')
- def testWithOptionalAndDefaultedDefModeSubstV04(self):
- assert decoder.decode(
- bytes((49, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), b'')
- def testWithOptionalAndDefaultedIndefModeSubst(self):
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), b'')
- def testWithOptionalAndDefaultedIndefModeSubstV04(self):
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), b'')
- def testTagFormat(self):
- try:
- decoder.decode(
- bytes((16, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- class SetDecoderWithSchemaTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Set(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('place-holder', univ.Null(b'')),
- namedtype.OptionalNamedType('first-name', univ.OctetString()),
- namedtype.DefaultedNamedType('age', univ.Integer(33)),
- )
- )
- def __init(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- def __initWithOptional(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- def __initWithDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def __initWithOptionalAndDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testDefMode(self):
- self.__init()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefMode(self):
- self.__init()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testDefModeChunked(self):
- self.__init()
- assert decoder.decode(
- bytes((49, 2, 5, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testIndefModeChunked(self):
- self.__init()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalDefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((49, 15, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalIndefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalDefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((49, 21, 5, 0, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalIndefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedDefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((49, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedIndefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedDefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((49, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithDefaultedIndefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((49, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefModeReordered(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((49, 18, 2, 1, 1, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 5, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefModeReordered(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((49, 128, 2, 1, 1, 5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedDefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((49, 24, 5, 0, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- bytes((49, 128, 5, 0, 36, 128, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- class SequenceOfWithExpTaggedOctetStringDecoder(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.SequenceOf(
- componentType=univ.OctetString().subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 3))
- )
- self.s.setComponentByPosition(0, 'q')
- self.s2 = univ.SequenceOf()
- def testDefModeSchema(self):
- s, r = decoder.decode(bytes((48, 5, 163, 3, 4, 1, 113)), asn1Spec=self.s)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testIndefModeSchema(self):
- s, r = decoder.decode(bytes((48, 128, 163, 128, 4, 1, 113, 0, 0, 0, 0)), asn1Spec=self.s)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testDefModeNoComponent(self):
- s, r = decoder.decode(bytes((48, 5, 163, 3, 4, 1, 113)), asn1Spec=self.s2)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testIndefModeNoComponent(self):
- s, r = decoder.decode(bytes((48, 128, 163, 128, 4, 1, 113, 0, 0, 0, 0)), asn1Spec=self.s2)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testDefModeSchemaless(self):
- s, r = decoder.decode(bytes((48, 5, 163, 3, 4, 1, 113)))
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testIndefModeSchemaless(self):
- s, r = decoder.decode(bytes((48, 128, 163, 128, 4, 1, 113, 0, 0, 0, 0)))
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- class SequenceWithExpTaggedOctetStringDecoder(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType(
- 'x', univ.OctetString().subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 3))
- )
- )
- )
- self.s.setComponentByPosition(0, 'q')
- self.s2 = univ.Sequence()
- def testDefModeSchema(self):
- s, r = decoder.decode(bytes((48, 5, 163, 3, 4, 1, 113)), asn1Spec=self.s)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testIndefModeSchema(self):
- s, r = decoder.decode(bytes((48, 128, 163, 128, 4, 1, 113, 0, 0, 0, 0)), asn1Spec=self.s)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testDefModeNoComponent(self):
- s, r = decoder.decode(bytes((48, 5, 163, 3, 4, 1, 113)), asn1Spec=self.s2)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testIndefModeNoComponent(self):
- s, r = decoder.decode(bytes((48, 128, 163, 128, 4, 1, 113, 0, 0, 0, 0)), asn1Spec=self.s2)
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testDefModeSchemaless(self):
- s, r = decoder.decode(bytes((48, 5, 163, 3, 4, 1, 113)))
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- def testIndefModeSchemaless(self):
- s, r = decoder.decode(bytes((48, 128, 163, 128, 4, 1, 113, 0, 0, 0, 0)))
- assert not r
- assert s == self.s
- assert s.tagSet == self.s.tagSet
- class ChoiceDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Choice(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('place-holder', univ.Null(b'')),
- namedtype.NamedType('number', univ.Integer(0)),
- namedtype.NamedType('string', univ.OctetString())
- )
- )
- def testBySpec(self):
- self.s.setComponentByPosition(0, univ.Null(b''))
- assert decoder.decode(
- bytes((5, 0)), asn1Spec=self.s
- ) == (self.s, b'')
- def testWithoutSpec(self):
- self.s.setComponentByPosition(0, univ.Null(b''))
- assert decoder.decode(bytes((5, 0))) == (self.s, b'')
- assert decoder.decode(bytes((5, 0))) == (univ.Null(b''), b'')
- def testUndefLength(self):
- self.s.setComponentByPosition(2, univ.OctetString('abcdefgh'))
- assert decoder.decode(bytes((36, 128, 4, 3, 97, 98, 99, 4, 3, 100, 101, 102, 4, 2, 103, 104, 0, 0)),
- asn1Spec=self.s) == (self.s, b'')
- def testExplicitTag(self):
- s = self.s.subtype(explicitTag=tag.Tag(tag.tagClassContext,
- tag.tagFormatConstructed, 4))
- s.setComponentByPosition(0, univ.Null(b''))
- assert decoder.decode(bytes((164, 2, 5, 0)), asn1Spec=s) == (s, b'')
- def testExplicitTagUndefLength(self):
- s = self.s.subtype(explicitTag=tag.Tag(tag.tagClassContext,
- tag.tagFormatConstructed, 4))
- s.setComponentByPosition(0, univ.Null(b''))
- assert decoder.decode(bytes((164, 128, 5, 0, 0, 0)), asn1Spec=s) == (s, b'')
- class AnyDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Any()
- def testByUntagged(self):
- assert decoder.decode(
- bytes((4, 3, 102, 111, 120)), asn1Spec=self.s
- ) == (univ.Any('\004\003fox'), b'')
- def testTaggedEx(self):
- s = univ.Any('\004\003fox').subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(bytes((164, 5, 4, 3, 102, 111, 120)), asn1Spec=s) == (s, b'')
- def testTaggedIm(self):
- s = univ.Any('\004\003fox').subtype(implicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(bytes((132, 5, 4, 3, 102, 111, 120)), asn1Spec=s) == (s, b'')
- def testByUntaggedIndefMode(self):
- assert decoder.decode(
- bytes((4, 3, 102, 111, 120)), asn1Spec=self.s
- ) == (univ.Any('\004\003fox'), b'')
- def testTaggedExIndefMode(self):
- s = univ.Any('\004\003fox').subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(bytes((164, 128, 4, 3, 102, 111, 120, 0, 0)), asn1Spec=s) == (s, b'')
- def testTaggedImIndefMode(self):
- s = univ.Any('\004\003fox').subtype(implicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(bytes((164, 128, 4, 3, 102, 111, 120, 0, 0)), asn1Spec=s) == (s, b'')
- def testByUntaggedSubst(self):
- assert decoder.decode(
- bytes((4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((4, 3, 102, 111, 120)), b'')
- def testByUntaggedSubstV04(self):
- assert decoder.decode(
- bytes((4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((4, 3, 102, 111, 120)), b'')
- def testTaggedExSubst(self):
- assert decoder.decode(
- bytes((164, 5, 4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (bytes((164, 5, 4, 3, 102, 111, 120)), b'')
- def testTaggedExSubstV04(self):
- assert decoder.decode(
- bytes((164, 5, 4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (bytes((164, 5, 4, 3, 102, 111, 120)), b'')
- class EndOfOctetsTestCase(BaseTestCase):
- def testUnexpectedEoo(self):
- try:
- decoder.decode(bytes((0, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'end-of-contents octets accepted at top level'
- def testExpectedEoo(self):
- result, remainder = decoder.decode(bytes((0, 0)), allowEoo=True)
- assert eoo.endOfOctets.isSameTypeWith(result) and result == eoo.endOfOctets and result is eoo.endOfOctets
- assert remainder == b''
- def testDefiniteNoEoo(self):
- try:
- decoder.decode(bytes((0x23, 0x02, 0x00, 0x00)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'end-of-contents octets accepted inside definite-length encoding'
- def testIndefiniteEoo(self):
- result, remainder = decoder.decode(bytes((0x23, 0x80, 0x00, 0x00)))
- assert result == () and remainder == b'', 'incorrect decoding of indefinite length end-of-octets'
- def testNoLongFormEoo(self):
- try:
- decoder.decode(bytes((0x23, 0x80, 0x00, 0x81, 0x00)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'end-of-contents octets accepted with invalid long-form length'
- def testNoConstructedEoo(self):
- try:
- decoder.decode(bytes((0x23, 0x80, 0x20, 0x00)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'end-of-contents octets accepted with invalid constructed encoding'
- def testNoEooData(self):
- try:
- decoder.decode(bytes((0x23, 0x80, 0x00, 0x01, 0x00)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'end-of-contents octets accepted with unexpected data'
- class NonStringDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('place-holder', univ.Null(b'')),
- namedtype.NamedType('first-name', univ.OctetString(b'')),
- namedtype.NamedType('age', univ.Integer(33))
- )
- )
- self.s.setComponentByPosition(0, univ.Null(b''))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- self.substrate = bytes([48, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1])
- def testOctetString(self):
- s = list(decoder.StreamingDecoder(
- univ.OctetString(self.substrate), asn1Spec=self.s))
- assert [self.s] == s
- def testAny(self):
- s = list(decoder.StreamingDecoder(
- univ.Any(self.substrate), asn1Spec=self.s))
- assert [self.s] == s
- class ErrorOnDecodingTestCase(BaseTestCase):
- def testErrorCondition(self):
- decode = decoder.SingleItemDecoder(
- tagMap=decoder.TAG_MAP, typeMap=decoder.TYPE_MAP)
- substrate = bytes((00, 1, 2))
- stream = streaming.asSeekableStream(substrate)
- try:
- asn1Object = next(decode(stream))
- except error.PyAsn1Error:
- pass
- else:
- assert False, 'Unexpected decoder result %r' % (asn1Object,)
- def testRawDump(self):
- substrate = bytes((31, 8, 2, 1, 1, 131, 3, 2, 1, 12))
- stream = streaming.asSeekableStream(substrate)
- class SingleItemEncoder(decoder.SingleItemDecoder):
- defaultErrorState = decoder.stDumpRawValue
- class StreamingDecoder(decoder.StreamingDecoder):
- SINGLE_ITEM_DECODER = SingleItemEncoder
- class OneShotDecoder(decoder.Decoder):
- STREAMING_DECODER = StreamingDecoder
- d = OneShotDecoder()
- asn1Object, rest = d(stream)
- assert isinstance(asn1Object, univ.Any), (
- 'Unexpected raw dump type %r' % (asn1Object,))
- assert asn1Object.asNumbers() == (31, 8, 2, 1, 1), (
- 'Unexpected raw dump value %r' % (asn1Object,))
- assert rest == bytes((131, 3, 2, 1, 12)), (
- 'Unexpected rest of substrate after raw dump %r' % rest)
- class BinaryFileTestCase(BaseTestCase):
- """Assure that decode works on open binary files."""
- def testOneObject(self):
- _, path = tempfile.mkstemp()
- try:
- with open(path, "wb") as out:
- out.write(bytes((2, 1, 12)))
- with open(path, "rb") as source:
- values = list(decoder.StreamingDecoder(source))
- assert values == [12]
- finally:
- os.remove(path)
- def testMoreObjects(self):
- _, path = tempfile.mkstemp()
- try:
- with open(path, "wb") as out:
- out.write(bytes((2, 1, 12, 35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)))
- with open(path, "rb") as source:
- values = list(decoder.StreamingDecoder(source))
- assert values == [12, (1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1)]
- finally:
- os.remove(path)
- def testInvalidFileContent(self):
- _, path = tempfile.mkstemp()
- try:
- with open(path, "wb") as out:
- out.write(bytes((2, 1, 12, 35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0, 7)))
- with open(path, "rb") as source:
- list(decoder.StreamingDecoder(source))
- except error.EndOfStreamError:
- pass
- finally:
- os.remove(path)
- class BytesIOTestCase(BaseTestCase):
- def testRead(self):
- source = bytes((2, 1, 12, 35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0))
- stream = io.BytesIO(source)
- values = list(decoder.StreamingDecoder(stream))
- assert values == [12, (1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1)]
- class UnicodeTestCase(BaseTestCase):
- def testFail(self):
- # This ensures that str objects cannot be parsed.
- source = bytes((2, 1, 12, 35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)).decode("latin-1")
- try:
- next(decoder.StreamingDecoder(source))
- except error.UnsupportedSubstrateError:
- pass
- else:
- assert False, 'Tolerated parsing broken unicode strings'
- class RestartableDecoderTestCase(BaseTestCase):
- class NonBlockingStream(io.BytesIO):
- block = False
- def read(self, size=-1):
- self.block = not self.block
- if self.block:
- return # this is what non-blocking streams sometimes do
- return io.BytesIO.read(self, size)
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.SequenceOf(componentType=univ.OctetString())
- self.s.setComponentByPosition(0, univ.OctetString('quick brown'))
- source = bytes(
- (48, 26,
- 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110,
- 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- self.stream = self.NonBlockingStream(source)
- def testPartialReadingFromNonBlockingStream(self):
- iterator = iter(decoder.StreamingDecoder(self.stream, asn1Spec=self.s))
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' not in res.context
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' not in res.context
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' in res.context
- assert isinstance(res.context['asn1Object'], univ.SequenceOf)
- assert res.context['asn1Object'].isValue
- assert len(res.context['asn1Object']) == 0
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' in res.context
- assert isinstance(res.context['asn1Object'], univ.SequenceOf)
- assert res.context['asn1Object'].isValue
- assert len(res.context['asn1Object']) == 0
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' in res.context
- assert isinstance(res.context['asn1Object'], univ.SequenceOf)
- assert res.context['asn1Object'].isValue
- assert len(res.context['asn1Object']) == 0
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' in res.context
- assert isinstance(res.context['asn1Object'], univ.SequenceOf)
- assert res.context['asn1Object'].isValue
- assert len(res.context['asn1Object']) == 1
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' in res.context
- assert isinstance(res.context['asn1Object'], univ.SequenceOf)
- assert res.context['asn1Object'].isValue
- assert len(res.context['asn1Object']) == 1
- res = next(iterator)
- assert isinstance(res, error.SubstrateUnderrunError)
- assert 'asn1Object' in res.context
- assert isinstance(res.context['asn1Object'], univ.SequenceOf)
- assert res.context['asn1Object'].isValue
- assert len(res.context['asn1Object']) == 1
- res = next(iterator)
- assert isinstance(res, univ.SequenceOf)
- assert res.isValue
- assert len(res) == 2
- try:
- next(iterator)
- except StopIteration:
- pass
- else:
- assert False, 'End of stream not raised'
- class CompressedFilesTestCase(BaseTestCase):
- def testGzip(self):
- _, path = tempfile.mkstemp(suffix=".gz")
- try:
- with gzip.open(path, "wb") as out:
- out.write(bytes((2, 1, 12, 35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)))
- with gzip.open(path, "rb") as source:
- values = list(decoder.StreamingDecoder(source))
- assert values == [12, (1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1)]
- finally:
- os.remove(path)
- def testZipfile(self):
- # It is a seekable stream.
- _, path = tempfile.mkstemp(suffix=".zip")
- try:
- with zipfile.ZipFile(path, "w") as myzip:
- myzip.writestr("data", bytes((2, 1, 12, 35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)))
- with zipfile.ZipFile(path, "r") as myzip:
- with myzip.open("data", "r") as source:
- values = list(decoder.StreamingDecoder(source))
- assert values == [12, (1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1)]
- finally:
- os.remove(path)
- def testZipfileMany(self):
- _, path = tempfile.mkstemp(suffix=".zip")
- try:
- with zipfile.ZipFile(path, "w") as myzip:
- #for i in range(100):
- myzip.writestr("data", bytes((2, 1, 12, 35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)) * 1000)
- with zipfile.ZipFile(path, "r") as myzip:
- with myzip.open("data", "r") as source:
- values = list(decoder.StreamingDecoder(source))
- assert values == [12, (1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1)] * 1000
- finally:
- os.remove(path)
- class NonStreamingCompatibilityTestCase(BaseTestCase):
- def setUp(self):
- from pyasn1 import debug
- BaseTestCase.setUp(self)
- debug.setLogger(None) # undo logger setup from BaseTestCase to work around unrelated issue
- def testPartialDecodeWithCustomSubstrateFun(self):
- snmp_req_substrate = bytes((
- 0x30, 0x22, 0x02, 0x01, 0x01, 0x04, 0x06, 0x70, 0x75, 0x62, 0x6c, 0x69, 0x63, 0xa0, 0x15, 0x02, 0x04, 0x69,
- 0x30, 0xdb, 0xeb, 0x02, 0x01, 0x00, 0x02, 0x01, 0x00, 0x30, 0x07, 0x30, 0x05, 0x06, 0x01, 0x01, 0x05, 0x00))
- seq, next_substrate = decoder.decode(
- snmp_req_substrate, asn1Spec=univ.Sequence(),
- recursiveFlag=False, substrateFun=lambda a, b, c: (a, b[:c])
- )
- assert seq.isSameTypeWith(univ.Sequence)
- assert next_substrate == snmp_req_substrate[2:]
- version, next_substrate = decoder.decode(
- next_substrate, asn1Spec=univ.Integer(), recursiveFlag=False,
- substrateFun=lambda a, b, c: (a, b[:c])
- )
- assert version == 1
- def testPartialDecodeWithDefaultSubstrateFun(self):
- substrate = bytes((
- 0x04, 0x0e, 0x30, 0x0c, 0x06, 0x0a, 0x2b, 0x06, 0x01, 0x04, 0x01, 0x82, 0x37, 0x3c, 0x03, 0x02
- ))
- result, rest = decoder.decode(substrate, recursiveFlag=False)
- assert result.isSameTypeWith(univ.OctetString)
- assert rest == substrate[2:]
- def testPropagateUserException(self):
- substrate = io.BytesIO(bytes((0x04, 0x00)))
- def userSubstrateFun(_asn1Object, _substrate, _length, _options):
- raise TypeError("error inside user function")
- try:
- decoder.decode(substrate, asn1Spec=univ.OctetString, substrateFun=userSubstrateFun)
- except TypeError as exc:
- assert str(exc) == "error inside user function"
- else:
- raise AssertionError("decode() must not hide TypeError from inside user provided callback")
- suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__])
- if __name__ == '__main__':
- unittest.TextTestRunner(verbosity=2).run(suite)
|