如何確定寫入（2）已將所有數據寫入套接字/文件描述符？

Question

我正在使用write（man 2寫入）將數據寫入具有已建立的阻塞和非常慢的TCP連接的套接字。我正在寫大量的數據。 write返回寫入的實際大小，當然它並不是所有的數據都是由於（可能）這個問題的範圍之外的原因而被寫入的。

可以肯定的，我封裝我寫的呼叫在一個小圈是這樣的：

do { 
    ssize_t ret = write(client, p, count); 
    if (ret <= 0) 
     break; 
    p += ret; 
    count -= ret; 
} while (count); 

if (count != 0) 
    return -ENODEV; 

有沒有更好的方式來做到這一點，喜歡上了文件描述符設置一個標誌，因而有下層處理呢？

Answer 1

size_t done; 
ssize_t ret; 

for (done = 0; done < size; done += ret) { 
    ret = write(client, buff + done, size-done); 

    if (ret == 0) return -ENODEV; 
    if (ret == -1 && errno == EINTR) { ret = 0; continue; } 
    if (ret == -1) return -errno; 
} 

Answer 2

我建議使用while循環而不是do {} while在count == 0的情況下具有一致的行爲。此外，一些失敗的案例並不錯誤：

while (count > 0) { 
    ssize_t ret = write(client, p, count); 
    if (ret <= 0) { 
     if (ret == 0) 
      return -ENODEV; 
     if (errno == EINTR) 
      continue; 
     else 
      return -errno; 
    }  
    p += ret; 
    count -= ret; 
} 

EINTR如果被寫入任何數據之前的系統調用被中斷signal設置。案件中應該重新啓動write。如果client句柄設置爲非阻塞，則還應該處理EAGAIN和EWOULDBLOCK。

從wildplasser的回答得出一個更爲簡潔和優雅的版本：

for (size_t done = 0; done < count;) { 
    ssize_t ret = write(client, p + done, count - done); 

    if (ret == 0) return -ENODEV; 
    if (ret < 0 && errno != EINTR) return -errno; 
    done += ret; 
}