Question

我使用python编写了我的脚本，它运行成功。我在终端中获取，识别并打印了addr1, addr2,和addr3。但是，在尝试将这些结果附加到字典格式的CSV文件时，我遇到了几个错误。

这是我之前在终端打印结果时的脚本：

#!/usr/bin/env python3
from scapy.all import *
import os

devices = set()
interface = "wlan0mon"
iface = "wlan0mon"

def sniffmgmnt(p):
    if p.haslayer(Dot11):
            devices.add(p.addr1)
            devices.add(p.addr2)
            devices.add(p.addr3)
            print(devices)

while True:
    for channel in range(1, 14):
            os.system("iwconfig" + iface + "channel" + str(channel))
            sniff(iface=interface, prn=sniffmgmnt, count=3, timeout=3, store=0)

然后我尝试将这些结果附加到字典格式的CSV文件中，其中addr3是键：

#!/usr/bin/env python3
from scapy.all import *
from snifferCSV import CSV
import os
aps = dict()
devices = set()

headers = ['addr3', 'addr2', 'addr1']
csvFile = CSV('data.csv', headers)

interface = "wlan0mon"
iface = "wlan0mon"

def sniffmgmnt(p):
    if p.haslayer(Dot11):
            source.append(p.addr1);
            receiving.append(p.addr2);
            destination.append(p.addr3);
            csvFile.addToCSV( [addr3, addr2, addr1] )
while True:
    for channel in range(1, 14):
            os.system("iwconfig" + iface + "channel" + str(channel))
            sniff(iface=interface, prn=sniffmgmnt, count=3, timeout=3, store=0)

但是脚本只输出终端中的标题，我的CSV文件仍为空。

下面我已经包含了我的CSV类 snifferCSV.py ：

      class CSV(object):
        csvFile = None
        fileName = data.csv
        def __init__(self,fileName,headers):
            self.fileName = fileName
            self.prepareCSVFile(headers)
    def prepareCSVFile(self,headers):
            self.csvFile = open(self.fileName, 'w')
            headerString = self.prepareValues(headers)
            self.writeToCSV(headerString)
    def prepareValues(self,valueList):
            line = str()
            for item in valueList[:-1]:
                    line = line + str(item) + ','
            line += valueList[-1]
            return line
    def addToCSV(self,valueList):
            valueLine = self.prepareValues(valueList)
            self.writeToCSV(valueLine)
    def writeToCSV(self,line):
            print(line)
            self.csvFile.write(line)

我的目标是将我的结果以字典格式附加到CSV文件中，其中addr3是关键。但是，我对这个过程的逻辑是错误的。有人能指出我正确的方向吗？

提前致谢 -JJ

Answer 1

我拿了@coder的建议并改了 open(self.fileName, 'w')到open(self.fileName, 'a')但是只有我的标头附加到csv文件。

我最后抓了我的snifferCSV.py类，而是在我的主文件中嵌入了字典函数，在那里我使用了pairwise dict构造函数和zip函数将我的数据附加到csv文件中：

#!/usr/bin/env python3
from scapy.all import *
import csv
import os

interface = "wlan0mon"
iface = "wlan0mon"

def sniffmgmnt(p):
    if p.haslayer(Dot11):
            keys = [(p.addr3)]
            values = [(p.addr1, p.addr2)]
            my_dict = dict(zip(keys, values))
            with open('data.csv','a') as f:
                    w = csv.writer(f)
                    w.writerows(my_dict.items())
 while True:
    for channel in range(1, 14):
            os.system("iwconfig " + iface + " channel " + str(channel))
            sniff(iface=interface, prn=sniffmgmnt)

以字典格式Python附加Scapy结果

1 个答案: