前幾天用PHP寫一個socket網絡服務,在文檔裡看到socket_read和socket_recv這兩個方法時有點暈,乍一看這不是一樣的嘛,干嗎還要給兩個不同的用法呢。看文檔沒看太明白,看了下源碼才搞清楚,在這裡記錄一下。
先看一下這兩個函數的聲明:
復制代碼 代碼如下:
string socket_read ( resource $socket , int $length [, int $type = PHP_BINARY_READ ] )
int socket_recv ( resource $socket , string &$buf , int $len , int $flags )
可以看到,從聲明可以看到,一個是把收到的數據通過執行結果返回,另一個是把收到的數據通過引用的形式返回。另一個區別就是,socket_read多了一個type,socket_recv多了一個flags(夠混亂的)。我們先來看看socket_recv的源碼吧!
復制代碼 代碼如下:
PHP_FUNCTION(socket_recv)
{
zval *php_sock_res, *buf;
char *recv_buf;
php_socket *php_sock;
int retval;
long len, flags;
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "rzll", &php_sock_res, &buf, &len, &flags) == FAILURE) {
return;
}
ZEND_FETCH_RESOURCE(php_sock, php_socket *, &php_sock_res, -1, le_socket_name, le_socket);
/* overflow check */
if ((len + 1) < 2) {
RETURN_FALSE;
}
recv_buf = emalloc(len + 1);
memset(recv_buf, 0, len + 1);
if ((retval = recv(php_sock->bsd_socket, recv_buf, len, flags)) < 1) {
efree(recv_buf);
zval_dtor(buf);
Z_TYPE_P(buf) = IS_NULL;
} else {
recv_buf[retval] = '\0';
/* Rebuild buffer zval */
zval_dtor(buf);
Z_STRVAL_P(buf) = recv_buf;
Z_STRLEN_P(buf) = retval;
Z_TYPE_P(buf) = IS_STRING;
}
if (retval == -1) {
PHP_SOCKET_ERROR(php_sock, "unable to read from socket", errno);
RETURN_FALSE;
}
RETURN_LONG(retval);
}
啰裡啰嗦一大堆,其實有一行最關鍵:
復制代碼 代碼如下:
if ((retval = recv(php_sock->bsd_socket, recv_buf, len, flags)) < 1) {
可以看到,實際上這個函數就是調用了系統的recv而已,只是把輸入參數和得到的結果都處理了一下,比較好理解。那我們再來看下socket_read,socket_read比系統的recv函數多了一個$type參數,這也是我認為這個函數存在的意義,從文檔裡可以看到,type有兩個值,分別是PHP_BINARY_READ和PHP_NORMAL_READ,文檔裡有寫,PHP_BINARY_READ表示直接用系統的recv方法,PHP_NORMAL_READ表示會一讀,直到遇到\n 或者 \r,我們來看下源碼:
復制代碼 代碼如下:
//省略一大堆
if (type == PHP_NORMAL_READ) {
retval = php_read(php_sock, tmpbuf, length, 0);
} else {
retval = recv(php_sock->bsd_socket, tmpbuf, length, 0);
}
可以看到,如果是PHP_NORMAL_READ模式,其實行為和socket_recv是一樣的,都是用的系統的recv函數,但是如果是PHP_NORMAL_READ,則有很大區別,用了自己實現的php_read函數,那這個php_read是干啥的呢?我們繼續看源碼:
復制代碼 代碼如下:
*t = '\0';
while (*t != '\n' && *t != '\r' && n < maxlen) {
if (m > 0) {
t++;
n++;
} else if (m == 0) {
no_read++;
if (nonblock && no_read >= 2) {
return n;
/* The first pass, m always is 0, so no_read becomes 1
* in the first pass. no_read becomes 2 in the second pass,
* and if this is nonblocking, we should return.. */
}
if (no_read > 200) {
set_errno(ECONNRESET);
return -1;
}
}
if (n < maxlen) {
m = recv(sock->bsd_socket, (void *) t, 1, flags);
}
if (errno != 0 && errno != ESPIPE && errno != EAGAIN) {
return -1;
}
set_errno(0);
}
還是指copy了關鍵部分,可以看到,這裡的實現是一直循環調用recv,直到遇到\r或者\n或者讀的數據長度到了指定的maxlen。
雖然這兩個函數比較混亂,但是看到這裡應該明白了吧!好了睡覺去啦!