电脑技术学习

在php上使用fork以及socket的sample

dn001
最近剛好遇到一個頭大的問題寫了這個code讓大家參考一下吧
家裏的無線AP功能不太好,他只提供把外部真實IP map 到 Nat裡面的某個IP
不能指定某個port map到某個內部IP的Port
可是我已經把外部的IP Map到內部的Linux Server上,
但是我又想從外部使用VNC連到內部的一台Windows電腦。
所以就寫了這個程式
原理是這樣

這個程式會在Linux Server上開一個Port作Listen的動作
當外部連到這個Port時,程式會再開啟另一個連線連到內部Windows的VNC上
把外部的封包原封不動的丟到VNC的連線上,然後把VNC連線傳回的資料原封不動的再丟回外部的Port

程式碼:

#!/usr/bin/php -q
<?php
$IP
='192.168.1.1'
//Windows電腦的IP
$Port='5900'
//VNC使用的Port
$ServerPort='9999'
//Linux Server對外使用的Port
$RemoteSocket=false
//連線到VNC的Socket
functionSignalFunction&
#40;$Signal&#41;
&
#123;
//這是主Process的訊息處理函數
global$PID
//Child Process的PID
switch &
#40;$Signal&#41;
&
#123;
caseSIGTRAP&
#58;
caseSIGTERM&
#58;
//收到結束程式的Signal
if&
#40;$PID&#41;
&
#123;
//送一個SIGTERM的訊號給Child告訴他趕快結束掉嘍
posix_kill&
#40;$PID,SIGTERM&#41;;
//等待Child Process結束,避免zombie
pcntl_wait&
#40;$Status&#41;;
&
#125;
//關閉主Process開啟的Socket
DestroySocket&
#40;&#41;;
exit&
#40;0&#41//結束主Process
break;
case
SIGCHLD&
#58;
/*
當Child Process結束掉時,Child會送一個SIGCHLD訊號給Parrent
當Parrent收到SIGCHLD,就知道Child Process已經結束嘍 ,該做一些
結束的動作*/
unset&
#40;$PID&#41//將$PID清空,表示Child Process已經結束
pcntl_wait&
#40;$Status&#41//避免Zombie
break;
default&
#58;
&
#125;
&
#125;
functionChildSignalFunction&
#40;$Signal&#41;
&
#123;
//這是Child Process的訊息處理函數
switch &
#40;$Signal&#41;
&
#123;
caseSIGTRAP&
#58;
caseSIGTERM&
#58;
//Child Process收到結束的訊息
DestroySocket&
#40;&#41//關閉Socket
exit&
#40;0&#41//結束Child Process
default&
#58;
&
#125;
&
#125;
functionProcessSocket&
#40;$ConnectedServerSocket&#41;
&
#123;
//Child Process Socket處理函數
//$ConnectedServerSocket -> 外部連進來的Socket
global$ServerSocket,$RemoteSocket,$IP,$Port

$ServerSocket=$ConnectedServerSocket

declare&
#40;ticks = 1&#41//這一行一定要加,不然沒辦法設定訊息處理函數。
//設定訊息處理函數
if&
#40;!pcntl_signal&#40;SIGTERM, "ChildSignalFunction"&#41;&#41; return;
if&
#40;!pcntl_signal&#40;SIGTRAP, "ChildSignalFunction"&#41;&#41; return;
//建立一個連線到VNC的Socket
$RemoteSocket=socket_create&
#40;AF_INET, SOCK_STREAM,SOL_TCP&#41;;
//連線到內部的VNC
@$RemoteConnected=socket_connect&
#40;$RemoteSocket,$IP,$Port&#41;;
if&
#40;!$RemoteConnected&#41; return; //無法連線到VNC 結束
//將Socket的處理設為Nonblock,避免程式被Block住
if&
#40;!socket_set_nonblock&#40;$RemoteSocket&#41;&#41; return;
if&
#40;!socket_set_nonblock&#40;$ServerSocket&#41;&#41; return;
while&
#40;true&#41;
&
#123;
//這邊我們採用pooling的方式去取得資料
$NoRecvData=false
//這個變數用來判別外部的連線是否有讀到資料
$NoRemoteRecvData=false
//這個變數用來判別VNC連線是否有讀到資料
@$RecvData=socket_read&
#40;$ServerSocket,4096,PHP_BINARY_READ&#41;;
//從外部連線讀取4096 bytes的資料
@$RemoteRecvData=socket_read&
#40;$RemoteSocket,4096,PHP_BINARY_READ&#41;;
//從vnc連線連線讀取4096 bytes的資料
if&
#40;$RemoteRecvData===''&#41;
&
#123;
//VNC連線中斷,該結束嘍
echo"Remote Connection Closen"

return;;;;
&
#125;
if&
#40;$RemoteRecvData===false&#41;
&
#123;
/*
由於我們是採用nonblobk模式
這裡的情況就是vnc連線沒有可供讀取的資料
*/
$NoRemoteRecvData=true

//清除掉Last Errror
socket_clear_error&
#40;$RemoteSocket&#41;;
&
#125;
if&
#40;$RecvData===''&#41;
&
#123;
//外部連線中斷,該結束嘍
echo"Client Connection Closen"

return;
&
#125;
if&
#40;$RecvData===false&#41;
&
#123;
/*
由於我們是採用nonblobk模式
這裡的情況就是外部連線沒有可供讀取的資料
*/
$NoRecvData=true

//清除掉Last Errror
socket_clear_error&
#40;$ServerSocket&#41;;
&
#125;
if&
#40;$NoRecvData&&$NoRemoteRecvData&#41;
&
#123;
//如果外部連線以及VNC連線都沒有資料可以讀取時,
//就讓程式睡個0.1秒,避免長期佔用CPU資源
usleep&
#40;100000&#41;;
//睡醒後,繼續作pooling的動作讀取socket
continue;
&
#125;
//Recv Data
if&
#40;!$NoRecvData&#41;
&
#123;
//外部連線讀取到資料
while&
#40;true&#41;
&
#123;
//把外部連線讀到的資料,轉送到VNC連線上
@$WriteLen=socket_write&
#40;$RemoteSocket,$RecvData&#41;;
if&
#40;$WriteLen===false&#41;
&
#123;
//由於網路傳輸的問題,目前暫時無法寫入資料
//先睡個0.1秒再繼續嘗試。
usleep&
#40;100000&#41;;
continue;
&
#125;
if&
#40;$WriteLen===0&#41;
&
#123;
//遠端連線中斷,程式該結束了
echo"Remote Write Connection Closen"

return;
&
#125;
//從外部連線讀取的資料,已經完全送給VNC連線時,中斷這個迴圈。
if&
#40;$WriteLen==strlen&#40;$RecvData&#41;&#41; break;
//如果資料一次送不完就得拆成好幾次傳送,直到所有的資料全部送出為止
$RecvData=substr&
#40;$RecvData,$WriteLen&#41;;
&
#125;
&
#125;
if&
#40;!$NoRemoteRecvData&#41;
&
#123;
//這邊是從VNC連線讀取到的資料,再轉送回外部的連線
//原理跟上面差不多不再贅述
while&
#40;true&#41;
&
#123;
@$WriteLen=socket_write&
#40;$ServerSocket,$RemoteRecvData&#41;;
if&
#40;$WriteLen===false&#41;
&
#123;
usleep&
#40;100000&#41;;
continue;
&
#125;
if&
#40;$WriteLen===0&#41;
&
#123;
echo"Remote Write Connection Closen"

return;
&
#125;
if&
#40;$WriteLen==strlen&#40;$RemoteRecvData&#41;&#41; break;
$RemoteRecvData=substr&
#40;$RemoteRecvData,$WriteLen&#41;;
&
#125;
&
#125;
&
#125;
&
#125;
functionDestroySocket&
#40;&#41;
&
#123;
//用來關閉已經開啟的Socket
global$ServerSocket,$RemoteSocket

if&
#40;$RemoteSocket&#41;
&
#123;
//如果已經開啟VNC連線
//在Close Socket前必須將Socket shutdown不然對方不知到你已經關閉連線了
@socket_shutdown&
#40;$RemoteSocket,2&#41;;
socket_clear_error&
#40;$RemoteSocket&#41;;
//關閉Socket
socket_close&
#40;$RemoteSocket&#41;;;;;
&
#125;
//關閉外部的連線
@socket_shutdown&
#40;$ServerSocket,2&#41;;
socket_clear_error&
#40;$ServerSocket&#41;;
socket_close&
#40;$ServerSocket&#41;;
&
#125;
//這裡是整個程式的開頭,程式從這邊開始執行
//這裡首先執行一次fork
$PID=pcntl_fork&
#40;&#41;;
if&
#40;$PID==-1&#41; die&#40;"could not fork"&#41;;
//如果$PID不為0表示這是Parrent Process
//$PID就是Child Process
//這是Parrent Process 自己結束掉,讓Child成為一個Daemon。
if&
#40;$PID&#41; die&#40;"Daemon PID&#58;$PIDn"&#41;;
//從這邊開始,就是Daemon模式在執行了
//將目前的Process跟終端機脫離成為daemon模式
if&
#40;!posix_setsid&#40;&#41;&#41; die&#40;"could not detach from terminaln"&#41;;
//設定daemon 的訊息處理函數
declare&
#40;ticks = 1&#41;;
if&
#40;!pcntl_signal&#40;SIGTERM, "SignalFunction"&#41;&#41; die&#40;"Error!!!n"&#41;;
if&
#40;!pcntl_signal&#40;SIGTRAP, "SignalFunction"&#41;&#41; die&#40;"Error!!!n"&#41;;
if&
#40;!pcntl_signal&#40;SIGCHLD, "SignalFunction"&#41;&#41; die&#40;"Error!!!n"&#41;;
//建立外部連線的Socket
$ServerSocket=socket_create&
#40;AF_INET, SOCK_STREAM,SOL_TCP&#41;;
//設定外部連線監聽的IP以及Port,IP欄位設0,表示經聽所有介面的IP
if&
#40;!socket_bind&#40;$ServerSocket,0,$ServerPort&#41;&#41; die&#40;"Cannot Bind Socket!n"&#41;;
//開始監聽Port
if&
#40;!socket_listen&#40;$ServerSocket&#41;&#41; die&#40;"Cannot Listen!n"&#41;;
//將Socket設為nonblock模式
if&
#40;!socket_set_nonblock&#40;$ServerSocket&#41;&#41; die&#40;"Cannot Set Server Socket to Block!n"&#41;;
//清空$PID變數,表示目前沒有任何的Child Process
unset&
#40;$PID&#41;;
while&
#40;true&#41;
&
#123;
//進入pooling模式,每隔1秒鐘就去檢查有沒有連線進來。
sleep&
#40;1&#41;;
//檢查有沒有連線進來
@$ConnectedServerSocket=socket_accept&
#40;$ServerSocket&#41;;
if&
#40;$ConnectedServerSocket!==false&#41;
&
#123;
//有人連進來嘍
//起始一個Child Process用來處理連線
$PID=pcntl_fork&
#40;&#41;;
if&
#40;$PID==-1&#41; die&#40;"could not fork"&#41;;
if&
#40;$PID&#41; continue;//這是daemon process,繼續回去監聽。
//這裡是Child Process開始
//執行Socket裡函數
ProcessSocket&
#40;$ConnectedServerSocket&#41;;
//處理完Socket後,結束掉Socket
DestroySocket&
#40;&#41;;
//結束Child Process
exit&
#40;0&#41;;
&
#125;
&
#125;
?>

标签: