python下的socket常用方法举例

python下的socket





1、简单的server和client端的socket代码

server.py:





#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import socket

ip_port = ('127.0.0.1',9999)





sk = socket.socket()

sk.bind(ip_port)

sk.listen(1)





while 1:

    print 'waiting'

    conn,addr = sk.accept()

    conn.sendall('welcome to 10086')

    conn.close()





client.py:





#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import socket





ip_port = ('127.0.0.1',9999)





sk = socket.socket()

sk.connect(ip_port)





data = sk.recv(1024)

sk.close()





print data





分别在两个窗口运行：

# python server.py

# python client.py





2、多线程，阻塞模式

server.py:





#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import socket

# 阻塞式

sk = socket.socket()

ip_port = ('192.168.8.51',50001)





sk.bind(ip_port)

sk.listen(2)





while True:

    conn,addr = sk.accept()

    print 'get a data from ',addr

    while True:

        

        data = conn.recv(1024)

        if not data : break

        conn.send(data.upper())

        print 'recevie:',data

sk.close()





client.py:





#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import socket





ip_port = ('192.168.8.51',50001)

sk = socket.socket()

sk.connect(ip_port)





while True:

    input_data = raw_input('')

    sk.send(input_data)

    data = sk.recv(1024)

    print 'recevi:',data

    

sk.close()





3、模拟实现ssh





ssh_socket_server.py:





#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import SocketServer

import commands,time





class MySocksServer(SocketServer.BaseRequestHandler):

    

    def handle(self):

        print 'get a new conn from ',self.client_address

        

        while True:

            cmd = self.request.recv(1024)

            if not cmd:

                print 'lost connetion with',self.client_address

                break

            cmd_result = commands.getstatusoutput(cmd)

            # send result size

            self.request.send(str(len(cmd_result[1])))

            #self.sleep(0.2)

            

            # send result

            self.request.sendall(cmd_result[1])

            

if __name__ == '__main__':

    h = '0.0.0.0'

    p = 50003

    s = SocketServer.ThreadingTCPServer((h,p),MySocksServer)





    s.serve_forever()





socket_client:





#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import socket

from _mysql import result





ip_port = ('192.168.8.51',50003)





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





sk.connect(ip_port)





def recv_all(obj,msg_length):

    raw_result = ''

    while msg_length != 0:

        if msg_length <= 4096:

            data = obj.recv(msg_length)

            msg_length = 0

            

        else:

            data = obj.recv(4096)

            msg_length -= 4096

        raw_result += data

        

    return raw_result





while True:

    input_data = raw_input('')

    if len(input_data) == 0: break

    sk.sendall(input_data)

    

    # recv response size

    res_size = int(sk.recv(100))

    print 'data size from server',res_size

    

    result = recv_all(sk,res_size)

    print result





sk.close()





4、模拟实现ftp账号远程登录通过获取数据





ftp客户端

ftp_client.py:

#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import socket

import MySQLdb

import os

import hashlib

from _mysql import result





ip_port = ('192.168.50.18',50004)

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

sk.connect(ip_port)





#连接数据库函数

def conn_db():

    try:

        return MySQLdb.connect(host='localhost',user='root',passwd='',port=3306)        

    except MySQLdb.Error,e:  

        print "Mysql Error %d: %s" % (e.args[0], e.args[1])

        

# md5加密的函数

def md5_create(name):

    code = hashlib.md5()

    code.update(name)

    return code.hexdigest()

     

#验证用户名密码是否正确

def login(username,password):

        #连接数据库

        conn = conn_db()

        

        cur = conn.cursor()

        conn.select_db('python')

        

        cur.execute('select password from user where username = "%s"' % username)

        result = cur.fetchone()

        md5code = md5_create(password)

        

        #验证密码是否正确

        if result[0] == md5code:

            return True

        else:

            return False





def recv_all(obj,msg_length):

    raw_result = ''

    while msg_length != 0:

        if msg_length <= 4096:

            data = obj.recv(msg_length)

            msg_length = 0

            

        else:

            data = obj.recv(4096)

            msg_length -= 4096

        raw_result += data

        

    return raw_result





#获取数据函数

def get_all(self,obj,msg_length,dst_file):

       

        while msg_length != 0:

            if msg_length <= 4096:

                data = obj.recv(msg_length)

                msg_length = 0

                print 'to less ',msg_length

            else:

                data = obj.recv(4096)

                msg_length -= 4096

                print 'else ',msg_length

            dst_file.write(data)

        

        return 'done'





while True:

    user_input = raw_input('please input your username and password eg: user pwd ')

    if len(user_input) == 0:

        print 'your username or password is error'

        break

    

    username,password = user_input.split()

    #验证用户密码是否正确

    login_rs = login(username,password)

    

    if login_rs:

        while True:

            user_input = raw_input('msg to send::')

            if len(user_input) == 0: continue

            

            user_cmd = user_input.split()

            

            #如果命令为put则进行传输数据

            if user_cmd[0] == 'put':

                if len(user_cmd) == 2:

                    f = file(user_cmd[1] , 'rb')

                    f_size = os.stat(user_cmd[1]).st_size

                    sk.send("%s %s %s" %(user_cmd[0],user_cmd[1],f_size))

                    

                    print 'going to send...'

                    sk.sendall(f.read())

                    print sk.recv(1024)

                    

            else:

                sk.sendall(user_cmd[0])

                print 'else',user_cmd[0]

                res_size = int(sk.recv(100))

                print 'data size from server',res_size

                

                result = recv_all(sk,res_size)

                print result

                





sk.close()





服务器端：

ftp_server.py





#!/usr/bin/env python

#_*_ coding:utf-8 _*_





import SocketServer

import commands

import socket





class MySocksServer(SocketServer.BaseRequestHandler):

    

    def recv_all(self,obj,msg_length,dst_file):

       

        while msg_length != 0:

            if msg_length <= 4096:

                data = obj.recv(msg_length)

                msg_length = 0

                print 'to less ',msg_length

            else:

                data = obj.recv(4096)

                msg_length -= 4096

                print 'else ',msg_length

            dst_file.write(data)

        

        return 'done'





    def handle(self):

        print 'get a new conn from ',self.client_address

        

        while True:

            cmd = self.request.recv(1024)

            if not cmd:

                print 'lost connetion with',self.client_address

                break

            #option,filename,file_size = cmd.split()

            cmd_c = cmd.split()

            

            if cmd_c[0] == 'put':

                # client want to upload file

                f = file('recv/%s' % cmd_c[1],'wb')

                write_to_file = self.recv_all(self.request, int(cmd_c[2]), f)

                if write_to_file == 'done':

                    self.request.send('file upload success')

                    f.close()

                    

            else:

                print 'start send cmd back info'

                cmd_result = commands.getstatusoutput(cmd_c[0])

                self.request.send(str(len(cmd_result[1])))

                self.request.sendall(cmd_result[1])

            

if __name__ == '__main__':

    h = '0.0.0.0'

    p = 50004

    s = SocketServer.ThreadingTCPServer((h,p),MySocksServer)

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

    s.serve_forever()