76 lines
2.0 KiB
C#
76 lines
2.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.ComponentModel;
|
|
using System.Data;
|
|
using System.Drawing;
|
|
using System.Text;
|
|
using System.Windows.Forms;
|
|
using tools;
|
|
|
|
namespace SuperSQLInjection
|
|
{
|
|
public partial class AddNode : Form
|
|
{
|
|
public AddNode()
|
|
{
|
|
InitializeComponent();
|
|
}
|
|
|
|
public TreeNode tn = null;
|
|
public TreeView tvw = null;
|
|
public int type = 0;
|
|
|
|
private void btn_addNode_Click(object sender, EventArgs e)
|
|
{
|
|
if (this.txt_node_text.TextLength <= 0)
|
|
{
|
|
MessageBox.Show("请输入节点的值！");
|
|
return;
|
|
}
|
|
TreeNode ctn = new TreeNode(this.txt_node_text.Text);
|
|
if (type == 1)
|
|
{
|
|
ctn.Tag = "dbs";
|
|
|
|
if (Tools.isExistsNode(tvw.Nodes,this.txt_node_text.Text))
|
|
{
|
|
MessageBox.Show("已存在相同的节点！");
|
|
}
|
|
else {
|
|
tvw.Nodes.Add(ctn);
|
|
}
|
|
}
|
|
|
|
else {
|
|
|
|
if (tn != null)
|
|
{
|
|
if ("dbs".Equals(tn.Tag))
|
|
{
|
|
ctn.Tag = "table";
|
|
}
|
|
else if ("table".Equals(tn.Tag))
|
|
{
|
|
ctn.Tag = "column";
|
|
}
|
|
if (Tools.isExistsNode(tn.Nodes, this.txt_node_text.Text))
|
|
{
|
|
MessageBox.Show("已存在相同的节点！");
|
|
}
|
|
else
|
|
{
|
|
tn.Nodes.Add(ctn);
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
MessageBox.Show("请选择添加表或列对应的数据库或表！");
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|
|
}
|