求IOS录屏并录声音的方案

小弟最后要做一个功能，就是要录IOS设备的屏显示和声音输出，还要录话筒声。 最后写成一个文件。
本人知道，假如录屏和声音分开录，最后合成一个文件，是没有问题的。但这样会比较慢。所以本人想边录边把声音和视频写成一个文件。
本人目前的做法是使用以下几个类来把流写到文件中。
AVAssetWriter ;
AVAssetWriterInput ;
AVAssetWriterInputPixelBufferAdaptor ;
其中，录视频的部分没有问题了。
音频部分，本人目前是参照 AVCaptureToAudioUnit 这个例子
通过AVCaptureAudioDataOutput，  AVCaptureSession，AVCaptureDeviceInput 的组合得到音频的CMSampleBufferRef
再通过 AVAssetWriterInput 写入文件。
但录出来的文件没有声音。
各位有经验的大牛给点思路吧。搞好多天了。