求IOS录屏并录声音的方案

小弟最后要做一个功能，就是要录IOS设备的屏显示和声音输出，还要录话筒声。 最后写成一个文件。
我知道，如果录屏和声音分开录，最后合成一个文件，是没有问题的。但这样会比较慢。所以我想边录边把声音和视频写成一个文件。


我目前的做法是使用以下几个类来把流写到文件中。
 AVAssetWriter ;
AVAssetWriterInput ;
AVAssetWriterInputPixelBufferAdaptor ;

其中，录视频的部分没有问题了。
音频部分，我目前是参照 AVCaptureToAudioUnit 这个例子
通过AVCaptureAudioDataOutput，  AVCaptureSession，AVCaptureDeviceInput 的组合得到音频的CMSampleBufferRef
再通过 AVAssetWriterInput 写入文件。

但录出来的文件没有声音。

各位有经验的大侠给点思路吧。。。搞好多天了。