Я думаю, у вас есть переменная-член, поддерживающая состояние вашего соединения. при успешном соединении установите для него значение true, запустите поток, который всегда будет читать байты из входного потока сокетов, и, если вы получаете IOException в этом потоке, установите для флага значение false.