你會(huì)用Linux網(wǎng)絡(luò)編程實(shí)現(xiàn)-readn、writen、readline等函數(shù)？

readn函數(shù)功能：在網(wǎng)絡(luò)編程的讀取數(shù)據(jù)中，通常會(huì)需要用到一個(gè)讀指定字節(jié)才返回的函數(shù)，linux系統(tǒng)調(diào)用中沒有給出，需要自己封裝。

readn實(shí)現(xiàn)代碼：

int readn(int fd, void *vptr, size_t n)
{
? ? size_t? ? ? ? ? nleft = n;? ? ? ? ? //readn函數(shù)還需要讀的字節(jié)數(shù)
? ? ssize_t? ? ? ? nread = 0;? ? ? ? ? //read函數(shù)讀到的字節(jié)數(shù)
? ? unsigned char? *ptr = (char *)vptr; //指向緩沖區(qū)的指針

while (nleft > 0)
? ? {
? ? ? ? nread = read(fd, ptr, nleft);
? ? ? ? if (-1 == nread)
? ? ? ? {
? ? ? ? ? ? if (EINTR == errno)
? ? ? ? ? ? ? ? nread = 0;
? ? ? ? ? ? else
? ? ? ? ? ? ? ? return -1;
? ? ? ? }
? ? ? ? else if (0 == nread)
? ? ? ? {
? ? ? ? ? ? break;
? ? ? ? }
? ? ? ? nleft -= nread;
? ? ? ? ptr += nread;
? ? }
? ? return n - nleft;
}

writen函數(shù)功能：讀滿n個(gè)字節(jié)才返回

writen代碼實(shí)現(xiàn)：

int writen(int fd, const void *vptr, size_t n)
{
? ? size_t? ? ? ? ? nleft = n;? //writen函數(shù)還需要寫的字節(jié)數(shù)
? ? ssize_t? ? ? ? nwrite = 0; //write函數(shù)本次向fd寫的字節(jié)數(shù)
? ? const char*? ? ptr = vptr; //指向緩沖區(qū)的指針

while (nleft > 0)
? ? {
? ? ? ? if ((nwrite = write(fd, ptr, nleft)) <= 0)
? ? ? ? {
? ? ? ? ? ? if (nwrite < 0 && EINTR == errno)
? ? ? ? ? ? ? ? nwrite = 0;
? ? ? ? ? ? else
? ? ? ? ? ? ? ? return -1;
? ? ? ? }
? ? ? ? nleft -= nwrite;
? ? ? ? ptr += nwrite;
? ? }
? ? return n;
}

readline函數(shù)功能：讀到'\n'或者讀滿緩沖區(qū)才返回

readline函數(shù)實(shí)現(xiàn)：

static ssize_t readch(int fd, char *ptr)
{
? ? static int? ? ? ? ? count = 0;
? ? static char*? ? ? ? read_ptr = 0;
? ? static char? ? ? ? read_buf[1024*4] = {0};

if (count <= 0)
? ? {
? ? again:
? ? ? ? count = read(fd, read_buf, sizeof(read_buf));
? ? ? ? if (-1 == count)
? ? ? ? ? ? if (EINTR == errno)
? ? ? ? ? ? ? ? goto again;
? ? ? ? ? ? else
? ? ? ? ? ? ? ? return -1;
? ? ? ? else if (0 == count)
? ? ? ? ? ? return 0;
? ? ? ? read_ptr = read_buf;
? ? }
? ? count--;
? ? *ptr = *read_ptr++;
? ? return 1;
}

ssize_t readline(int fd, void *vptr, size_t maxlen)
{
? ? ssize_t? ? ? ? i = 0;
? ? ssize_t? ? ? ? ret = 0;
? ? char? ? ? ? ? ? ch = '\0';
? ? char*? ? ? ? ? ptr = NULL;

ptr = (char *)vptr;

for (i = 1; i < maxlen; ++i)
? ? {
? ? ? ? ret = readch(fd, &ch);
? ? ? ? if (1 == ret)
? ? ? ? {
? ? ? ? ? ? *ptr++ = ch;
? ? ? ? ? ? if ('\n' == ch)
? ? ? ? ? ? ? ? break;
? ? ? ? }
? ? ? ? else if (0 == ret)
? ? ? ? {
? ? ? ? ? ? *ptr = 0;
? ? ? ? ? ? return i-1;
? ? ? ? }
? ? ? ? else
? ? ? ? ? ? return -1;
? ? }
? ? *ptr = 0;
? ? return i;
}

?