如何進(jìn)行socket通信

本篇文章為大家展示了如何進(jìn)行socket通信，內(nèi)容簡明扼要并且容易理解，絕對(duì)能使你眼前一亮，通過這篇文章的詳細(xì)介紹希望你能有所收獲。

一、使用socket實(shí)現(xiàn)一對(duì)一的簡單通信  

socket就是一個(gè)開往網(wǎng)絡(luò)應(yīng)用必備的功能模塊。通過這個(gè)模塊我們可以自己寫程序的server端和client端，可以自己定義server端對(duì)外提供服務(wù)器的ip地址和端口。學(xué)會(huì)使用這個(gè)模塊的之后我們不僅可以開往網(wǎng)絡(luò)程序，還可以利用socket模塊開發(fā)出端口掃描的程序。

先看一段創(chuàng)建服務(wù)器端的代碼

1. # -*- coding: utf-8 -*-
   

2. import socket
   

3. import os

4. 
   

5. #定義服務(wù)器端啟動(dòng)后要綁定的ip和端口
   

6. ip_port = ('0.0.0.0',9999)
   

7. # 創(chuàng)建socket對(duì)象并指定連接的網(wǎng)絡(luò)類型和傳輸協(xié)議
   

8. sk=sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
   

9. #綁定ip和端口號(hào)
   

10. sk.bind(ip_port)
    

11. #啟動(dòng)監(jiān)聽，并設(shè)置最多可以通知連入連接數(shù)
    

12. sk.listen(5)
    

13. #用while使程序一直運(yùn)行
    

14. while True:
    

15.     #print 'server waiting...'
    

16.     """
    

17.     進(jìn)入accpet阻塞狀態(tài)來等待客戶端的連接請(qǐng)求
    

18.     保存客戶端的連接狀態(tài)和客戶端的地址
    

19.     """
    

20.     conn,addr = sk.accept()
    

21.     print addr
    

22.     #如果有客戶端發(fā)來請(qǐng)求就每次都只接受1024個(gè)字節(jié)的內(nèi)容，注意recv()也是阻塞的
    

23.     client_data = conn.recv(10240)
    

24.     #打印從客戶端接收到的字符串
    

25.     print client_data                ----客戶端ip地址

26.     #向客戶端發(fā)送字符串
    

27.     conn.sendall("I`m server")
    

28.     #關(guān)閉連接
    

29.     conn.close()

客戶端代碼

1. # -*- coding: utf-8 -*-
   

2. import socket
   #指定server端的ip地址和端口號(hào)
   ip_port = ('127.0.0.1',9999)
   #創(chuàng)建socket實(shí)例，參數(shù)不填的話就采用ipv4協(xié)議通過tcp傳輸
   sk = socket.socket()
   #發(fā)起連接到server端
   sk.connect(ip_port)
   #給server端發(fā)消息
   sk.sendall("I`m client")
   #發(fā)送完成后接收server端的反饋
   server_reply = sk.recv(1024)
   #打印server端反饋的信息
   print server_reply
   #關(guān)閉連接
   sk.close()

3. 
   

通過以上兩段代碼的運(yùn)行，我們就實(shí)現(xiàn)了server端與client端的通信

在server端可以到如下運(yùn)行結(jié)果，接收到客戶端請(qǐng)求之后又恢復(fù)到了等待狀態(tài)

1. server waiting...
   

2. I`m client
   

3. server waiting...

在client端也看到了類似的結(jié)果，成功接收到了server端的發(fā)來的信息之后退出了程序

點(diǎn)擊(此處)折疊或打開

1. I`m server

拓展：實(shí)時(shí)解析客戶端的日志文件并發(fā)送至服務(wù)端保存

server端改寫

1. # -*- coding: utf-8 -*-
   

2. import socket
   

3. import os
   

4. 
   

5. def save_to_txt(addr,data):
   

6.                 path='/tmp/'+addr+'/slow.log'
   

7.                 file_path = os.path.split(path)
   

8.                 if not os.path.exists( file_path[0] ):
   

9.                         os.makedirs( file_path[0] )
   

10.                         print '創(chuàng)建目錄成功'
    

11.                 if len(data)>0:
    

12.                         try:
    

13.                                 with open( path.encode('utf-8'),'ab' ) as f:
    

14.                                         data = data+"\n"
    

15.                                         f.write( data.encode('utf-8') )
    

16.                                         
    

17.                         except:
    

18.                                 print '數(shù)據(jù)插入失敗'
    

19.                 else:
    

20.                         print '沒有數(shù)據(jù)'
    

21. 
    

22. 
    

23. #定義服務(wù)器端啟動(dòng)后要綁定的ip和端口
    

24. ip_port = ('0.0.0.0',9999)
    

25. # 創(chuàng)建socket對(duì)象并指定連接的網(wǎng)絡(luò)類型和傳輸協(xié)議
    

26. sk=sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    

27. #綁定ip和端口號(hào)
    

28. sk.bind(ip_port)
    

29. #啟動(dòng)監(jiān)聽，并設(shè)置最多可以通知連入連接數(shù)
    

30. sk.listen(5)
    

31. #用while使程序一直運(yùn)行
    

32. while True:
    

33.     #print 'server waiting...'
    

34.     """
    

35.     進(jìn)入accpet阻塞狀態(tài)來等待客戶端的連接請(qǐng)求
    

36.     保存客戶端的連接狀態(tài)和客戶端的地址
    

37.     """
    

38.     conn,addr = sk.accept()
    

39.     print addr
    

40.     #如果有客戶端發(fā)來請(qǐng)求就每次都只接受1024個(gè)字節(jié)的內(nèi)容，注意recv()也是阻塞的
    

41.     client_data = conn.recv(10240)
    

42.     #打印從客戶端接收到的字符串
    

43.     print client_data
    

44.     save_to_txt(addr[0],client_data) ----接送并寫入日志文件
    

45.     #向客戶端發(fā)送字符串
    

46.     conn.sendall("I`m server")
    

47.     #關(guān)閉連接
    

48.     conn.close()

client改寫

1. # -*- coding: utf-8 -*-
   

2. import socket
   

3. def sendtxt(line):
   

4.     ip_port = ('192.168.2.136',9999)
   

5.     #創(chuàng)建socket實(shí)例，參數(shù)不填的話就采用ipv4協(xié)議通過tcp傳輸
   

6.     sk = socket.socket()
   

7.     #發(fā)起連接到server端
   

8.     sk.connect(ip_port)
   

9.     #給server端發(fā)消息
   

10.     sk.sendall(line)
    

11.     #發(fā)送完成后接收server端的反饋
    

12.     server_reply = sk.recv(10240)
    

13.     #打印server端反饋的信息
    

14.     print server_reply
    

15.     #關(guān)閉連接
    

16.     sk.close()
    

17. 
    

18. 
    

19. import fileinput
    

20. import time
    

21. import os
    

22. 
    

23. target_file = 'log.txt'
    

24. init_flag = True  # 初次加載程序
    

25. 
    

26. record_count = 0
    

27. 
    

28. while True:
    

29.         if init_flag:
    

30.             #讀取整個(gè)文件
    

31.             # for eachline in fileinput.input(target_file):
    

32.             #    print eachline
    

33.             #    record_count += 1
    

34.             total_count = os.popen('wc -l %s' % target_file).read().split()[0]
    

35.             total_count = int(total_count)
    

36.             record_count = total_count
    

37. 
    

38.             init_flag = False
    

39.         else:
    

40.             #如果總行數(shù)小于當(dāng)前行，那么認(rèn)為文件更新了，從第一行開始讀。
    

41.             total_count = os.popen('wc -l %s' % target_file).read().split()[0]
    

42.             total_count = int(total_count)
    

43.             if total_count < record_count:
    

44.                 record_count = 0
    

45. 
    

46.             for eachline in fileinput.input(target_file):
    

47.                 line_no = fileinput.filelineno()
    

48.                 if line_no > record_count:
    

49.                    if len(eachline.strip('\n')) > 0:
    

50.                        print eachline.strip('\n')
    

51.                        sendtxt(eachline.strip('\n')) --發(fā)送

52.                        record_count += 1
    

53.                    else:
    

54.                        sendtxt('        ')
    

55.                        record_count += 1

上述內(nèi)容就是如何進(jìn)行socket通信，你們學(xué)到知識(shí)或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識(shí)儲(chǔ)備，歡迎關(guān)注億速云行業(yè)資訊頻道。