Swift中闭包作为Blcok或C函数指针参数

最近在用Swift来播放系统震动音和使用Aspects来hook方法时遇到了点问题,所以学习并记录下。

一、Swift闭包作为block或者c函数指针参数时需使用@convention

@convention(swift) : 表明这个是一个swift的闭包。
@convention(block) :表明这个是一个兼容oc的block的闭包。
@convention(c) : 表明这个是兼容c的函数指针的闭包。

1.1 作为C语言函数指针

下面AudioServicesAddSystemSoundCompletion是一个系统播放震动的函数,其中第四个参数是一个c函数指针,此时如果将一个提前声明的闭包变量传入,则闭包声明时需要使用@convention(c)修饰,否则会报如下错误:
A C function pointer can only be formed from a reference to a 'func' or a literal closure. 翻译过来就是一个C函数指针只能传一个func或者闭包字面量,即只能传一个Swift函数或者直接写一个闭包在参数位置。

func testConvention2() {
    // 如果把closure放在外面定义好再作为参数传进去,则需要使用'@convention(c)'修饰
    // 不然会报错
    // A C function pointer can only be formed from a reference to a 'func' or a literal closure
    
    // 解决方法一:使用'@convention(c)'修饰闭包
    let completion: @convention(c) (SystemSoundID, UnsafeMutableRawPointer?) -> Void = { soundId, pointer in
        print("@convention的闭包 ----")
    }
    AudioServicesAddSystemSoundCompletion(kSystemSoundID_Vibrate, nil, nil, completion, nil)
    
    // 解决方法二:定义一个swift的函数作为参数
    func swiftFunc(id: SystemSoundID, pointer: UnsafeMutableRawPointer?) {
        print("swiftFunc---")
    }
    AudioServicesAddSystemSoundCompletion(kSystemSoundID_Vibrate, nil, nil, swiftFunc, nil)
    
    // 解决方法三:参数位置写闭包字面量
    AudioServicesAddSystemSoundCompletion(kSystemSoundID_Vibrate, nil, nil, { soundID, pointer in
        print("字面量闭包--")
    }, nil)
}
1.2 作为OC的block

相对于上面C语言函数指针,对于OC中的block,大部分情况下Swift闭包可以直接传入使用。但在Aspects中hook方法需要使用@convention(block)修饰闭包,因为里面涉及到了block的方法签名。下面是一个Swift中使用Aspects来hook方法的例子:

/// 做交换: 用一个全局的常量来实现的dispatchOnce效果
let doSwizzles: () = {
    let oriSel1 = #selector(UIViewController.viewWillAppear(_:))
  
    // '@convention(block)'修饰闭包作为参数传入才不会触发Aspects的断言奔溃
    // 因为closure没有block的方法签名
    let wrappedBlock: @convention(block) (AspectInfo, Bool) -> Void = { aspectInfo, _ in
        print("wrappedBlock---")
    }
    _ = try? UIViewController.aspect_hook(oriSel1, with: AspectOptions.positionBefore, usingBlock: wrappedBlock)
}
二、闭包作为C语言指针传入时捕获变量怎么处理?

在上面例子中,闭包作为block参数时,是一样可以捕获局部变量的;但是,当闭包作为C语言函数指针时,闭包中如果捕获局部变量(一般是使用了self或者局部变量),会编译报错,这种情况怎么处理呢?

class ABCClass {
var times = 0
func testSound() {
    // C函数指针的闭包中引用了局部变量或者self
    /* 1.如下代码,
     * 在给下面函数的第4个参数之间传一个对应类型的closure, 如果closure中使用了self或者局部变量会报错:
     * A C function pointer cannot be formed from a closure that captures context
     * 即不能从捕获了变量的closure形成C函数指针
     * Swift中的closure跟OC中的block一样,会捕获局部变量(self也属于局部变量),不捕获全局变量和静态变量。 */
    // 解决方法:
    // 根据这个系统函数的特点,最后一个参数是一个指针,且系统在调用闭包时会将这个指针作为第二个参数,
    // 所以我们可以将要捕获的变量转为指针传入这个系统函数的最后一个参数。
    // 方法一:
    // 注意:(如果在闭包中捕获的是一个局部变量,那么这个局部变量需要自己保证不会太早被释放。)
    // var times2 = 10 // 如果将这个局部变量转为指针传入,则在闭包调用前已经被销毁,这不是预期的结果
    // 所以下面的times需要是成员变量
    let times2Pointer = withUnsafeMutablePointer(to: &times, { UnsafeMutableRawPointer($0) })
    AudioServicesAddSystemSoundCompletion(kSystemSoundID_Vibrate, nil, nil,  { _, pointer in
        // 这里是控制音效播放3次后停止
        var times22 = pointer?.assumingMemoryBound(to: Int.self)
        times22?.pointee += 1
        if times22?.pointee == 3 {
            AudioServicesRemoveSystemSoundCompletion(kSystemSoundID_Vibrate)
            return
        }
        AudioServicesPlaySystemSound(kSystemSoundID_Vibrate)
    }, times2Pointer)
    AudioServicesPlaySystemSound(kSystemSoundID_Vibrate)
  
    // 方法二:直接将self转为指针作为最后一个参数传入
    // 2.1 需要外部强引用self,保证在闭包回调时不会被释放
    let selfPointer2 = unsafeBitCast(self, to: UnsafeMutableRawPointer.self)
    AudioServicesAddSystemSoundCompletion(kSystemSoundID_Vibrate, nil, nil,  { _, pointer in
        guard let pointer = pointer else { return }
        // 下面是控制音效播放3次后停止
        let self_p = unsafeBitCast(pointer, to: AbbVC.self)
        self_p.times += 1
        if self_p.times == 5 {
            AudioServicesRemoveSystemSoundCompletion(kSystemSoundID_Vibrate)
            return
        }
        AudioServicesPlaySystemSound(kSystemSoundID_Vibrate)
    }, selfPointer2)
    AudioServicesPlaySystemSound(kSystemSoundID_Vibrate)

    // 方法三:在闭包中使用全局变量,不使用会被捕获的内容。这个不推荐,因为全局变量会一直存在内存里。
}
}
三、使用Unmanaged来管理self的生命和指针转换

在上面的例子中,播放震动完毕的C函数回调中,如果闭包作为参数要使用self,我们还可以通过Unmanaged来完成,例子如下:

    // 方法二:
    // 2.1 如果外部没有持有self,则这里需要passRetained来引用,以保证在closure调用时没有被销毁
    let selfPointer = Unmanaged<AbbVC>.passRetained(self).toOpaque()
    
    /* 2.2 如果外部已经持有self,则这里可以使用passUnretained,
     let selfPointer = Unmanaged<AbbVC>.passUnretained(self).toOpaque()
     且在使用完后不需要调用Unmanaged<AbbVC>.fromOpaque(pointer).release()
     */
    AudioServicesAddSystemSoundCompletion(kSystemSoundID_Vibrate, nil, nil,  { _, pointer in
        guard let pointer = pointer else { return }
        // 下面是控制音效播放3次后停止
        let self_p = Unmanaged<AbbVC>.fromOpaque(pointer).takeUnretainedValue()
        print(self_p.times)
        self_p.times += 1
        print(self_p.times)
        if self_p.times == 5 {
            Unmanaged<AbbVC>.fromOpaque(pointer).release()
            AudioServicesRemoveSystemSoundCompletion(kSystemSoundID_Vibrate)
            return
        }
        AudioServicesPlaySystemSound(kSystemSoundID_Vibrate)
    }, selfPointer)
    AudioServicesPlaySystemSound(kSystemSoundID_Vibrate)

代码解释:

1.Unmanaged<AbbVC>.passRetained(self) 是创建一个Unmanaged类型的retain引用,会强引用self。
2.Unmanaged<AbbVC>.passUnretained(self)创建一个Unmanaged类型的引用,不会强引用self。
3.Unmanaged<AbbVC>.fromOpaque(pointer).release()是对之前创建的Unmanaged类型的强引用做一次release,不这样做的话,self不会释放。

推荐阅读:
swift的@convention
Swift中指针

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,014评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,796评论 3 386
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,484评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,830评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,946评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,114评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,182评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,927评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,369评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,678评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,832评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,533评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,166评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,885评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,128评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,659评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,738评论 2 351

推荐阅读更多精彩内容