iOS基础:block 内如何修改 block 外部变量

block 原理已有很多优秀的博客介绍过了,这里是对 block 相关知识的复习巩固

在 block 内部修改其外部变量,大家都知道要使用 __block 关键字,其原理简单的说就是:使用了 __blcok 之后,在 block 被 copy 到堆上的同时也会将捕获的外部变量 copy 到堆上,之后便可以在 block 内部对外部变量进行修改。具体情况下面分析。

无 __block 关键字的参数捕获

没有使用 __block 的关键字我们可以在 block 中使用但是不能修改,具体原因可以查看下以下源码。使用 clang 编译之后,有一个 __main_block_impl_0 的结构体,它就是 block 的 c++ 实现,它里面有一行 NSInteger a; 如果 block 中没有使用局部变量时,是没有这段代码的,同时其构造函数参数列表里面也多了个参数 a。
可以明显的看出这里的参数捕获只是完成了一次值得传递,当你在 block 中修改这个变量的时候,编译器就会报错。其实这里的变量 a, 与 block 外面的局部变量 a 已经不是同一个变量了,在 block 内部对其进行修改对外部没有任何影响。

int main(int argc, char * argv[]) {
    @autoreleasepool {
        NSInteger a = 10;
        void (^blk)(void) = ^{printf("block\n, a = %ld",a);};
        blk();
    }
    return 0;
}
struct __main_block_impl_0 {
  struct __block_impl impl;
  struct __main_block_desc_0* Desc;
  NSInteger a;
  __main_block_impl_0(void *fp, struct __main_block_desc_0 *desc, NSInteger _a, int flags=0) : a(_a) {
    impl.isa = &_NSConcreteStackBlock;
    impl.Flags = flags;
    impl.FuncPtr = fp;
    Desc = desc;
  }
};
static void __main_block_func_0(struct __main_block_impl_0 *__cself) {
  NSInteger a = __cself->a; // bound by copy
printf("block\n, a = %ld",a);}

static struct __main_block_desc_0 {
  size_t reserved;
  size_t Block_size;
} __main_block_desc_0_DATA = { 0, sizeof(struct __main_block_impl_0)};
int main(int argc, char * argv[]) {
    /* @autoreleasepool */ { __AtAutoreleasePool __autoreleasepool; 

        NSInteger a = 10;
        void (*blk)(void) = ((void (*)())&__main_block_impl_0((void *)__main_block_func_0, &__main_block_desc_0_DATA, a));
        ((void (*)(__block_impl *))((__block_impl *)blk)->FuncPtr)((__block_impl *)blk);

    }
    return 0;
}

具有 __block 关键字的参数捕获

查看编译之后的代码,添加 __block 关键字之后,在 __main_block_impl_0 函数中可以看到出现一段代码 __Block_byref_a_0 *a; 变量 a 在被捕获之后被包装成为一个 __Block_byref_a_0 类型对象,__Block_byref_a_0 是一个结构体,具有 isa 指针,因此也是一个对象。
当block被copy到堆中时,拷贝辅助函数 __main_block_copy_0 会将__Block_byref_a_0 拷贝至堆中,所以即使局部变量所在堆被销毁,block依然能对堆中的局部变量进行操作。其中 __Block_byref_a_0 成员指针 __forwarding 用来指向它在堆中的拷贝,此时在栈上的变量的地址也指向堆,这样就保证了修改的对象始终是堆中的。


iOS基础:block 内如何修改 block 外部变量_第1张图片
36DB7C8E-E944-451F-A8F6-1F7C1921C49A.png

编译之后

struct __Block_byref_a_0 {
  void *__isa;
__Block_byref_a_0 *__forwarding;
 int __flags;
 int __size;
 int a;
};

struct __main_block_impl_0 {
  struct __block_impl impl;
  struct __main_block_desc_0* Desc;
  __Block_byref_a_0 *a; // by ref
  __main_block_impl_0(void *fp, struct __main_block_desc_0 *desc, __Block_byref_a_0 *_a, int flags=0) : a(_a->__forwarding) {
    impl.isa = &_NSConcreteStackBlock;
    impl.Flags = flags;
    impl.FuncPtr = fp;
    Desc = desc;
  }
};
static void __main_block_func_0(struct __main_block_impl_0 *__cself) {
  __Block_byref_a_0 *a = __cself->a; // bound by ref
printf("block\n"); (a->__forwarding->a) = 10;}
static void __main_block_copy_0(struct __main_block_impl_0*dst, struct __main_block_impl_0*src) {_Block_object_assign((void*)&dst->a, (void*)src->a, 8/*BLOCK_FIELD_IS_BYREF*/);}

static void __main_block_dispose_0(struct __main_block_impl_0*src) {_Block_object_dispose((void*)src->a, 8/*BLOCK_FIELD_IS_BYREF*/);}

static struct __main_block_desc_0 {
  size_t reserved;
  size_t Block_size;
  void (*copy)(struct __main_block_impl_0*, struct __main_block_impl_0*);
  void (*dispose)(struct __main_block_impl_0*);
} __main_block_desc_0_DATA = { 0, sizeof(struct __main_block_impl_0), __main_block_copy_0, __main_block_dispose_0};
int main(int argc, char * argv[]) {
    /* @autoreleasepool */ { __AtAutoreleasePool __autoreleasepool; 
        __attribute__((__blocks__(byref))) __Block_byref_a_0 a = {(void*)0,(__Block_byref_a_0 *)&a, 0, sizeof(__Block_byref_a_0)};
;
        void (*blk)(void) = ((void (*)())&__main_block_impl_0((void *)__main_block_func_0, &__main_block_desc_0_DATA, (__Block_byref_a_0 *)&a, 570425344));
        ((void (*)(__block_impl *))((__block_impl *)blk)->FuncPtr)((__block_impl *)blk);

    }
    return 0;
}

在 block 内修改全局变量

对于全局变量来说,其存储实在静态数据存储区,在程序结束之前都不会被释放,在 block 中可以直接对其进行修改。

NSInteger a;


struct __main_block_impl_0 {
  struct __block_impl impl;
  struct __main_block_desc_0* Desc;
  __main_block_impl_0(void *fp, struct __main_block_desc_0 *desc, int flags=0) {
    impl.isa = &_NSConcreteStackBlock;
    impl.Flags = flags;
    impl.FuncPtr = fp;
    Desc = desc;
  }
};
static void __main_block_func_0(struct __main_block_impl_0 *__cself) {
 a = 10; printf("block\n, a = %ld",a);}

static struct __main_block_desc_0 {
  size_t reserved;
  size_t Block_size;
} __main_block_desc_0_DATA = { 0, sizeof(struct __main_block_impl_0)};
int main(int argc, char * argv[]) {
    /* @autoreleasepool */ { __AtAutoreleasePool __autoreleasepool; 


        void (*blk)(void) = ((void (*)())&__main_block_impl_0((void *)__main_block_func_0, &__main_block_desc_0_DATA));
        ((void (*)(__block_impl *))((__block_impl *)blk)->FuncPtr)((__block_impl *)blk);

    }
    return 0;
}
static struct IMAGE_INFO { unsigned version; unsigned flag; } _OBJC_IMAGE_INFO = { 0, 2 };

捕获变量之后 block 的变化(无 __block 关键字)

局部变量:
__block_impl_0 实现中会增加一个成员变量,用来存储 block 外部变量的值,仅仅是一次值传递,在 block 内部对其操作会报错。
全局变量:
全局变量在 __block_impl_0 实现中并没有出现,因为全局变量存储在静态数据区,程序结束前并不会被销毁,block 可直接访问,因此在 __block_impl_0 结构体中没有体现。
局部静态变量:
__block_impl_0 结构体中会增加一个指针变量,在 _block_func_0 中通过局部变量的地址可以对其进行访问修改,但其作用域就是当前所在函数的作用域

更多关于 block 的知识参考文章:Block技巧与底层解析、谈 Objective-C block 的实现

你可能感兴趣的:(iOS基础:block 内如何修改 block 外部变量)