非ARC下返回Block

非ARC下返回Block

首先，对于没有引用外部变量的Block，无论在ARC还是非ARC下，类型都是NSGlobalBlock，这种类型的block可以理解成一种全局的block，不需要考虑作用域问题。同时，对他进行copy或者retain操作也是无效的，比如这样一个返回Block的函数：

typedef int(^MyBlock)();
MyBlock func()
{
    return ^{ return 123; };
}

在非ARC下执行如下代码：

//非ARC
MyBlock block = func();
NSLog(@"%d", block());
NSLog(@"%@", [block class]);

MyBlock block2 = [block copy];
//Copy操作对__NSGlobalBlock__类型无效
NSLog(@"%d", block == block2);

输出：

123
__NSGlobalBlock__
1

可以看到，copy后的Block和原来是同一个对象的。

而对于引用了外部变量的Block，如果没有对他进行copy，他的作用域只会在声明他的函数栈内（类型是NSStackBlock），如果想在非ARC下直接返回此类Block，Xcode会提示编译错误的，如下图：

（Xcode提示Returning block that lives on the local stack）

而在ARC环境下，上述代码会编译通过，因为ARC会自动加入copy操作。

比如可以在ARC下运行如下代码：

//ARC
MyBlock block = func();
NSLog(@"%d", block());
NSLog(@"%@", [block class]);

输出：

123
__NSMallocBlock__

类型是NSMallocBlock，说明Block已经被copy到了堆中了。

当然其实在非ARC下，也可以使上面有错误的函数编译通过。如下代码：

typedef int(^MyBlock)();
MyBlock func()
{
    int i = 123;
    //非ARC下不要这样！！！
    MyBlock ret = ^{ return i; };
    return ret;
}

我们把原来的返回值赋给一个变量，然后再返回这个变量，就可以编译通过了。不过虽然编译通过了，这个返回的Block作用域仍是在函数栈中的，因此一旦函数运行完毕后再使用这个Block很可能会引发BAD_ACCESS错误。

所以在非ARC下，必须把Block复制到堆中才可以在函数外使用Block，如下正确的代码：

typedef int(^MyBlock)();
MyBlock func()
{
    //非ARC
    int i = 123;
    return [^{ return i; } copy];
}

我们可以直接通过输出变量的指针，就可以验证Block被copy后，他所引用的变量被复制到了堆中的情况，如下代码（非ARC下）：

//非ARC
void func()
{
    int a = 123;
    __block int b = 123;
    NSLog(@"%@", @"=== block copy前");
    NSLog(@"&a = %p, &b = %p", &a, &b);

    void(^block)() = ^{
        NSLog(@"%@", @"=== Block");
        NSLog(@"&a = %p, &b = %p", &a, &b);
        NSLog(@"a = %d, b = %d", a, b = 456);
    };
    block = [block copy];
    block();

    NSLog(@"%@", @"=== block copy后");
    NSLog(@"&a = %p, &b = %p", &a, &b);
    NSLog(@"a = %d, b = %d", a, b);

    [block release];
}

输出：

=== block copy前
&a = 0x7fff5fbff8bc, &b = 0x7fff5fbff8b0
=== Block
&a = 0x100201048, &b = 0x100201068
a = 123, b = 456
=== block copy后
&a = 0x7fff5fbff8bc, &b = 0x100201068
a = 123, b = 456

可以看到，在Block执行中，他所引用的变量a和b都被复制到了堆上。而被标记__block的变量事实上应该说是被移动到了堆上，因此，当Block执行后，函数栈内访问b的地址会变成堆中的地址。而变量a，仍会指向函数栈内原有的变量a的空间。