这个禅宗公案的出处在哪里？

查到两个版本，仅供参考
《五灯会元?道悟传》：
道悟，渚宫人，姓崔氏，子玉之后胤也。年十五依长沙寺昙翥律师出家，二十三诣嵩山受戒，三十参石头，频沐指示，曾未投机，次谒忠国师。三十四与国师侍者应真南还谒马祖。……师常云：“快活！快活！”及临终时叫：“苦！苦！”又云：“阎罗王来取我也。”院主问曰：“和尚当时被节度使抛向水中，神色不动，如今何得恁么地？”师举枕子云：“汝道当时是？如今是？”院主无对，便入灭。