12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973 |
- #
- # 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.compat.octets import ints2octs, str2octs, null
- from pyasn1 import error
- class LargeTagDecoderTestCase(BaseTestCase):
- def testLargeTag(self):
- assert decoder.decode(ints2octs((127, 141, 245, 182, 253, 47, 3, 2, 1, 1))) == (1, null)
- def testLongTag(self):
- assert decoder.decode(ints2octs((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(ints2octs((0x9f, 0x80, 0x00, 0x02, 0x01, 0x02)), asn1Spec=integer) == decoder.decode(
- ints2octs((0x9f, 0x00, 0x02, 0x01, 0x02)), asn1Spec=integer)
- class DecoderCacheTestCase(BaseTestCase):
- def testCache(self):
- assert decoder.decode(ints2octs((0x1f, 2, 1, 0))) == decoder.decode(ints2octs((0x1f, 2, 1, 0)))
- class IntegerDecoderTestCase(BaseTestCase):
- def testPosInt(self):
- assert decoder.decode(ints2octs((2, 1, 12))) == (12, null)
- def testNegInt(self):
- assert decoder.decode(ints2octs((2, 1, 244))) == (-12, null)
- def testZero(self):
- assert decoder.decode(ints2octs((2, 0))) == (0, null)
- def testZeroLong(self):
- assert decoder.decode(ints2octs((2, 1, 0))) == (0, null)
- def testMinusOne(self):
- assert decoder.decode(ints2octs((2, 1, 255))) == (-1, null)
- def testPosLong(self):
- assert decoder.decode(
- ints2octs((2, 9, 0, 255, 255, 255, 255, 255, 255, 255, 255))
- ) == (0xffffffffffffffff, null)
- def testNegLong(self):
- assert decoder.decode(
- ints2octs((2, 9, 255, 0, 0, 0, 0, 0, 0, 0, 1))
- ) == (-0xffffffffffffffff, null)
- def testSpec(self):
- try:
- decoder.decode(
- ints2octs((2, 1, 12)), asn1Spec=univ.Null()
- ) == (12, null)
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong asn1Spec worked out'
- assert decoder.decode(
- ints2octs((2, 1, 12)), asn1Spec=univ.Integer()
- ) == (12, null)
- def testTagFormat(self):
- try:
- decoder.decode(ints2octs((34, 1, 12)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- class BooleanDecoderTestCase(BaseTestCase):
- def testTrue(self):
- assert decoder.decode(ints2octs((1, 1, 1))) == (1, null)
- def testTrueNeg(self):
- assert decoder.decode(ints2octs((1, 1, 255))) == (1, null)
- def testExtraTrue(self):
- assert decoder.decode(ints2octs((1, 1, 1, 0, 120, 50, 50))) == (1, ints2octs((0, 120, 50, 50)))
- def testFalse(self):
- assert decoder.decode(ints2octs((1, 1, 0))) == (0, null)
- def testTagFormat(self):
- try:
- decoder.decode(ints2octs((33, 1, 1)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- class BitStringDecoderTestCase(BaseTestCase):
- def testDefMode(self):
- assert decoder.decode(
- ints2octs((3, 3, 1, 169, 138))
- ) == ((1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1), null)
- def testIndefMode(self):
- assert decoder.decode(
- ints2octs((3, 3, 1, 169, 138))
- ) == ((1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1), null)
- def testDefModeChunked(self):
- assert decoder.decode(
- ints2octs((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), null)
- def testIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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), null)
- def testDefModeChunkedSubst(self):
- assert decoder.decode(
- ints2octs((35, 8, 3, 2, 0, 169, 3, 2, 1, 138)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (ints2octs((3, 2, 0, 169, 3, 2, 1, 138)), str2octs(''))
- def testDefModeChunkedSubstV04(self):
- assert decoder.decode(
- ints2octs((35, 8, 3, 2, 0, 169, 3, 2, 1, 138)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (ints2octs((3, 2, 0, 169, 3, 2, 1, 138)), str2octs(''))
- def testIndefModeChunkedSubst(self):
- assert decoder.decode(
- ints2octs((35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)),
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (ints2octs((3, 2, 0, 169, 3, 2, 1, 138, 0, 0)), str2octs(''))
- def testIndefModeChunkedSubstV04(self):
- assert decoder.decode(
- ints2octs((35, 128, 3, 2, 0, 169, 3, 2, 1, 138, 0, 0)),
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (ints2octs((3, 2, 0, 169, 3, 2, 1, 138, 0, 0)), str2octs(''))
- def testTypeChecking(self):
- try:
- decoder.decode(ints2octs((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(
- ints2octs((4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120))
- ) == (str2octs('Quick brown fox'), null)
- def testIndefMode(self):
- assert decoder.decode(
- ints2octs((36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0, 0))
- ) == (str2octs('Quick brown fox'), null)
- def testDefModeChunked(self):
- assert decoder.decode(
- ints2octs(
- (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))
- ) == (str2octs('Quick brown fox'), null)
- def testIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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))
- ) == (str2octs('Quick brown fox'), null)
- def testDefModeChunkedSubst(self):
- assert decoder.decode(
- ints2octs(
- (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)
- ) == (ints2octs((4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120)), str2octs(''))
- def testDefModeChunkedSubstV04(self):
- assert decoder.decode(
- ints2octs(
- (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:])
- ) == (ints2octs((4, 4, 81, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 4, 111, 119, 110, 32, 4, 3, 102, 111, 120)), str2octs(''))
- def testIndefModeChunkedSubst(self):
- assert decoder.decode(
- ints2octs((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)
- ) == (ints2octs(
- (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)), str2octs(''))
- def testIndefModeChunkedSubstV04(self):
- assert decoder.decode(
- ints2octs((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:])
- ) == (ints2octs(
- (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)), str2octs(''))
- 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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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)
- ) == (ints2octs((4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120)), str2octs(''))
- def testDefModeSubstV04(self):
- assert decoder.decode(
- ints2octs((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:])
- ) == (ints2octs((4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120)), str2octs(''))
- def testIndefModeSubst(self):
- assert decoder.decode(
- ints2octs((
- 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)
- ) == (ints2octs(
- (36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0, 0, 0, 0)), str2octs(''))
- def testIndefModeSubstV04(self):
- assert decoder.decode(
- ints2octs((
- 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:])
- ) == (ints2octs(
- (36, 128, 4, 15, 81, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 32, 102, 111, 120, 0, 0, 0, 0)), str2octs(''))
- class NullDecoderTestCase(BaseTestCase):
- def testNull(self):
- assert decoder.decode(ints2octs((5, 0))) == (null, null)
- def testTagFormat(self):
- try:
- decoder.decode(ints2octs((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(
- ints2octs((6, 6, 43, 6, 0, 191, 255, 126))
- ) == ((1, 3, 6, 0, 0xffffe), null)
- def testEdge1(self):
- assert decoder.decode(
- ints2octs((6, 1, 39))
- ) == ((0, 39), null)
- def testEdge2(self):
- assert decoder.decode(
- ints2octs((6, 1, 79))
- ) == ((1, 39), null)
- def testEdge3(self):
- assert decoder.decode(
- ints2octs((6, 1, 120))
- ) == ((2, 40), null)
- def testEdge4(self):
- assert decoder.decode(
- ints2octs((6, 5, 0x90, 0x80, 0x80, 0x80, 0x4F))
- ) == ((2, 0xffffffff), null)
- def testEdge5(self):
- assert decoder.decode(
- ints2octs((6, 1, 0x7F))
- ) == ((2, 47), null)
- def testEdge6(self):
- assert decoder.decode(
- ints2octs((6, 2, 0x81, 0x00))
- ) == ((2, 48), null)
- def testEdge7(self):
- assert decoder.decode(
- ints2octs((6, 3, 0x81, 0x34, 0x03))
- ) == ((2, 100, 3), null)
- def testEdge8(self):
- assert decoder.decode(
- ints2octs((6, 2, 133, 0))
- ) == ((2, 560), null)
- def testEdge9(self):
- assert decoder.decode(
- ints2octs((6, 4, 0x88, 0x84, 0x87, 0x02))
- ) == ((2, 16843570), null)
- def testNonLeading0x80(self):
- assert decoder.decode(
- ints2octs((6, 5, 85, 4, 129, 128, 0)),
- ) == ((2, 5, 4, 16384), null)
- def testLeading0x80Case1(self):
- try:
- decoder.decode(
- ints2octs((6, 5, 85, 4, 128, 129, 0))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testLeading0x80Case2(self):
- try:
- decoder.decode(
- ints2octs((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(
- ints2octs((6, 2, 0x80, 1))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testLeading0x80Case4(self):
- try:
- decoder.decode(
- ints2octs((6, 2, 0x80, 0x7F))
- )
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'Leading 0x80 tolerated'
- def testTagFormat(self):
- try:
- decoder.decode(ints2octs((38, 1, 239)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- def testZeroLength(self):
- try:
- decoder.decode(ints2octs((6, 0, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'zero length tolerated'
- def testIndefiniteLength(self):
- try:
- decoder.decode(ints2octs((6, 128, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'indefinite length tolerated'
- def testReservedLength(self):
- try:
- decoder.decode(ints2octs((6, 255, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'reserved length tolerated'
- def testLarge1(self):
- assert decoder.decode(
- ints2octs((0x06, 0x11, 0x83, 0xC6, 0xDF, 0xD4, 0xCC, 0xB3, 0xFF, 0xFF, 0xFE, 0xF0, 0xB8, 0xD6, 0xB8, 0xCB, 0xE2, 0xB7, 0x17))
- ) == ((2, 18446744073709551535184467440737095), null)
- def testLarge2(self):
- assert decoder.decode(
- ints2octs((0x06, 0x13, 0x88, 0x37, 0x83, 0xC6, 0xDF, 0xD4, 0xCC, 0xB3, 0xFF, 0xFF, 0xFE, 0xF0, 0xB8, 0xD6, 0xB8, 0xCB, 0xE2, 0xB6, 0x47))
- ) == ((2, 999, 18446744073709551535184467440737095), null)
- class RealDecoderTestCase(BaseTestCase):
- def testChar(self):
- assert decoder.decode(
- ints2octs((9, 7, 3, 49, 50, 51, 69, 49, 49))
- ) == (univ.Real((123, 10, 11)), null)
- def testBin1(self): # check base = 2
- assert decoder.decode( # (0.5, 2, 0) encoded with base = 2
- ints2octs((9, 3, 128, 255, 1))
- ) == (univ.Real((1, 2, -1)), null)
- def testBin2(self): # check base = 2 and scale factor
- assert decoder.decode( # (3.25, 2, 0) encoded with base = 8
- ints2octs((9, 3, 148, 255, 13))
- ) == (univ.Real((26, 2, -3)), null)
- def testBin3(self): # check base = 16
- assert decoder.decode( # (0.00390625, 2, 0) encoded with base = 16
- ints2octs((9, 3, 160, 254, 1))
- ) == (univ.Real((1, 2, -8)), null)
- def testBin4(self): # check exponent = 0
- assert decoder.decode( # (1, 2, 0) encoded with base = 2
- ints2octs((9, 3, 128, 0, 1))
- ) == (univ.Real((1, 2, 0)), null)
- def testBin5(self): # case of 2 octs for exponent and negative exponent
- assert decoder.decode( # (3, 2, -1020) encoded with base = 16
- ints2octs((9, 4, 161, 255, 1, 3))
- ) == (univ.Real((3, 2, -1020)), null)
- # TODO: this requires Real type comparison fix
- # def testBin6(self):
- # assert decoder.decode(
- # ints2octs((9, 5, 162, 0, 255, 255, 1))
- # ) == (univ.Real((1, 2, 262140)), null)
- # def testBin7(self):
- # assert decoder.decode(
- # ints2octs((9, 7, 227, 4, 1, 35, 69, 103, 1))
- # ) == (univ.Real((-1, 2, 76354972)), null)
- def testPlusInf(self):
- assert decoder.decode(
- ints2octs((9, 1, 64))
- ) == (univ.Real('inf'), null)
- def testMinusInf(self):
- assert decoder.decode(
- ints2octs((9, 1, 65))
- ) == (univ.Real('-inf'), null)
- def testEmpty(self):
- assert decoder.decode(
- ints2octs((9, 0))
- ) == (univ.Real(0.0), null)
- def testTagFormat(self):
- try:
- decoder.decode(ints2octs((41, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'wrong tagFormat worked out'
- def testShortEncoding(self):
- try:
- decoder.decode(ints2octs((9, 1, 131)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'accepted too-short real'
- class UniversalStringDecoderTestCase(BaseTestCase):
- def testDecoder(self):
- assert decoder.decode(ints2octs((28, 12, 0, 0, 0, 97, 0, 0, 0, 98, 0, 0, 0, 99))) == (char.UniversalString(sys.version_info[0] >= 3 and 'abc' or unicode('abc')), null)
- class BMPStringDecoderTestCase(BaseTestCase):
- def testDecoder(self):
- assert decoder.decode(ints2octs((30, 6, 0, 97, 0, 98, 0, 99))) == (char.BMPString(sys.version_info[0] >= 3 and 'abc' or unicode('abc')), null)
- class UTF8StringDecoderTestCase(BaseTestCase):
- def testDecoder(self):
- assert decoder.decode(ints2octs((12, 3, 97, 98, 99))) == (char.UTF8String(sys.version_info[0] >= 3 and 'abc' or unicode('abc')), null)
- 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(
- ints2octs((48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- ) == (self.s, null)
- def testIndefMode(self):
- assert decoder.decode(
- ints2octs((48, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0))
- ) == (self.s, null)
- def testDefModeChunked(self):
- assert decoder.decode(
- ints2octs((48, 19, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110))
- ) == (self.s, null)
- def testIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testSchemalessDecoder(self):
- assert decoder.decode(
- ints2octs((48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=univ.SequenceOf()
- ) == (self.s, null)
- class ExpTaggedSequenceOfDecoderTestCase(BaseTestCase):
- def testWithSchema(self):
- s = univ.SequenceOf().subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatConstructed, 3))
- s2, r = decoder.decode(
- ints2octs((163, 15, 48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=s
- )
- assert not r
- assert s2 == [str2octs('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(
- ints2octs((163, 15, 48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- )
- assert not r
- assert s2 == [str2octs('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(
- ints2octs((48, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, null)
- def testIndefMode(self):
- assert decoder.decode(
- ints2octs((48, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testDefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- 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(
- ints2octs((49, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110))
- ) == (self.s, null)
- def testIndefMode(self):
- assert decoder.decode(
- ints2octs((49, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0))
- ) == (self.s, null)
- def testDefModeChunked(self):
- assert decoder.decode(
- ints2octs((49, 19, 36, 17, 4, 4, 113, 117, 105, 99, 4, 4, 107, 32, 98, 114, 4, 3, 111, 119, 110))
- ) == (self.s, null)
- def testIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testSchemalessDecoder(self):
- assert decoder.decode(
- ints2octs((49, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=univ.SetOf()
- ) == (self.s, null)
- 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(
- ints2octs((49, 13, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, null)
- def testIndefMode(self):
- assert decoder.decode(
- ints2octs((49, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testDefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- class SequenceDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Sequence(
- componentType=namedtype.NamedTypes(
- namedtype.NamedType('place-holder', univ.Null(null)),
- namedtype.NamedType('first-name', univ.OctetString(null)),
- namedtype.NamedType('age', univ.Integer(33))
- )
- )
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testWithOptionalAndDefaultedDefMode(self):
- assert decoder.decode(
- ints2octs((48, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1))
- ) == (self.s, null)
- def testWithOptionalAndDefaultedIndefMode(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedDefModeChunked(self):
- assert decoder.decode(
- ints2octs(
- (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, null)
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedDefModeSubst(self):
- assert decoder.decode(
- ints2octs((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)
- ) == (ints2octs((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), str2octs(''))
- def testWithOptionalAndDefaultedDefModeSubstV04(self):
- assert decoder.decode(
- ints2octs((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:])
- ) == (ints2octs((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), str2octs(''))
- def testWithOptionalAndDefaultedIndefModeSubst(self):
- assert decoder.decode(
- ints2octs((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)
- ) == (ints2octs(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), str2octs(''))
- def testWithOptionalAndDefaultedIndefModeSubstV04(self):
- assert decoder.decode(
- ints2octs((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:])
- ) == (ints2octs(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), str2octs(''))
- def testTagFormat(self):
- try:
- decoder.decode(
- ints2octs((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(null)),
- namedtype.OptionalNamedType('first-name', univ.OctetString()),
- namedtype.DefaultedNamedType('age', univ.Integer(33)),
- )
- )
- def __init(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- def __initWithOptional(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- def __initWithDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def __initWithOptionalAndDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testDefMode(self):
- self.__init()
- assert decoder.decode(
- ints2octs((48, 2, 5, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testIndefMode(self):
- self.__init()
- assert decoder.decode(
- ints2octs((48, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testDefModeChunked(self):
- self.__init()
- assert decoder.decode(
- ints2octs((48, 2, 5, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testIndefModeChunked(self):
- self.__init()
- assert decoder.decode(
- ints2octs((48, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithOptionalDefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((48, 15, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithOptionaIndefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalDefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalIndefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithDefaultedDefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((48, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithDefaultedIndefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((48, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithDefaultedDefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((48, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithDefaultedIndefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((48, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithOptionalAndDefaultedDefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedIndefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedDefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs(
- (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, null)
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- 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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((48, 6, 2, 1, 1, 2, 1, 12)), asn1Spec=self.s
- )
- assert not r
- assert s[0] == 1
- assert s[1] == ints2octs((2, 1, 12))
- def testDontDecodeOpenTypesChoiceTwo(self):
- s, r = decoder.decode(
- ints2octs((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] == ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((48, 8, 2, 1, 1, 49, 3, 2, 1, 12)), asn1Spec=self.s
- )
- assert not r
- assert s[0] == 1
- assert s[1][0] == ints2octs((2, 1, 12))
- def testDontDecodeOpenTypesChoiceTwo(self):
- s, r = decoder.decode(
- ints2octs((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] == ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs((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(
- ints2octs( (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(null)),
- namedtype.NamedType('first-name', univ.OctetString(null)),
- namedtype.NamedType('age', univ.Integer(33))
- )
- )
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testWithOptionalAndDefaultedDefMode(self):
- assert decoder.decode(
- ints2octs((49, 18, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1))
- ) == (self.s, null)
- def testWithOptionalAndDefaultedIndefMode(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedDefModeChunked(self):
- assert decoder.decode(
- ints2octs(
- (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, null)
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedDefModeSubst(self):
- assert decoder.decode(
- ints2octs((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)
- ) == (ints2octs((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), str2octs(''))
- def testWithOptionalAndDefaultedDefModeSubstV04(self):
- assert decoder.decode(
- ints2octs((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:])
- ) == (ints2octs((5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 2, 1, 1)), str2octs(''))
- def testWithOptionalAndDefaultedIndefModeSubst(self):
- assert decoder.decode(
- ints2octs((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)
- ) == (ints2octs(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), str2octs(''))
- def testWithOptionalAndDefaultedIndefModeSubstV04(self):
- assert decoder.decode(
- ints2octs((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:])
- ) == (ints2octs(
- (5, 0, 36, 128, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110, 0, 0, 2, 1, 1, 0, 0)), str2octs(''))
- def testTagFormat(self):
- try:
- decoder.decode(
- ints2octs((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(null)),
- namedtype.OptionalNamedType('first-name', univ.OctetString()),
- namedtype.DefaultedNamedType('age', univ.Integer(33)),
- )
- )
- def __init(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- def __initWithOptional(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- def __initWithDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def __initWithOptionalAndDefaulted(self):
- self.s.clear()
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- def testDefMode(self):
- self.__init()
- assert decoder.decode(
- ints2octs((49, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testIndefMode(self):
- self.__init()
- assert decoder.decode(
- ints2octs((49, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testDefModeChunked(self):
- self.__init()
- assert decoder.decode(
- ints2octs((49, 2, 5, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testIndefModeChunked(self):
- self.__init()
- assert decoder.decode(
- ints2octs((49, 128, 5, 0, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithOptionalDefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((49, 15, 5, 0, 4, 11, 113, 117, 105, 99, 107, 32, 98, 114, 111, 119, 110)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithOptionalIndefMode(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalDefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalIndefModeChunked(self):
- self.__initWithOptional()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithDefaultedDefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((49, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithDefaultedIndefMode(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((49, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithDefaultedDefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((49, 5, 5, 0, 2, 1, 1)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithDefaultedIndefModeChunked(self):
- self.__initWithDefaulted()
- assert decoder.decode(
- ints2octs((49, 128, 5, 0, 2, 1, 1, 0, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithOptionalAndDefaultedDefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedDefModeReordered(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedIndefMode(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedIndefModeReordered(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedDefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- def testWithOptionalAndDefaultedIndefModeChunked(self):
- self.__initWithOptionalAndDefaulted()
- assert decoder.decode(
- ints2octs((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, null)
- 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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(ints2octs((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(null)),
- namedtype.NamedType('number', univ.Integer(0)),
- namedtype.NamedType('string', univ.OctetString())
- )
- )
- def testBySpec(self):
- self.s.setComponentByPosition(0, univ.Null(null))
- assert decoder.decode(
- ints2octs((5, 0)), asn1Spec=self.s
- ) == (self.s, null)
- def testWithoutSpec(self):
- self.s.setComponentByPosition(0, univ.Null(null))
- assert decoder.decode(ints2octs((5, 0))) == (self.s, null)
- assert decoder.decode(ints2octs((5, 0))) == (univ.Null(null), null)
- def testUndefLength(self):
- self.s.setComponentByPosition(2, univ.OctetString('abcdefgh'))
- assert decoder.decode(ints2octs((36, 128, 4, 3, 97, 98, 99, 4, 3, 100, 101, 102, 4, 2, 103, 104, 0, 0)),
- asn1Spec=self.s) == (self.s, null)
- def testExplicitTag(self):
- s = self.s.subtype(explicitTag=tag.Tag(tag.tagClassContext,
- tag.tagFormatConstructed, 4))
- s.setComponentByPosition(0, univ.Null(null))
- assert decoder.decode(ints2octs((164, 2, 5, 0)), asn1Spec=s) == (s, null)
- def testExplicitTagUndefLength(self):
- s = self.s.subtype(explicitTag=tag.Tag(tag.tagClassContext,
- tag.tagFormatConstructed, 4))
- s.setComponentByPosition(0, univ.Null(null))
- assert decoder.decode(ints2octs((164, 128, 5, 0, 0, 0)), asn1Spec=s) == (s, null)
- class AnyDecoderTestCase(BaseTestCase):
- def setUp(self):
- BaseTestCase.setUp(self)
- self.s = univ.Any()
- def testByUntagged(self):
- assert decoder.decode(
- ints2octs((4, 3, 102, 111, 120)), asn1Spec=self.s
- ) == (univ.Any('\004\003fox'), null)
- def testTaggedEx(self):
- s = univ.Any('\004\003fox').subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(ints2octs((164, 5, 4, 3, 102, 111, 120)), asn1Spec=s) == (s, null)
- def testTaggedIm(self):
- s = univ.Any('\004\003fox').subtype(implicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(ints2octs((132, 5, 4, 3, 102, 111, 120)), asn1Spec=s) == (s, null)
- def testByUntaggedIndefMode(self):
- assert decoder.decode(
- ints2octs((4, 3, 102, 111, 120)), asn1Spec=self.s
- ) == (univ.Any('\004\003fox'), null)
- def testTaggedExIndefMode(self):
- s = univ.Any('\004\003fox').subtype(explicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(ints2octs((164, 128, 4, 3, 102, 111, 120, 0, 0)), asn1Spec=s) == (s, null)
- def testTaggedImIndefMode(self):
- s = univ.Any('\004\003fox').subtype(implicitTag=tag.Tag(tag.tagClassContext, tag.tagFormatSimple, 4))
- assert decoder.decode(ints2octs((164, 128, 4, 3, 102, 111, 120, 0, 0)), asn1Spec=s) == (s, null)
- def testByUntaggedSubst(self):
- assert decoder.decode(
- ints2octs((4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (ints2octs((4, 3, 102, 111, 120)), str2octs(''))
- def testByUntaggedSubstV04(self):
- assert decoder.decode(
- ints2octs((4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (ints2octs((4, 3, 102, 111, 120)), str2octs(''))
- def testTaggedExSubst(self):
- assert decoder.decode(
- ints2octs((164, 5, 4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c, d: streaming.readFromStream(b, c)
- ) == (ints2octs((164, 5, 4, 3, 102, 111, 120)), str2octs(''))
- def testTaggedExSubstV04(self):
- assert decoder.decode(
- ints2octs((164, 5, 4, 3, 102, 111, 120)),
- asn1Spec=self.s,
- substrateFun=lambda a, b, c: (b, b[c:])
- ) == (ints2octs((164, 5, 4, 3, 102, 111, 120)), str2octs(''))
- class EndOfOctetsTestCase(BaseTestCase):
- def testUnexpectedEoo(self):
- try:
- decoder.decode(ints2octs((0, 0)))
- except error.PyAsn1Error:
- pass
- else:
- assert 0, 'end-of-contents octets accepted at top level'
- def testExpectedEoo(self):
- result, remainder = decoder.decode(ints2octs((0, 0)), allowEoo=True)
- assert eoo.endOfOctets.isSameTypeWith(result) and result == eoo.endOfOctets and result is eoo.endOfOctets
- assert remainder == null
- def testDefiniteNoEoo(self):
- try:
- decoder.decode(ints2octs((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(ints2octs((0x23, 0x80, 0x00, 0x00)))
- assert result == () and remainder == null, 'incorrect decoding of indefinite length end-of-octets'
- def testNoLongFormEoo(self):
- try:
- decoder.decode(ints2octs((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(ints2octs((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(ints2octs((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(null)),
- namedtype.NamedType('first-name', univ.OctetString(null)),
- namedtype.NamedType('age', univ.Integer(33))
- )
- )
- self.s.setComponentByPosition(0, univ.Null(null))
- self.s.setComponentByPosition(1, univ.OctetString('quick brown'))
- self.s.setComponentByPosition(2, univ.Integer(1))
- self.substrate = ints2octs([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 = ints2octs((00, 1, 2))
- stream = streaming.asSeekableStream(substrate)
- try:
- asn1Object = next(decode(stream))
- except error.PyAsn1Error:
- exc = sys.exc_info()[1]
- assert isinstance(exc, error.PyAsn1Error), (
- 'Unexpected exception raised %r' % (exc,))
- else:
- assert False, 'Unexpected decoder result %r' % (asn1Object,)
- def testRawDump(self):
- substrate = ints2octs((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 == ints2octs((131, 3, 2, 1, 12)), (
- 'Unexpected rest of substrate after raw dump %r' % rest)
- @unittest.skipIf(sys.version_info < (3,), "Unsupported on Python 2")
- 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(ints2octs((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(ints2octs((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(ints2octs((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 = ints2octs((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 unicode objects in Python 2 & str objects in Python 3.7 cannot be parsed.
- source = ints2octs((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 = ints2octs(
- (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(ints2octs((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):
- # File from ZIP archive is a good example of non-seekable stream in Python 2.7
- # In Python 3.7, it is a seekable stream.
- _, path = tempfile.mkstemp(suffix=".zip")
- try:
- with zipfile.ZipFile(path, "w") as myzip:
- myzip.writestr("data", ints2octs((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", ints2octs((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 = ints2octs((
- 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 = ints2octs((
- 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(ints2octs((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)
|