mirror of
https://github.com/mirror/processhacker
synced 2026-06-08 16:03:24 +00:00
wj32
cbf214e775
* changed elevation dialog icon to Warning git-svn-id: svn://svn.code.sf.net/p/processhacker/code@1077 21ef857c-d57f-4fe0-8362-d861dc6d29cd
319 lines
11 KiB
C#
319 lines
11 KiB
C#
/*
|
|
* Process Hacker -
|
|
* network list
|
|
*
|
|
* Copyright (C) 2008-2009 wj32
|
|
*
|
|
* This file is part of Process Hacker.
|
|
*
|
|
* Process Hacker is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Process Hacker is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Process Hacker. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
using System;
|
|
using System.Diagnostics;
|
|
using System.Reflection;
|
|
using System.Windows.Forms;
|
|
using System.Drawing;
|
|
using System.Collections.Generic;
|
|
using ProcessHacker.UI;
|
|
|
|
namespace ProcessHacker.Components
|
|
{
|
|
public partial class NetworkList : UserControl
|
|
{
|
|
private NetworkProvider _provider;
|
|
private HighlightingContext _highlightingContext;
|
|
public new event KeyEventHandler KeyDown;
|
|
public new event MouseEventHandler MouseDown;
|
|
public new event MouseEventHandler MouseUp;
|
|
public new event EventHandler DoubleClick;
|
|
public event EventHandler SelectedIndexChanged;
|
|
|
|
public NetworkList()
|
|
{
|
|
InitializeComponent();
|
|
|
|
_highlightingContext = new HighlightingContext(listNetwork);
|
|
listNetwork.SetTheme("explorer");
|
|
listNetwork.ListViewItemSorter = new SortedListComparer(listNetwork);
|
|
listNetwork.KeyDown += new KeyEventHandler(NetworkList_KeyDown);
|
|
listNetwork.MouseDown += new MouseEventHandler(listNetwork_MouseDown);
|
|
listNetwork.MouseUp += new MouseEventHandler(listNetwork_MouseUp);
|
|
listNetwork.DoubleClick += new EventHandler(listNetwork_DoubleClick);
|
|
listNetwork.SelectedIndexChanged += new System.EventHandler(listNetwork_SelectedIndexChanged);
|
|
}
|
|
|
|
private void listNetwork_DoubleClick(object sender, EventArgs e)
|
|
{
|
|
if (this.DoubleClick != null)
|
|
this.DoubleClick(sender, e);
|
|
}
|
|
|
|
private void listNetwork_MouseUp(object sender, MouseEventArgs e)
|
|
{
|
|
if (this.MouseUp != null)
|
|
this.MouseUp(sender, e);
|
|
}
|
|
|
|
private void listNetwork_MouseDown(object sender, MouseEventArgs e)
|
|
{
|
|
if (this.MouseDown != null)
|
|
this.MouseDown(sender, e);
|
|
}
|
|
|
|
private void listNetwork_SelectedIndexChanged(object sender, System.EventArgs e)
|
|
{
|
|
if (this.SelectedIndexChanged != null)
|
|
this.SelectedIndexChanged(sender, e);
|
|
}
|
|
|
|
private void NetworkList_KeyDown(object sender, KeyEventArgs e)
|
|
{
|
|
if (this.KeyDown != null)
|
|
this.KeyDown(sender, e);
|
|
}
|
|
|
|
#region Properties
|
|
|
|
public new bool DoubleBuffered
|
|
{
|
|
get
|
|
{
|
|
return (bool)typeof(ListView).GetProperty("DoubleBuffered",
|
|
BindingFlags.NonPublic | BindingFlags.Instance).GetValue(listNetwork, null);
|
|
}
|
|
set
|
|
{
|
|
typeof(ListView).GetProperty("DoubleBuffered",
|
|
BindingFlags.NonPublic | BindingFlags.Instance).SetValue(listNetwork, value, null);
|
|
}
|
|
}
|
|
|
|
public override bool Focused
|
|
{
|
|
get
|
|
{
|
|
return listNetwork.Focused;
|
|
}
|
|
}
|
|
|
|
public override ContextMenu ContextMenu
|
|
{
|
|
get { return listNetwork.ContextMenu; }
|
|
set { listNetwork.ContextMenu = value; }
|
|
}
|
|
|
|
public override ContextMenuStrip ContextMenuStrip
|
|
{
|
|
get { return listNetwork.ContextMenuStrip; }
|
|
set { listNetwork.ContextMenuStrip = value; }
|
|
}
|
|
|
|
public ListView List
|
|
{
|
|
get { return listNetwork; }
|
|
}
|
|
|
|
public NetworkProvider Provider
|
|
{
|
|
get { return _provider; }
|
|
set
|
|
{
|
|
if (_provider != null)
|
|
{
|
|
_provider.DictionaryAdded -= new NetworkProvider.ProviderDictionaryAdded(provider_DictionaryAdded);
|
|
_provider.DictionaryModified -= new NetworkProvider.ProviderDictionaryModified(provider_DictionaryModified);
|
|
_provider.DictionaryRemoved -= new NetworkProvider.ProviderDictionaryRemoved(provider_DictionaryRemoved);
|
|
_provider.Updated -= new NetworkProvider.ProviderUpdateOnce(provider_Updated);
|
|
}
|
|
|
|
_provider = value;
|
|
|
|
listNetwork.Items.Clear();
|
|
|
|
if (_provider != null)
|
|
{
|
|
foreach (Win32.NetworkConnection item in _provider.Dictionary.Values)
|
|
{
|
|
provider_DictionaryAdded(item);
|
|
}
|
|
|
|
_provider.UseInvoke = true;
|
|
_provider.Invoke = new NetworkProvider.ProviderInvokeMethod(this.BeginInvoke);
|
|
_provider.DictionaryAdded += new NetworkProvider.ProviderDictionaryAdded(provider_DictionaryAdded);
|
|
_provider.DictionaryModified += new NetworkProvider.ProviderDictionaryModified(provider_DictionaryModified);
|
|
_provider.DictionaryRemoved += new NetworkProvider.ProviderDictionaryRemoved(provider_DictionaryRemoved);
|
|
_provider.Updated += new NetworkProvider.ProviderUpdateOnce(provider_Updated);
|
|
}
|
|
}
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Interfacing
|
|
|
|
public void BeginUpdate()
|
|
{
|
|
listNetwork.BeginUpdate();
|
|
}
|
|
|
|
public void EndUpdate()
|
|
{
|
|
listNetwork.EndUpdate();
|
|
}
|
|
|
|
public ListView.ListViewItemCollection Items
|
|
{
|
|
get { return listNetwork.Items; }
|
|
}
|
|
|
|
public ListView.SelectedListViewItemCollection SelectedItems
|
|
{
|
|
get { return listNetwork.SelectedItems; }
|
|
}
|
|
|
|
#endregion
|
|
|
|
private void ResetImageKeys()
|
|
{
|
|
foreach (ListViewItem lvItem in listNetwork.Items)
|
|
{
|
|
string t = lvItem.ImageKey;
|
|
|
|
lvItem.ImageKey = "";
|
|
lvItem.ImageKey = t;
|
|
}
|
|
}
|
|
|
|
private void provider_Updated()
|
|
{
|
|
_highlightingContext.Tick();
|
|
}
|
|
|
|
private void provider_DictionaryAdded(Win32.NetworkConnection item)
|
|
{
|
|
HighlightedListViewItem litem = new HighlightedListViewItem(_highlightingContext);
|
|
|
|
litem.Name = item.ID;
|
|
litem.Tag = item.PID;
|
|
|
|
if (Program.ProcessProvider.Dictionary.ContainsKey(item.PID))
|
|
{
|
|
if (imageList.Images.ContainsKey(item.PID.ToString()))
|
|
imageList.Images.RemoveByKey(item.PID.ToString());
|
|
|
|
var icon = Program.ProcessProvider.Dictionary[item.PID].Icon;
|
|
|
|
if (icon != null)
|
|
{
|
|
imageList.Images.Add(item.PID.ToString(), icon);
|
|
litem.ImageKey = item.PID.ToString();
|
|
}
|
|
else
|
|
{
|
|
litem.ImageKey = "generic_process";
|
|
}
|
|
}
|
|
|
|
if (Program.ProcessProvider.Dictionary.ContainsKey(item.PID))
|
|
litem.Text = Program.ProcessProvider.Dictionary[item.PID].Name;
|
|
else
|
|
litem.Text = "(" + item.PID.ToString() + ")";
|
|
|
|
if (item.Local != null && item.Local.ToString() != "0.0.0.0:0")
|
|
litem.SubItems.Add(new ListViewItem.ListViewSubItem(litem, item.Local.ToString()));
|
|
else
|
|
litem.SubItems.Add(new ListViewItem.ListViewSubItem(litem, ""));
|
|
|
|
if (item.Remote != null && item.Remote.ToString() != "0.0.0.0:0")
|
|
litem.SubItems.Add(new ListViewItem.ListViewSubItem(litem, item.Remote.ToString()));
|
|
else
|
|
litem.SubItems.Add(new ListViewItem.ListViewSubItem(litem, ""));
|
|
|
|
litem.SubItems.Add(new ListViewItem.ListViewSubItem(litem, item.Protocol.ToString().ToUpper()));
|
|
litem.SubItems.Add(new ListViewItem.ListViewSubItem(litem, item.State != 0 ? item.State.ToString() : ""));
|
|
|
|
listNetwork.Items.Add(litem);
|
|
this.ResetImageKeys();
|
|
}
|
|
|
|
private void provider_DictionaryModified(Win32.NetworkConnection oldItem, Win32.NetworkConnection newItem)
|
|
{
|
|
lock (listNetwork)
|
|
{
|
|
ListViewItem litem = listNetwork.Items[newItem.ID];
|
|
|
|
if (litem == null)
|
|
return;
|
|
|
|
if (newItem.Local != null && newItem.Local.ToString() != "0.0.0.0:0")
|
|
{
|
|
if (newItem.LocalString != null)
|
|
litem.SubItems[1].Text = newItem.LocalString + ":" + newItem.Local.Port.ToString() +
|
|
" (" + newItem.Local.ToString() + ")";
|
|
else
|
|
litem.SubItems[1].Text = newItem.Local.ToString();
|
|
}
|
|
|
|
if (newItem.Remote != null && newItem.Remote.ToString() != "0.0.0.0:0")
|
|
{
|
|
if (newItem.RemoteString != null)
|
|
litem.SubItems[2].Text = newItem.RemoteString + ":" + newItem.Remote.Port.ToString() +
|
|
" (" + newItem.Remote.ToString() + ")";
|
|
else
|
|
litem.SubItems[2].Text = newItem.Remote.ToString();
|
|
}
|
|
|
|
litem.SubItems[4].Text = newItem.State != 0 ? newItem.State.ToString() : "";
|
|
listNetwork.Sort();
|
|
}
|
|
}
|
|
|
|
private void provider_DictionaryRemoved(Win32.NetworkConnection item)
|
|
{
|
|
int index = listNetwork.Items[item.ID].Index;
|
|
bool selected = listNetwork.Items[item.ID].Selected;
|
|
int selectedCount = listNetwork.SelectedItems.Count;
|
|
ListViewItem litem = listNetwork.Items[item.ID];
|
|
bool imageStillUsed = false;
|
|
|
|
if (litem.ImageKey == "generic_process")
|
|
{
|
|
imageStillUsed = true;
|
|
}
|
|
else
|
|
{
|
|
foreach (ListViewItem lvItem in listNetwork.Items)
|
|
{
|
|
if (lvItem != litem && lvItem.ImageKey == item.PID.ToString())
|
|
{
|
|
imageStillUsed = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!imageStillUsed)
|
|
{
|
|
imageList.Images.RemoveByKey(item.PID.ToString());
|
|
|
|
// Reset all the image keys (by now most items' icons have screwed up).
|
|
this.ResetImageKeys();
|
|
}
|
|
|
|
litem.Remove();
|
|
}
|
|
}
|
|
}
|