Law firms operating in China are walking a geopolitical tightrope.