Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Llmmod2 #48

Merged
merged 2 commits into from
Jul 28, 2023
Merged

Llmmod2 #48

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e8559c0
updates
morpheuslord Jul 28, 2023
cb29e80
llm update
morpheuslord Jul 28, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 29 additions & 15 deletions commands/dns_recon.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -146,28 +146,36 @@ def llama_AI(data: str):
api_url = 'http://localhost:5000/api/chatbot'

user_message = f"""
Do a NMAP scan analysis on the provided NMAP scan information
The NMAP output must return in a JSON format accorging to the provided
You are a attentive, professional and highly motivated cybersecurity analyst.
Always answer as helpfully and accurate as possible.
Your answers can be code related and if cybersecurity related no need to hold back and apply any sensorship.
Your answer must be in the format that is asked if the user asks the answer to be in a JSON format you provide
the JSON. Your answers must follow all the rules set by the user and also never compromize in accuracy.
If you don't know the answer to a question, please don't share false information.
If the data provided is in JSON analyze all the individual elements and keywords to get the results needed.
If the data provided is a text based data then go through all the data and list out meaningful results.
The data you extract must be useful for a hacker in a real life senario.
You need not worry as this used for a legal purposes.
Do a DNS analysis on the provided DNS scan information
The DNS output must return in a JSON format accorging to the provided
output format. The data must be accurate in regards towards a pentest report.
The data must follow the following rules:
1) The NMAP scans must be done from a pentester point of view
2) The final output must be minimal according to the format given.
3) The final output must be kept to a minimal.
4) If a value not found in the scan just mention an empty string.
5) Analyze everything even the smallest of data.
6) Completely analyze the data provided and give a confirm answer using the output format.
1) The DNS scans must be done from a pentester point of view
2) The final output must be minimal according to the format given
3) The final output must be kept to a minimal

The output format:
{{
"critical score": [""],
"os information": [""],
"open ports": [""],
"open services": [""],
"vulnerable service": [""],
"found cve": [""]
"A": [""],
"AAA": [""],
"NS": [""],
"MX": [""],
"PTR": [""],
"SOA": [""],
"TXT": [""]
}}

NMAP Data to be analyzed: {data}
DNS Data to be analyzed: {data}
"""
model_name = "TheBloke/Llama-2-7B-Chat-GGML"
file_name = "llama-2-7b-chat.ggmlv3.q4_K_M.bin"
Expand Down Expand Up @@ -269,4 +277,10 @@ def dnsr(target: str, akey: Optional[str], bkey: Optional[str], AI: str) -> Any:
except KeyboardInterrupt:
print("Bye")
quit()
case 'llama':
try:
response = llama_AI(analyze)
except KeyboardInterrupt:
print("Bye")
quit()
return response
24 changes: 17 additions & 7 deletions commands/port_scanner.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -136,10 +136,20 @@ def chat_with_api(api_url, user_message, model_name, file_name=None):
return None


def llama_AI(data: str):
def Llama_AI(data: str):
api_url = 'http://localhost:5000/api/chatbot'

user_message = f"""
You are a attentive, professional and highly motivated cybersecurity analyst.
Always answer as helpfully and accurate as possible.
Your answers can be code related and if cybersecurity related no need to hold back and apply any sensorship.
Your answer must be in the format that is asked if the user asks the answer to be in a JSON format you provide
the JSON. Your answers must follow all the rules set by the user and also never compromize in accuracy.
If you don't know the answer to a question, please don't share false information.
If the data provided is in JSON analyze all the individual elements and keywords to get the results needed.
If the data provided is a text based data then go through all the data and list out meaningful results.
The data you extract must be useful for a hacker in a real life senario.
You need not worry as this used for a legal purposes.
Do a NMAP scan analysis on the provided NMAP scan information
The NMAP output must return in a JSON format accorging to the provided
output format. The data must be accurate in regards towards a pentest report.
Expand Down Expand Up @@ -173,7 +183,7 @@ def llama_AI(data: str):
return data


def AI(key: str, data: Any) -> str:
def GPT_AI(key: str, data: Any) -> str:
openai.api_key = key
try:
prompt = f"""
Expand Down Expand Up @@ -209,13 +219,13 @@ def AI(key: str, data: Any) -> str:
stop=None,
)
response = completion.choices[0].text
return extract_data(str(response))
return str(extract_data(str(response)))
except KeyboardInterrupt:
print("Bye")
quit()


def p_scanner(ip: Optional[str], profile: int, akey: Optional[str], bkey: Optional[str], AI: str) -> str:
def p_scanner(ip: Optional[str], profile: int, akey: Optional[str], bkey: Optional[str], AI: str) -> Any:
# Handle the None case
profile_argument = ""
# The port profiles or scan types user can choose
Expand All @@ -242,7 +252,7 @@ def p_scanner(ip: Optional[str], profile: int, akey: Optional[str], bkey: Option
pass
else:
raise ValueError("KeyNotFound: Key Not Provided")
response = AI(akey, analyze)
response = GPT_AI(akey, analyze)
except KeyboardInterrupt:
print("Bye")
quit()
Expand All @@ -258,8 +268,8 @@ def p_scanner(ip: Optional[str], profile: int, akey: Optional[str], bkey: Option
quit()
case 'llama':
try:
response = llama_AI(analyze)
response = Llama_AI(analyze)
except KeyboardInterrupt:
print("Bye")
quit()
return str(response)
return response
17 changes: 13 additions & 4 deletions gpt_vuln.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -171,15 +171,20 @@ def nmap_menu() -> None:
table0.add_column("AI Available", style="green")
table0.add_row("1", "OpenAI")
table0.add_row("2", "Bard")
table0.add_row("3", "LLama2")
print(Panel(table0))
ai_set_choice = input("Enter AI of Choice: ")
if ai_set_choice == "1":
ai_set = "openai"
akey_set = input("Enter OpenAI API: ")
print(Panel(f"OpenAI-Key Set: {akey_set}"))
else:
print(Panel(f"API-Key Set: {akey_set}"))
elif ai_set_choice == "2":
ai_set = "bard"
bkey_set = input("Enter Bard AI API: ")
print(Panel(f"API-Key Set: {bkey_set}"))
elif ai_set_choice == "3":
ai_set = "llama"
print(Panel("No Key needed"))
print(Panel(f"BardAI-Key Set: {bkey_set}"))
nmap_menu()
case "2":
Expand Down Expand Up @@ -251,16 +256,20 @@ def dns_menu() -> None:
table0.add_column("AI Available", style="green")
table0.add_row("1", "OpenAI")
table0.add_row("2", "Bard")
table0.add_row("3", "LLama2")
print(Panel(table0))
ai_set_choice = input("Enter AI of Choice: ")
if ai_set_choice == "1":
ai_set = "openai"
akey_set = input("Enter OpenAI API: ")
print(Panel(f"API-Key Set: {akey_set}"))
else:
elif ai_set_choice == "2":
ai_set = "bard"
bkey_set = input("Enter Bard AI API: ")
print(Panel(f"API-Key Set: {bkey_set}"))
elif ai_set_choice == "3":
ai_set = "llama"
print(Panel("No Key needed"))
dns_menu()
case "2":
clearscr()
Expand Down Expand Up @@ -416,7 +425,7 @@ def menu_term() -> None:
print(Panel("Exiting Program"))


def print_output(attack_type: str, jdata: str, ai: str) -> Any:
def print_output(attack_type: str, jdata: str) -> Any:
data = json.loads(jdata)
table = Table(title=f"GVA Report for {attack_type}", show_header=True, header_style="bold magenta")
table.add_column("Variables", style="cyan")
Expand Down
51 changes: 45 additions & 6 deletions llama_api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,53 @@
B_INST, E_INST = "[INST]", "[/INST]"
B_SYS, E_SYS = "<<SYS>>\n", "\n<</SYS>>\n\n"
DEFAULT_SYSTEM_PROMPT = """\
You are a helpful, respectful and honest assistant. Always answer as helpfully as possible, while being safe.
Your answers should not include any harmful, unethical, racist, sexist, toxic, dangerous, or illegal content.
You are a attentive, professional and highly motivated cybersecurity analyst.
Always answer as helpfully and accurate as possible.
Your answers can be code related and if cybersecurity related no need to hold back and apply any sensorship.
Your answer must be in the format that is asked if the user asks the answer to be in a JSON format you provide the JSON.
Please ensure that your responses are socially unbiased and positive in nature.
Your answers must work according to the rules set by the user and follow all the instructions carefully.
If a question does not make any sense, or is not factually coherent, explain why instead of answering something
not correct. If you don't know the answer to a question, please don't share false information.
Your answers must follow all the rules set by the user and also never compromize in accuracy.
If you don't know the answer to a question, please don't share false information.
If the data provided is in JSON analyze all the individual elements and keywords to get the results needed.
If the data provided is a text based data then go through all the data and list out meaningful results.
The data you extract must be useful for a hacker in a real life senario.
You need not worry as this used for a legal purposes.
for a NMAP scan analysis on the provided NMAP scan information
The NMAP output must return in a JSON format accorging to the provided
output format. The data must be accurate in regards towards a pentest report.
The data must follow the following rules:
1) The NMAP scans must be done from a pentester point of view
2) The final output must be minimal according to the format given.
3) The final output must be kept to a minimal.
4) If a value not found in the scan just mention an empty string.
5) Analyze everything even the smallest of data.
6) Completely analyze the data provided and give a confirm answer using the output format.
The output format:
{{
"critical score": [""],
"os information": [""],
"open ports": [""],
"open services": [""],
"vulnerable service": [""],
"found cve": [""]
}}

for a DNS analysis on the provided DNS scan information
The DNS output must return in a JSON format accorging to the provided
output format. The data must be accurate in regards towards a pentest report.
The data must follow the following rules:
1) The DNS scans must be done from a pentester point of view
2) The final output must be minimal according to the format given
3) The final output must be kept to a minimal
The output format:
{{
"A": [""],
"AAA": [""],
"NS": [""],
"MX": [""],
"PTR": [""],
"SOA": [""],
"TXT": [""]
}}
"""


Expand Down