Only problem is if they are working in the official capacity the DOJ can elect to transfer the case to the Federal system then decline to prosecute. I believe it is 28 U.S.C. § 1442
uscode.house.gov/view.xhtml?hl=false&edition=prelim&req=granuleid%3AUSC-prelim-title28-section1442&num=0&saved=%7CZ3JhbnVsZWlkOlVTQy1wcmVsaW0tdGl0bGUyOC1zZWN0aW9uMTQ0MQ%3D%3D%7C%7C%7C0%7Cfalse%7Cprelim