修改代码，支持内存控制。