Previously an error in a TCP write might accidentally get covered up as the
`write` function didn't check for `State::Error`. This updates that logic to
propagate the error out to ensure if we see an error it goes upwards.