Difference between revisions of "Manual:API Python3"

From MikroTik Wiki
Jump to: navigation, search
m (code)
(add default user/paass; add new stype login; fix large sentence length decoding)
Line 16: Line 16:
 
         self.sk = sk
 
         self.sk = sk
 
         self.currenttag = 0
 
         self.currenttag = 0
       
+
 
 
     def login(self, username, pwd):
 
     def login(self, username, pwd):
         for repl, attrs in self.talk(["/login"]):
+
         for repl, attrs in self.talk(["/login", "=name=" + username,
             chal = binascii.unhexlify((attrs['=ret']).encode('UTF-8'))
+
                                      "=password=" + pwd]):
        md = hashlib.md5()
+
          if repl == '!trap':
        md.update(b'\x00')
+
            return False
        md.update(pwd.encode('UTF-8'))
+
          elif '=ret' in attrs.keys():
        md.update(chal)
+
        #for repl, attrs in self.talk(["/login"]):
        self.talk(["/login", "=name=" + username,
+
             chal = binascii.unhexlify((attrs['=ret']).encode(sys.stdout.encoding))
                   "=response=00" + binascii.hexlify(md.digest()).decode('UTF-8') ])
+
            md = hashlib.md5()
 +
            md.update(b'\x00')
 +
            md.update(pwd.encode(sys.stdout.encoding))
 +
            md.update(chal)
 +
            for repl2, attrs2 in self.talk(["/login", "=name=" + username,
 +
                   "=response=00" + binascii.hexlify(md.digest()).decode(sys.stdout.encoding) ]):
 +
              if repl2 == '!trap':
 +
                return False
 +
        return True
  
 
     def talk(self, words):
 
     def talk(self, words):
Line 58: Line 66:
 
             if w == '': return r
 
             if w == '': return r
 
             r.append(w)
 
             r.append(w)
           
+
 
 
     def writeWord(self, w):
 
     def writeWord(self, w):
 
         print(("<<< " + w))
 
         print(("<<< " + w))
Line 82: Line 90:
 
             self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
         elif l < 0x10000000:      
+
         elif l < 0x10000000:
             l |= 0xE0000000        
+
             l |= 0xE0000000
 
             self.writeByte(((l >> 24) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 24) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 16) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 16) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
         else:                      
+
         else:
 
             self.writeByte((0xF0).to_bytes(1, sys.byteorder))
 
             self.writeByte((0xF0).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 24) & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte(((l >> 24) & 0xFF).to_bytes(1, sys.byteorder))
Line 95: Line 103:
 
             self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
 
             self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
  
     def readLen(self):            
+
     def readLen(self):
         c = ord(self.readStr(1))  
+
         c = ord(self.readStr(1))
         if (c & 0x80) == 0x00:    
+
        # print (">rl> %i" % c)
             pass                  
+
         if (c & 0x80) == 0x00:
         elif (c & 0xC0) == 0x80:  
+
             pass
             c &= ~0xC0            
+
         elif (c & 0xC0) == 0x80:
             c <<= 8                
+
             c &= ~0xC0
             c += ord(self.readStr(1))  
+
             c <<= 8
         elif (c & 0xE0) == 0xC0:  
+
             c += ord(self.readStr(1))
             c &= ~0xE0            
+
         elif (c & 0xE0) == 0xC0:
             c <<= 8                
+
             c &= ~0xE0
             c += ord(self.readStr(1))  
+
             c <<= 8
             c <<= 8                
+
             c += ord(self.readStr(1))
             c += ord(self.readStr(1))  
+
             c <<= 8
         elif (c & 0xF0) == 0xE0:  
+
             c += ord(self.readStr(1))
             c &= ~0xF0            
+
         elif (c & 0xF0) == 0xE0:
             c <<= 8                
+
             c &= ~0xF0
             c += ord(self.readStr(1))  
+
             c <<= 8
             c <<= 8                
+
             c += ord(self.readStr(1))
             c += ord(self.readStr(1))  
+
             c <<= 8
             c <<= 8                
+
             c += ord(self.readStr(1))
             c += ord(self.readStr(1))  
+
             c <<= 8
         elif (c & 0xF8) == 0xF0:  
+
             c += ord(self.readStr(1))
             c = ord(self.readStr(1))    
+
         elif (c & 0xF8) == 0xF0:
             c <<= 8                
+
             c = ord(self.readStr(1))
             c += ord(self.readStr(1))  
+
             c <<= 8
             c <<= 8                
+
             c += ord(self.readStr(1))
             c += ord(self.readStr(1))  
+
             c <<= 8
             c <<= 8                
+
             c += ord(self.readStr(1))
             c += ord(self.readStr(1))  
+
             c <<= 8
         return c                  
+
             c += ord(self.readStr(1))
 +
         return c
  
     def writeStr(self, str):      
+
     def writeStr(self, str):
 
         n = 0;
 
         n = 0;
 
         while n < len(str):
 
         while n < len(str):
 
             r = self.sk.send(bytes(str[n:], 'UTF-8'))
 
             r = self.sk.send(bytes(str[n:], 'UTF-8'))
 
             if r == 0: raise RuntimeError("connection closed by remote end")
 
             if r == 0: raise RuntimeError("connection closed by remote end")
             n += r                
+
             n += r
  
     def writeByte(self, str):      
+
     def writeByte(self, str):
 
         n = 0;
 
         n = 0;
 
         while n < len(str):
 
         while n < len(str):
 
             r = self.sk.send(str[n:])
 
             r = self.sk.send(str[n:])
 
             if r == 0: raise RuntimeError("connection closed by remote end")
 
             if r == 0: raise RuntimeError("connection closed by remote end")
             n += r                
+
             n += r
  
     def readStr(self, length):    
+
     def readStr(self, length):
 
         ret = ''
 
         ret = ''
 +
        # print ("length: %i" % length)
 
         while len(ret) < length:
 
         while len(ret) < length:
 
             s = self.sk.recv(length - len(ret))
 
             s = self.sk.recv(length - len(ret))
             if s == '': raise RuntimeError("connection closed by remote end")
+
             if s == b'': raise RuntimeError("connection closed by remote end")
             ret += s.decode('UTF-8', 'replace')
+
             # print (b">>>" + s)
 +
            # atgriezt kaa byte ja nav ascii chars
 +
            if s >= (128).to_bytes(1, "big") :
 +
              return s
 +
            # print((">>> " + s.decode(sys.stdout.encoding, 'ignore')))
 +
            ret += s.decode(sys.stdout.encoding, "replace")
 
         return ret
 
         return ret
  
Line 170: Line 185:
  
 
     apiros = ApiRos(s);
 
     apiros = ApiRos(s);
     apiros.login(sys.argv[2], sys.argv[3]);
+
 
 +
     # use default username and pasword if not specified
 +
    if len(sys.argv) == 4:
 +
      if not apiros.login(sys.argv[2], sys.argv[3]):
 +
        return
 +
    elif len(sys.argv) == 3:
 +
      if not apiros.login(sys.argv[2], ""):
 +
        return
 +
    else :
 +
      if not apiros.login("admin", ""):
 +
        return
  
 
     inputsentence = []
 
     inputsentence = []
Line 195: Line 220:
 
if __name__ == '__main__':
 
if __name__ == '__main__':
 
     main()
 
     main()
 
 
  
 
</pre>
 
</pre>

Revision as of 15:15, 13 July 2018

Summary

Since python language have introduced changes to syntax when going from 2.x to 3.x some adjustments had to be made for old code from API.

Code for Python3

code

#!/usr/bin/python3
# -*- coding: latin-1 -*-
import sys, posix, time, binascii, socket, select
import hashlib

class ApiRos:
    "Routeros api"
    def __init__(self, sk):
        self.sk = sk
        self.currenttag = 0

    def login(self, username, pwd):
        for repl, attrs in self.talk(["/login", "=name=" + username,
                                      "=password=" + pwd]):
          if repl == '!trap':
            return False
          elif '=ret' in attrs.keys():
        #for repl, attrs in self.talk(["/login"]):
            chal = binascii.unhexlify((attrs['=ret']).encode(sys.stdout.encoding))
            md = hashlib.md5()
            md.update(b'\x00')
            md.update(pwd.encode(sys.stdout.encoding))
            md.update(chal)
            for repl2, attrs2 in self.talk(["/login", "=name=" + username,
                   "=response=00" + binascii.hexlify(md.digest()).decode(sys.stdout.encoding) ]):
              if repl2 == '!trap':
                return False
        return True

    def talk(self, words):
        if self.writeSentence(words) == 0: return
        r = []
        while 1:
            i = self.readSentence();
            if len(i) == 0: continue
            reply = i[0]
            attrs = {}
            for w in i[1:]:
                j = w.find('=', 1)
                if (j == -1):
                    attrs[w] = ''
                else:
                    attrs[w[:j]] = w[j+1:]
            r.append((reply, attrs))
            if reply == '!done': return r

    def writeSentence(self, words):
        ret = 0
        for w in words:
            self.writeWord(w)
            ret += 1
        self.writeWord('')
        return ret

    def readSentence(self):
        r = []
        while 1:
            w = self.readWord()
            if w == '': return r
            r.append(w)

    def writeWord(self, w):
        print(("<<< " + w))
        self.writeLen(len(w))
        self.writeStr(w)

    def readWord(self):
        ret = self.readStr(self.readLen())
        print((">>> " + ret))
        return ret

    def writeLen(self, l):
        if l < 0x80:
            self.writeByte((l).to_bytes(1, sys.byteorder))
        elif l < 0x4000:
            l |= 0x8000
            tmp = (l >> 8) & 0xFF
            self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
        elif l < 0x200000:
            l |= 0xC00000
            self.writeByte(((l >> 16) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
        elif l < 0x10000000:
            l |= 0xE0000000
            self.writeByte(((l >> 24) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte(((l >> 16) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))
        else:
            self.writeByte((0xF0).to_bytes(1, sys.byteorder))
            self.writeByte(((l >> 24) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte(((l >> 16) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte(((l >> 8) & 0xFF).to_bytes(1, sys.byteorder))
            self.writeByte((l & 0xFF).to_bytes(1, sys.byteorder))

    def readLen(self):
        c = ord(self.readStr(1))
        # print (">rl> %i" % c)
        if (c & 0x80) == 0x00:
            pass
        elif (c & 0xC0) == 0x80:
            c &= ~0xC0
            c <<= 8
            c += ord(self.readStr(1))
        elif (c & 0xE0) == 0xC0:
            c &= ~0xE0
            c <<= 8
            c += ord(self.readStr(1))
            c <<= 8
            c += ord(self.readStr(1))
        elif (c & 0xF0) == 0xE0:
            c &= ~0xF0
            c <<= 8
            c += ord(self.readStr(1))
            c <<= 8
            c += ord(self.readStr(1))
            c <<= 8
            c += ord(self.readStr(1))
        elif (c & 0xF8) == 0xF0:
            c = ord(self.readStr(1))
            c <<= 8
            c += ord(self.readStr(1))
            c <<= 8
            c += ord(self.readStr(1))
            c <<= 8
            c += ord(self.readStr(1))
        return c

    def writeStr(self, str):
        n = 0;
        while n < len(str):
            r = self.sk.send(bytes(str[n:], 'UTF-8'))
            if r == 0: raise RuntimeError("connection closed by remote end")
            n += r

    def writeByte(self, str):
        n = 0;
        while n < len(str):
            r = self.sk.send(str[n:])
            if r == 0: raise RuntimeError("connection closed by remote end")
            n += r

    def readStr(self, length):
        ret = ''
        # print ("length: %i" % length)
        while len(ret) < length:
            s = self.sk.recv(length - len(ret))
            if s == b'': raise RuntimeError("connection closed by remote end")
            # print (b">>>" + s)
            # atgriezt kaa byte ja nav ascii chars
            if s >= (128).to_bytes(1, "big") :
               return s
            # print((">>> " + s.decode(sys.stdout.encoding, 'ignore')))
            ret += s.decode(sys.stdout.encoding, "replace")
        return ret

def main():
    s = None
    for res in socket.getaddrinfo(sys.argv[1], "8728", socket.AF_UNSPEC, socket.SOCK_STREAM):
        af, socktype, proto, canonname, sa = res
        try:
             s = socket.socket(af, socktype, proto)
        except (socket.error, msg):
            s = None
            continue
        try:
            s.connect(sa)
        except (socket.error, msg):
            s.close()
            s = None
            continue
        break
    if s is None:
        print ('could not open socket')
        sys.exit(1)

    apiros = ApiRos(s);

    # use default username and pasword if not specified
    if len(sys.argv) == 4:
      if not apiros.login(sys.argv[2], sys.argv[3]):
        return
    elif len(sys.argv) == 3:
      if not apiros.login(sys.argv[2], ""):
        return
    else :
      if not apiros.login("admin", ""):
        return

    inputsentence = []

    while 1:
        r = select.select([s, sys.stdin], [], [], None)
        if s in r[0]:
            # something to read in socket, read sentence
            x = apiros.readSentence()

        if sys.stdin in r[0]:
            # read line from input and strip off newline
            l = sys.stdin.readline()
            l = l[:-1]

            # if empty line, send sentence and start with new
            # otherwise append to input sentence
            if l == '':
                apiros.writeSentence(inputsentence)
                inputsentence = []
            else:
                inputsentence.append(l)

if __name__ == '__main__':
    main()

file

api client in python3

See also

API