目前,我正在使用基本的 API 從雲端安全供應商取得日誌,但這不是一個非常乾淨的過程。我想過渡到他們提供的事件流 (AMQPS) 服務。
我已經設置了隊列並在雲端擁有相關信息,但我遇到的問題是找出接受該流並將其轉儲到系統日誌的最佳方法。我現在正在使用 syslog-ng,但看起來它只能發送到 RabbitMQ 而不能接受它。
我目前正在研究 Pika,看看它是否可以接受 AMQPS,然後將其傳輸到日誌文件,但這就是我遇到一些問題的地方。任何幫助將不勝感激。
謝謝
答案1
我找到了 walbit 製作的 github 頁面(https://github.com/walbit/AMP_API),這對我來說似乎工作正常。我正在使用“consume_event_stream_by_name.py”,它將所有輸入轉儲到標準輸出。我做了一點修改,將其也發送到 syslog,現在 syslog-ng 正在接受它。
目前我對這段程式碼的唯一問題是,如果連線沒有進入,它似乎會不時超時或整體死亡。
#!/usr/bin/env python
import argparse
import pika
import pprint
import requests
import sys
import logging
import logging.handlers
# YOU NEED TO CREATE AN auth.py FILE WITH CLIENT_ID AND API_KEY STRINGS
from auth import CLIENT_ID, API_KEY
parser = argparse.ArgumentParser()
parser.add_argument('event_stream_name', metavar='event_stream_name',
nargs=1, help='event stream name')
parser.parse_args()
event_stream_name = parser.parse_args().event_stream_name[0]
api_endpoint = 'https://api.amp.cisco.com/v1/event_streams'
session = requests.Session()
session.auth = (CLIENT_ID, API_KEY)
event_streams = session.get(api_endpoint).json()['data']
event_stream = {}
for e in event_streams:
if e['name'] is event_stream_name:
event_stream = e
amqp_url = 'amqps://{user_name}:{password}@{host}:{port}'.format(
**e['amqp_credentials'])
queue = e['amqp_credentials']['queue_name']
parameters = pika.URLParameters(amqp_url)
connection = pika.BlockingConnection(parameters)
channel = connection.channel()
parameters = pika.URLParameters(amqp_url)
connection = pika.BlockingConnection(parameters)
channel = connection.channel()
my_logger = logging.getLogger('MyLogger')
my_logger.setLevel(logging.INFO)
handler = logging.handlers.SysLogHandler(address = '/dev/log')
my_logger.addHandler(handler)
def callback(ch, method, properties, body):
# print(" [x] Received meth:\t%r" % method)
# print(" [x] Received prop:\t%r" % properties)
# print(" [x] Received body:\t%r" % body)
# print(body)
my_logger.info('Fireamp: ' + body)
channel.basic_consume(callback, queue, no_ack=True)
print(" [*] Connecting to:\t%r" % amqp_url)
print(" [*] Waiting for messages. To exit press CTRL+C")
channel.start_consuming()