python解析minicap

上篇知道了minicap发送图片的格式，照着官网的app.js代码，改用一个python版的来解析它，适当扩展，可以做个小工具实时显示手机屏幕.

步骤：

一、手机开启minicap服务

adb shell LD_LIBRARY_PATH=/data/local/tmp /data/local/tmp/minicap -P 1080x1920@1080x1920/

二、本机端口映射

adb forward tcp: localabstract:minicap

三、执行下面的代码

# coding: utf8

'''
    by jc 2018/1/19
'''
import socket, sys
import struct
from collections import OrderedDict
import time

class Banner:
    def __init__(self):
        self.__banner = OrderedDict(
            [('version', 0),
            ('length', 0),
            ('pid', 0),
            ('realWidth', 0),
            ('realHeight', 0),
            ('virtualWidth', 0),
            ('virtualHeight', 0),
            ('orientation', 0),
            ('quirks', 0)
        ])
    def __setitem__(self, key, value):
        self.__banner[key] = value
    def __getitem__(self, key):
        return self.__banner[key]
    def keys(self):
        return self.__banner.keys()
    def __str__(self):
        return str(self.__banner)

class Minicap(object):
    BUFFER_SIZE = 4096
    def __init__(self, host, port, banner):
        self.host = host
        self.port = port
        self.banner = banner
    def connect(self):
        try:
            self.__socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        except socket.error, e:
            print e
            sys.exit(1)
        self.__socket.connect((self.host, self.port))

    def on_image_transfered(self, data):
        file_name = str(time.time()) + '.jpg'
        with open(file_name, 'wb') as f:
            for b in data:
                f.write(b)

    def consume(self):

        readBannerBytes = 0
        bannerLength = 24
        readFrameBytes = 0
        frameBodyLength = 0
        data = []
        while 1:
            try:
                chunk = self.__socket.recv(self.BUFFER_SIZE)
            except socket.error, e:
                print "Error receiving data: %s" % e
                sys.exit(1)
            cursor = 0
            buf_len = len(chunk)
            while cursor < buf_len:
                if readBannerBytes < bannerLength:
                    map(lambda i, val: self.banner.__setitem__(self.banner.keys()[i], val),  [i for i in range(len(self.banner.keys()))], struct.unpack("<2b5ibB", chunk))
                    cursor = buf_len
                    readBannerBytes = bannerLength
                    print self.banner
                elif readFrameBytes < 4:
                    frameBodyLength += (struct.unpack('B', chunk[cursor])[0] << (readFrameBytes * 8)) >> 0
                    cursor += 1
                    readFrameBytes += 1else:
                    print "frame length:%d buf_len:%d cursor:%d" % (frameBodyLength, buf_len, cursor)
                    # pic end
                    if buf_len - cursor >= frameBodyLength:
                        data.extend(chunk[cursor:cursor+frameBodyLength])

                        self.on_image_transfered(data)

                        cursor += frameBodyLength
                        frameBodyLength = readFrameBytes = 0
                        data = []
                    else:
                        data.extend(chunk[cursor:buf_len])
                        frameBodyLength -= buf_len - cursor
                        readFrameBytes += buf_len - cursor
                        cursor = buf_len

if '__main__' == __name__:
    mc = Minicap('localhost', 1313, Banner())
    mc.connect()
    mc.consume()

这样，在当前目录下就会有手机截屏生成.