check for end-of-octets in version

1 files changed, 3 insertions, 1 deletions

diff --git a/pysnmp/proto/api/verdec.py b/pysnmp/proto/api/verdec.py
index 34d5c8f..8895f27 100644
--- a/pysnmp/proto/api/verdec.py
+++ b/pysnmp/proto/api/verdec.py
@@ -1,5 +1,5 @@
 from pyasn1.type import univ
-from pyasn1.codec.ber import decoder
+from pyasn1.codec.ber import decoder, eoo
 from pyasn1.error import PyAsn1Error
 from pysnmp.proto.error import ProtocolError
 
@@ -11,6 +11,8 @@ def decodeMessageVersion(wholeMsg):
         ver, wholeMsg = decoder.decode(
             wholeMsg, asn1Spec=univ.Integer(), recursiveFlag=0
         )
+        if eoo.endOfOctets.isSameTypeWith(ver):
+            raise ProtocolError('EOO at SNMP version component')
         return ver
     except PyAsn1Error:
         raise ProtocolError('Invalid BER at SNMP version component')