Socket in Python

Socket in Python

   凌晨12:56       Rain Lin

---

之前就稍微學過c的socket program, 最近剛好再接觸python,也就順道一起研究
這次會介紹透過ip互相溝通與process間互相溝通的兩種寫法(寫法幾乎相同 只差在參數的調用部分)


1.透過ip互相溝通:

Server code:

#!/usr/bin/python

import socket,os

HOST = ''                 

PORT = 5566              

if __name__ == '__main__':

        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #註1

        s.bind((HOST, PORT)) #註2

        s.listen(5) #註3

        while(1):

                conn, addr = s.accept() #註4

                print 'Connected by', addr

                try:

                        conn.settimeout(10) #註5

                        data = conn.recv(1024) #註6

                        if not data:

                                print 'Error'

                                conn.send('Error')

                        elif data =='exit':

                                conn.send('bye')

                                break

                        else:

                                print "======start====="

                                os.system(data)

                                conn.send('Executed '+data) #註7

                except socket.timeout:

                        print 'time out'

                conn.close()

註1:AF_INET表示使用internet來進行傳輸，SOCK_STREAM表示使用TCP協定
註2:將socket綁定到指定的HOST和PORT
註3:設定server的隊列大小
註4:等待client連接
註5:設定timeout時間為10秒
註6:接收command的最大size為1024byte
註7:送字串傳去client

Client code:

#!/usr/bin/python

import socket,sys

if __name__ == '__main__':

        if len(sys.argv) < 3:

                print 'please check the insertion'

        elif sys.argv[2].strip() == '':

                print 'please check the command'

        else:

                HOST = sys.argv[1]    

                PORT = 5566          

   

                s = socket.socket(socket.AF_INET,

socket.SOCK_STREAM)

                s.connect((HOST, PORT))

                s.sendall(sys.argv[2])

                data = s.recv(1024)

                s.close()

                print 'Received', repr(data)

send和sendall 的差別：
send: 最後會回傳總共傳送了多少byte,如果只有一些字元被傳送,application會繼續負責把剩餘的資料傳完。
sendall: 這個函數會負責把文字傳送到另一端，但如果沒傳送完全會跳excepcttion，且沒辦法知道總共傳送了多少字元，回傳0

是代表傳送成功。



2.process間互相溝通:

Server code:

#!/usr/bin/python

import socket,os,sys

if __name__ == '__main__':

        sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) #註8

        if os.path.exists('/mnt/pc.d'): #註9

                os.unlink('/mnt/pc.d')

        sock.bind('/mnt/pc.d')

        sock.listen(5)

        while True:

                conn, addr = sock.accept()

                data = conn.recv(1024)

                os.system(data)

                conn.send('Execute '+data)

                conn.close()

Client code:

#!/usr/bin/python

import socket,time,sys

if __name__ == '__main__':

        sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)

        sock.connect("/mnt/pc.d")

        sock.send(sys.argv[1])

        print sock.recv(1024)

        sock.close()

#註8: AF_UNIX 是用於同一台電腦中process與process互相溝通時所調用的參數
#註9: 這邊會把HOST和PORT轉換成指定到電腦上的某一個檔案 所以要保證此檔案不存在 不然會發生錯誤


總結:
python程式比起c來說真的是好寫太多，c要用近百行的程式碼來寫socket，而python只需十幾行，真的很簡單，但是python是沒有multi-thread的，所以在loading重的情況下，容易造成lag，要改成multi-process。

Read More