add sample module

Author: Tw1sm

CHANGELOG.md

@@ -1,4 +1,9 @@
 # Changelog
+## [v0.2.0] - 06/26/2024
+### Added
+- `sample` module to retrive table data without manual SQL query
+
+
 ## [v0.1.4] - 02/03/2024
 ### Fixed
 - Issue [#9](https://github.com/Tw1sm/PySQLRecon/issues/9)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pysqlrecon"
-version = "0.1.4"
+version = "0.2.0"
 description = "Offensive MSSQL Python toolkit"
 authors = ["Matt Creel <[email protected]>"]
 readme = "README.md"

pysqlrecon/__init__.py

@@ -1 +1 @@
-__version__ = '0.1.4'
+__version__ = '0.2.0'

pysqlrecon/modules/__init__.py

@@ -2,7 +2,7 @@
                                 links, query, rows, search, smb, tables, users, whoami, \
                                 enablexp, disablexp, xpcmd, enablerpc, disablerpc, enableole,\
                                 disableole, enableclr, disableclr, olecmd, agentstatus, agentcmd, \
-                                clr, adsi
+                                clr, adsi, sample
 
 __all__ = [
     checkrpc,
@@ -13,6 +13,7 @@
     links,
     query,
     rows,
+    sample,
     search,
     smb,
     tables,

pysqlrecon/modules/sample.py

@@ -0,0 +1,42 @@
+import typer
+
+from pysqlrecon.logger import logger
+from pysqlrecon.lib import PySqlRecon
+
+app = typer.Typer()
+COMMAND_NAME = "sample"
+HELP = "[bright_black][NORM][/] Query a sample of table data [I,L]"
+LINK_COMPATIBLE = True
+IMPERSONATE_COMPATIBLE = True
+
+
+@app.callback(invoke_without_command=True, no_args_is_help=True)
+def main(
+    ctx     : typer.Context,
+    table   : str = typer.Option(..., '--table', help='Table name'),
+    count   : int = typer.Option(5, '--count', help='Number of rows to return')):
+    
+    pysqlrecon: PySqlRecon = ctx.obj['pysqlrecon']
+    use_basic_tables = ctx.obj['basic_tables']
+
+    # verify opts are compatible with module before connecting
+    if not PySqlRecon.validate_opts(
+        LINK_COMPATIBLE,
+        IMPERSONATE_COMPATIBLE,
+        pysqlrecon.link,
+        pysqlrecon.impersonate
+    ):
+        exit()
+
+    pysqlrecon.connect()
+
+    if pysqlrecon.link is not None:
+        logger.info(f"Sampling data from '{table}' in '{pysqlrecon.db}' on {pysqlrecon.link} via {pysqlrecon.target}")
+    else:
+        logger.info(f"Sampling data from '{table}' in '{pysqlrecon.db}' on {pysqlrecon.target}")
+    
+    query = f"use {pysqlrecon.db}; SELECT TOP {count} * FROM {table};"
+    pysqlrecon.query_handler(query, use_rpc_query=True)
+    pysqlrecon.print_results(use_basic_tables)
+
+    pysqlrecon.disconnect()