Sounds like you may benefit from an NDA with a non-compete clause. These will give you a contract stating that your co-founder will keep your code secret and not use it to compete against you if the relationship turns sour in the future.

Links for reference: http://en.wikipedia.org/wiki/Non-disclosure_agreement http://en.wikipedia.org/wiki/Non-compete_clause