使用Pycurl调用Arbor抗DDos设备API接口

2021-08-01  本文已影响0人  净坛使者_猪悟能

一、Pycurl介绍

  pycurl是从python语言中调用libcurl的接口模块。libcurl是一组功能丰富的网络客户端库,支持许多应用层的网络协议,如HTTP,HTTPS, FTP,IMAP,SMTP等。
安装Pycurl

pip install pycurl

基本使用

import pycurl
c = pycurl.Curl()#创建一个HTTP客户端
url = 'http://www.jd.com/'
c.setopt(pycurl.URL, url)#设置要访问的url
##设置请求头
c.setopt(pycurl.HTTPHEADER, [
        'User-Agent: Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/29.0.1547.76 Safari/537.36',
        'Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8'
        'Accept-Encoding: gzip,deflate',
        'Connection: keep-alive',
     ])
##发送请求,访问url
  try:
        c.perform()
        status_code = c.getinfo(c.HTTP_CODE)
        # 处理网页的代码...
    except Exception as e:
        # raise e
        pass

Get方法
  PycURL的简单网络操作是使用其URL从给定服务器检索信息。 这被称为GET请求,因为它用于get网络资源。
  通过导入BytesIO模块并创建其对象,可以使用PycURL执行简单的GET请求。 创建了CURL对象以通过URL传输数据和文件。使用setopt()函数(用作setopt(option, value))设置所需的URL。 option参数指定要设置的选项,例如 URL,WRITEDATA等,并且value参数指定提供给该特定选项的值。然后将从设置的URL中检索的数据以字节的形式写入BytesIO对象。 然后使用getvalue()函数从BytesIO对象中读取字节,然后将其解码以将HTML打印到控制台。

import pycurl
from io import BytesIO
headers = ["X-Arbux-APIToken:MLJzvJw5NJ3Jf4zWSGivHu_MMepAx5MspMbt3Xb_"]
b_obj = BytesIO()
crl = pycurl.Curl()
# 设置URl
crl.setopt(crl.URL, 'https://192.168.5.243/api/aps/v3/protection-groups/denied-hosts/')

# 设置使用bytes类型写入数据,编码为UTF-8
crl.setopt(crl.WRITEDATA, b_obj)

#不对SSL证书进行验证
crl.setopt(pycurl.SSL_VERIFYPEER, False)
crl.setopt(pycurl.SSL_VERIFYHOST, False)

crl.setopt(pycurl.HTTPHEADER,headers)
crl.perform()
crl.close()

#读取缓存中的资源并解码
get_body = b_obj.getvalue()
print('Output of GET request:\n%s ' % get_body.decode('utf8'))

Post方法
  POST请求是通过将数据封装在HTTP请求的主体中来将数据发送到Web服务器的请求。 当您上传文件或提交表单时,基本上是在向指定服务器发送POST请求。
  可以使用PycURL执行POST请求,方法是首先设置URL,以通过setopt函数将表单数据发送到该URL。 首先以字典的形式(以键值对的形式)存储要提交的数据,然后使用urllib.parse模块中的urlencode函数对URL进行编码。我们在发送表单数据时使用POSTFIELDS选项,因为它会自动将HTTP请求方法设置为POST,并且也会处理pf数据。

import pycurl,json

my_user = 'admin'
my_password = 'Admin123'
url='https://192.168.5.243/api/aps/v3/protection-groups/denied-hosts/'
data = {"pgid":-1, "hostAddress":"3.3.3.3","annotation":"Quad-one Malware"}
headers = ['Content-Type:application/json',"X-Arbux-APIToken:MLJzvJw5NJ3Jf4zWSGivHu_MMepAx5MspMbt3Xb_"]

crl = pycurl.Curl()
crl.setopt(crl.URL, url)

crl.setopt(pycurl.SSL_VERIFYPEER, False)
crl.setopt(pycurl.SSL_VERIFYHOST, False)
#使用post提交数据
data_post = json.dumps(data)
crl.setopt(pycurl.HTTPHEADER,headers)
crl.setopt(crl.POSTFIELDS, data_post)
crl.setopt(pycurl.POST, 1)

crl.perform()
crl.close()


二、具体调用脚本

  下面脚本可调用Arbor设备的API接口提交Deny IP,并查看当前denied-hosts列表。

import pycurl,json
from io import BytesIO

my_token = 'MLJzvJw5NJ3Jf4zWSGivHu_MMepAx5MspMbt3Xb_'
my_host="192.168.5.243"

curls=[
    {
        'uri':'/api/aps/v3/protection-groups/denied-hosts/',
        'data': {"pgid":-1, "hostAddress":["2.2.2.2",'11.2.1.12'],"annotation":"Quad-one Malware"},
        'post': 1,
    },
    {
        'uri': '/api/aps/v3/protection-groups/denied-urls/',
        'data': {"pgid": -1, "url": "maps.google.com/maps", "annotation": "Google Maps"},
        'post': 1,
    },
]

def get_url(uri):
    '''
        构建一个URL
    '''
    sep = ''
    if not uri.startswith('/'):
        sep = '/'
    return 'https://{0}{1}{2}'.format(my_host, sep, uri)

def set_curl_opts(crl):
    '''
        设置与REST API交互的选项(不对SSL证书进行校验)
    '''
    crl.setopt(pycurl.SSL_VERIFYPEER, False)
    crl.setopt(pycurl.SSL_VERIFYHOST, False)

def my_get(curl):
    '''
        处理GET请求
        curl:传入URL地址
    '''
    headers = ["X-Arbux-APIToken:%s" % my_token]
    b_obj = BytesIO()
    crl = pycurl.Curl()#为GET请求定义一个请求链接
    crl.setopt(crl.URL, curl)
    crl.setopt(pycurl.HTTPHEADER, headers)
    set_curl_opts(crl)
    crl.setopt(crl.WRITEDATA, b_obj)
    crl.perform()
    crl.close()
    get_body = b_obj.getvalue()
    print('\nOutput of GET request:>>>\n%s ' % get_body.decode('utf8'))

def my_post(curl,post,data_in):
    '''
        处理POST请求
    '''
    headers = ['Content-Type:application/json', "X-Arbux-APIToken:%s"%my_token]
    crl.setopt(crl.URL, curl)
    set_curl_opts(crl)
    data_post = json.dumps(data_in)
    crl.setopt(pycurl.HTTPHEADER, headers)
    crl.setopt(crl.POSTFIELDS, data_post)
    crl.setopt(pycurl.POST, post )
    crl.perform()

if __name__== '__main__':

    crl = pycurl.Curl()
    for urlblock in curls:
        uri,data,post = urlblock['uri'],urlblock['data'],urlblock['post']
        url = get_url(uri)
        print(url)
        if post ==1:
            my_post(url,post,data)
        my_get( url)
    crl.close()

上一篇下一篇

猜你喜欢

热点阅读