iOS后台抓取全局屏幕的实现

抓取屏幕的方法很多，但是大多都是抓取UIView的，没有抓取全局屏幕的，只有一个利用硬件缓冲区的方式来抓取全局屏幕的方法，也有代码，但是需要有些功底的人才能够用起来，因为需要修改一些地方才可以用使用。下面我就把我整理的代码写上来，供大家参考使用：

#import <IOMobileFramebuffer.h>
#import <IOKit/IOKitLib.h>
#import <IOSurface/IOSurface.h>
#import <QuartzCore/QuartzCore.h>

UIImage* snapshot() {  // 这个是抓屏的主函数，通过调用此函数，就可以获取全局屏幕的一帧图像
    io_service_t framebufferService = IOServiceGetMatchingService(kIOMasterPortDefault, IOServiceMatching("AppleH1CLCD"));
    if(!framebufferService)
    {
        framebufferService = IOServiceGetMatchingService(kIOMasterPortDefault, IOServiceMatching("AppleM2CLCD"));
    }
    if(!framebufferService)
    {
        framebufferService = IOServiceGetMatchingService(kIOMasterPortDefault, IOServiceMatching("AppleCLCD"));
    }
    
    kern_return_t result;
    IOMobileFramebufferConnection connect;
    CoreSurfaceBufferRef screenSurface = NULL;
    
    result = IOMobileFramebufferOpen(framebufferService, mach_task_self(), 0, &connect);
    result = IOMobileFramebufferGetLayerDefaultSurface(connect, 0, &screenSurface);
    
    uint32_t aseed;
    IOSurfaceLock((IOSurfaceRef)screenSurface, kIOSurfaceLockReadOnly, &aseed);
    size_t width = IOSurfaceGetWidth((IOSurfaceRef)screenSurface);
    size_t height = IOSurfaceGetHeight((IOSurfaceRef)screenSurface);
    int bytes = 4;
    size_t pitch = width * bytes, size = width * height * bytes;
    
    char pixelFormat[4] = {'A','R','G','B'};
    
    CFMutableDictionaryRef dict = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
    CFDictionarySetValue(dict, kIOSurfaceIsGlobal,        kCFBooleanTrue);
    CFDictionarySetValue(dict, kIOSurfaceBytesPerRow,     CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &pitch));
    CFDictionarySetValue(dict, kIOSurfaceBytesPerElement, CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &bytes));
    CFDictionarySetValue(dict, kIOSurfaceWidth,           CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &width));
    CFDictionarySetValue(dict, kIOSurfaceHeight,          CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &height));
    CFDictionarySetValue(dict, kIOSurfacePixelFormat,     CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, pixelFormat));
    CFDictionarySetValue(dict, kIOSurfaceAllocSize,       CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &size));
    
    IOSurfaceAcceleratorRef outAcc;
    IOSurfaceRef destSurf = IOSurfaceCreate(dict);
    IOSurfaceAcceleratorCreate(NULL, 0, &outAcc);
    IOSurfaceAcceleratorTransferSurface(outAcc, screenSurface, destSurf, dict, NULL);
    IOSurfaceUnlock((IOSurfaceRef)screenSurface, kIOSurfaceLockReadOnly, &aseed);
    CFRelease(outAcc);
    
    CGDataProviderRef provider =  CGDataProviderCreateWithData(NULL,  IOSurfaceGetBaseAddress(destSurf), size, NULL);
    CGImageRef cgImage = CGImageCreate(width, height, 8, 8 * bytes, IOSurfaceGetBytesPerRow(destSurf), CGColorSpaceCreateDeviceRGB(),
                                       kCGImageAlphaNoneSkipFirst | kCGBitmapByteOrder32Little, provider, NULL, YES, kCGRenderingIntentDefault);
    UIImage *image = [UIImage imageWithCGImage:cgImage];
    
    CGImageRelease(cgImage);
    CFDictionaryRemoveAllValues(dict);
    CGDataProviderRelease(provider);
    return image;
}

UIImage* rotation(UIImage *image, UIImageOrientation orientation) {  // 因为抓取出来的图像是横着的，对于移动设备来说，需要旋转一下，所以添加了这个函数（来自互联网）
    long double rotate = 0.0;
    CGRect rect;
    
    float translateX = 0, translateY = 0, scaleX = 1.0, scaleY = 1.0;
    
    switch (orientation)
    {
        case UIImageOrientationLeft:
            rotate     = M_PI_2;
            rect       = CGRectMake(0, 0, image.size.height, image.size.width);
            translateX = 0;
            translateY = -rect.size.width;
            scaleY     = rect.size.width / rect.size.height;
            scaleX     = rect.size.height / rect.size.width;
            break;
            
        case UIImageOrientationRight:
            rotate     = 3 * M_PI_2;
            rect       = CGRectMake(0, 0, image.size.height, image.size.width);
            translateX = -rect.size.height;
            translateY = 0;
            scaleY     = rect.size.width / rect.size.height;
            scaleX     = rect.size.height / rect.size.width;
            break;
            
        case UIImageOrientationDown:
            rotate     = M_PI;
            rect       = CGRectMake(0, 0, image.size.width, image.size.height);
            translateX = -rect.size.width;
            translateY = -rect.size.height;
            break;
            
        default:
            rotate     = 0.0;
            rect       = CGRectMake(0, 0, image.size.width, image.size.height);
            translateX = 0;
            translateY = 0;
            break;
    }
    
    UIGraphicsBeginImageContext(rect.size);
    CGContextRef context = UIGraphicsGetCurrentContext();
    
    // ctm transform
    CGContextTranslateCTM(context, 0.0, rect.size.height);
    CGContextScaleCTM(context, 1.0, -1.0);
    CGContextRotateCTM(context, rotate);
    CGContextTranslateCTM(context, translateX, translateY);
    
    CGContextScaleCTM(context, scaleX, scaleY);
    
    // draw image
    CGContextDrawImage(context, CGRectMake(0, 0, rect.size.width, rect.size.height), image.CGImage);
    UIImage *newImg = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return newImg;
}

主要代码就写到这里，但是这和多数人从网上找到的差不多，编译时你会发现，编译出错了，因为IOSurfaceAcceleratorRef这个定义和IOSurfaceAcceleratorCreate，IOSurfaceAcceleratorTransferSurface这个两个函数你找不到，为什么呢？难道代码有误？

其实不是，是因为最初的编辑者少上传了一个头文件导致的，下面我就把这个头文件给贴上来：

//
// IOSurfaceAccelerator.h
//
#ifndef _IOSURFACE_ACCELERATOR_H
#define _IOSURFACE_ACCELERATOR_H 1

#include <IOSurface/IOSurfaceAPI.h>
#include <IOKit/IOReturn.h>

#if __cplusplus
extern "C" {
#endif

typedef IOReturn IOSurfaceAcceleratorReturn;

enum {
	kIOSurfaceAcceleratorSuccess = 0,
};

typedef struct __IOSurfaceAccelerator *IOSurfaceAcceleratorRef;

IOSurfaceAcceleratorReturn IOSurfaceAcceleratorCreate(CFAllocatorRef allocator, uint32_t type, IOSurfaceAcceleratorRef *outAccelerator);
IOSurfaceAcceleratorReturn IOSurfaceAcceleratorTransferSurface(IOSurfaceAcceleratorRef accelerator, IOSurfaceRef sourceSurface, IOSurfaceRef destSurface, CFDictionaryRef dict, void *unknown);

#if __cplusplus
}
#endif

#endif

就是它了，把它拷贝下去，写到IOSurfaceAccelerator.h中，并且放到headers/IOSurface目录下，修改一下IOSurface.h文件，在文件最后添加上，

#include <IOSurface/IOSurfaceAccelerator.h>

就算完工了。剩下的就是编译了连接了。

连接的时候，你需要添加IOKit.framework,IOMobileFramebuffer.framework和IOSurface.framework这三个包，才能够顺利的编译过去。

画外音：不过经过我测试，这个方法的效率不是很高，250ms一帧的效率，还是有一点点不能接受的（可能是我压缩jpg导致的），但是除了这个方法外，也没其他办法了，我也只能忍了。

开始build吧，希望你也能成功。

附上头文件下载地址：头文件下载地址