Mexico to take over criminal probe of ex-security chief