According to SSA you may not work while on disability UNLESS you enter an approved return to work program which, once completed, you will be expected to leave disability (SSD), and continue working. If your well enough to work for the 9 month program you are not sick enough for being considered total disability.
for SSI, yes, you can work but must discuss with SSA first.