test_server.cpython-310.pyc•5.6 kB
o
�UhD � @ s� d Z ddlZddlZddlZddlmZ ej�deee �j
�� ddl Z e j
�dd�Ze j
�e�Zej�e� ejZejZdd� Zdd � Zd
d � Zed krRe� dS dS )
z�
Test script for Wireshark MCP Server
This script tests the core functionality of the Wireshark MCP server
without requiring a full MCP client setup.
� N)�Path�wireshark_mcp_serverzwireshark-mcp-server.pyc
� s� �t d� t d� z't d� t� } t d| j� �� | jr$t d| j� �� | jr/t d| j� �� W n tyJ } zt d|� �� W Y d}~d S d}~ww t d
� zM| �� }|d d kr�t d
|d � d�� t|d dd� �D ]\}}t d|d � d|� �� qn|d dkr�t d|d d � d�� n t d|d � �� W n ty� } z
t d|� �� W Y d}~nd}~ww t d� g d�}|D ] }t� |�}|r�d nd }t d |� d |� d |r�d!nd"� �� q�g d#�} | D ] }
t�
|
�}|r�d nd }t d |� d$|
� d |r�d!nd"� �� q�t d%� zV| �� }|d d k�rY|d �rY|d d& } d| v �r,| � d�d& } nd'} t d(| � �� | j | dd)d*�}
|
d d k�rOt d+|
d, � d-�� nt d.|
d � �� nt d/� W n t�yx } z
t d0|� �� W Y d}~nd}~ww t d1� g d2�}|D ]#}t�
|�}|�r�d nd }|�r�|nd3}t d |� d4|� d |� �� �q�t d5� t d6� t d7� t d8� t d9� t d:� d;S )<z+Test the Wireshark interface functionality.u, 🧪 Testing Wireshark MCP Server Componentsz2==================================================z01. Testing Wireshark Interface Initialization...u ✅ TShark found at: u ✅ Dumpcap found at: u ✅ Capinfos found at: u/ ❌ Error initializing Wireshark interface: NFz(
2. Testing Network Interface Listing...�status�successu
✅ Found �countz network interfaces:�
interfaces� z � z. z ... and z moreu$ ⚠️ Interface listing issue: �messageu! ❌ Error listing interfaces: z"
3. Testing Security Validation...)�eth0�wlan0z
Ethernet 2zWi-Fi�1zinvalid;command�
../etc/passwdu ✅u ❌z z Interface 'z': �Valid�Invalid)z tcp port 80zhost 192.168.1.1z
tcp; rm -rf /z
normal filterz Filter 'z2
4. Testing Live Capture (if permissions allow)...r r
z! Testing capture on interface: �
)r �timeoutu ✅ Successfully captured � packet_countz packetsuN ⚠️ Capture test failed (this is expected without proper permissions): u. ⚠️ No interfaces available for testingu? ⚠️ Capture test failed (expected without permissions): z!
5. Testing File Path Security...)z/tmp/test.pcapr z C:\Windows\System32\test.pcapz./nonexistent.pcapztest.txt�Rejectedz Path 'u"
🎉 Component testing completed!z
Next steps:z81. Install requirements: pip install -r requirements.txtz42. Configure Claude Desktop with the provided configz03. Test with Claude Desktop or other MCP clientsz44. Ensure proper network capture permissions are setT)�print�WiresharkInterface� tshark_path� dumpcap_path�
capinfos_path� Exception�get_interfaces� enumerate�SecurityValidator�validate_interface�validate_capture_filter�split�capture_packets�sanitize_filepath)� wireshark�e�interfaces_result�i� interface�test_interfaces�iface�is_validr � test_filters� filter_expr�first_interface� interface_id�capture_result�
test_paths�path� sanitized�result� r4 �test_server.py�test_wireshark_interface s� ���� ����
$
$
���
r6 c
� s� �t d� z)tj} t d� | � }t d� t d� t d� t d� t d� t d� t d � W d
S tyL } zt d |� �� t d � W Y d
}~dS d
}~w tye } zt d|� �� W Y d
}~dS d
}~ww )z,Test the full MCP server (requires FastMCP).z%
6. Testing MCP Server Integration...u- ✅ MCP Server class imported successfullyu* ✅ MCP Server initialized successfullyu 📋 Registered tools:z - get_network_interfaces()z - capture_live_packets()z - analyze_pcap_file()z! - get_protocol_statistics()z - get_capture_file_info()Tu" ⚠️ FastMCP not installed: u) 💡 Install with: pip install fastmcpNFu! ❌ Error testing MCP server: )r r �WiresharkMCPServer� ImportErrorr )r7 �serverr$ r4 r4 r5 �test_mcp_server{ s0 ����r: c
C s� t d� t d� t�� } t�| � zXz | �t� �}| �t� � |r*t d� t d� nt d� W n% ty< t d� Y n tyT } z
t d|� �� W Y d}~nd}~ww W | � � dS W | � � dS W | � � dS | � � w ) zMain test function.z&Wireshark MCP Server - Component Testsz%=====================================u"
✅ Basic component tests passed!z6The server should be ready for MCP client integration.u/
❌ Some tests failed. Check the output above.u!
⏹️ Test interrupted by useru&
❌ Unexpected error during testing: N)
r �asyncio�new_event_loop�set_event_loop�run_until_completer6 r: �KeyboardInterruptr �close)�loopr r$ r4 r4 r5 �main� s0
� ��� � �rB �__main__)�__doc__r; �json�sys�pathlibr r1 �insert�str�__file__�parent�importlib.util� importlib�util�spec_from_file_location�spec�module_from_specr �loader� exec_moduler r r6 r: rB �__name__r4 r4 r4 r5 �<module> s$ a
�