Skip to content

Commit

Permalink
feat: Add QL Queries and update workflow
Browse files Browse the repository at this point in the history
  • Loading branch information
GeekMasher committed Jun 28, 2024
1 parent ec0ea08 commit 3768349
Show file tree
Hide file tree
Showing 4 changed files with 65 additions and 49 deletions.
34 changes: 34 additions & 0 deletions .github/codeql/python/Prompt.ql
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
/**
* @name AI Prompt Usage
* @description AI Prompt Usage
* @kind problem
* @problem.severity error
* @security-severity 2.0
* @sub-severity medium
* @precision medium
* @id githubsecuritylab/prompt-usage
* @tags security
*/

import python
private import semmle.python.ApiGraphs
private import semmle.python.Concepts
private import semmle.python.dataflow.new.DataFlow

class AzureOpenAIClient extends DataFlow::Node {
AzureOpenAIClient() {
this =
API::moduleImport("openai")
.getMember(["BaseAzureClient", "AzureOpenAI", "AsyncAzureOpenAI"])
.getASubclass*()
.getReturn()
.getMember("chat")
.getMember("completions")
.getMember("with_raw_response")
.getMember("create")
.getACall()
}
}

from AzureOpenAIClient sinks
select sinks, "AI prompt used"
24 changes: 24 additions & 0 deletions .github/codeql/python/codeql-pack.lock.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
---
lockVersion: 1.0.0
dependencies:
codeql/dataflow:
version: 1.0.2
codeql/mad:
version: 1.0.2
codeql/python-all:
version: 1.0.2
codeql/regex:
version: 1.0.2
codeql/ssa:
version: 1.0.2
codeql/tutorial:
version: 1.0.2
codeql/typetracking:
version: 1.0.2
codeql/util:
version: 1.0.2
codeql/xml:
version: 1.0.2
codeql/yaml:
version: 1.0.2
compiled: false
5 changes: 5 additions & 0 deletions .github/codeql/python/qlpack.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
library: false
name: octodemo/sample
version: 0.0.1
dependencies:
codeql/python-all: '*'
51 changes: 2 additions & 49 deletions .github/workflows/codeql.yml
Original file line number Diff line number Diff line change
@@ -1,14 +1,3 @@
# For most projects, this workflow file will not need changing; you simply need
# to commit it to your repository.
#
# You may wish to alter this file to override the set of languages analyzed,
# or to provide custom queries or build logic.
#
# ******** NOTE ********
# We have attempted to detect the languages in your repository. Please check
# the `language` matrix defined below to confirm you have the correct set of
# supported CodeQL languages.
#
name: "CodeQL"

on:
Expand All @@ -22,13 +11,7 @@ on:
jobs:
analyze:
name: Analyze (${{ matrix.language }})
# Runner size impacts CodeQL analysis time. To learn more, please see:
# - https://gh.io/recommended-hardware-resources-for-running-codeql
# - https://gh.io/supported-runners-and-hardware-resources
# - https://gh.io/using-larger-runners (GitHub.com only)
# Consider using larger runners or machines with greater resources for possible analysis time improvements.
runs-on: ${{ (matrix.language == 'swift' && 'macos-latest') || 'ubuntu-latest' }}
timeout-minutes: ${{ (matrix.language == 'swift' && 120) || 360 }}
runs-on: 'ubuntu-latest'
permissions:
# required for all workflows
security-events: write
Expand All @@ -48,46 +31,16 @@ jobs:
build-mode: none
- language: python
build-mode: none
# CodeQL supports the following values keywords for 'language': 'c-cpp', 'csharp', 'go', 'java-kotlin', 'javascript-typescript', 'python', 'ruby', 'swift'
# Use `c-cpp` to analyze code written in C, C++ or both
# Use 'java-kotlin' to analyze code written in Java, Kotlin or both
# Use 'javascript-typescript' to analyze code written in JavaScript, TypeScript or both
# To learn more about changing the languages that are analyzed or customizing the build mode for your analysis,
# see https://docs.github.com/en/code-security/code-scanning/creating-an-advanced-setup-for-code-scanning/customizing-your-advanced-setup-for-code-scanning.
# If you are analyzing a compiled language, you can modify the 'build-mode' for that language to customize how
# your codebase is analyzed, see https://docs.github.com/en/code-security/code-scanning/creating-an-advanced-setup-for-code-scanning/codeql-code-scanning-for-compiled-languages
steps:
- name: Checkout repository
uses: actions/checkout@v4

# Initializes the CodeQL tools for scanning.
- name: Initialize CodeQL
uses: github/codeql-action/init@v3
with:
languages: ${{ matrix.language }}
build-mode: ${{ matrix.build-mode }}
# If you wish to specify custom queries, you can do so here or in a config file.
# By default, queries listed here will override any specified in a config file.
# Prefix the list here with "+" to use these queries and those in the config file.

# For more details on CodeQL's query packs, refer to: https://docs.github.com/en/code-security/code-scanning/automatically-scanning-your-code-for-vulnerabilities-and-errors/configuring-code-scanning#using-queries-in-ql-packs
# queries: security-extended,security-and-quality

# If the analyze step fails for one of the languages you are analyzing with
# "We were unable to automatically build your code", modify the matrix above
# to set the build mode to "manual" for that language. Then modify this step
# to build your code.
# ℹ️ Command-line programs to run using the OS shell.
# 📚 See https://docs.github.com/en/actions/using-workflows/workflow-syntax-for-github-actions#jobsjob_idstepsrun
- if: matrix.build-mode == 'manual'
shell: bash
run: |
echo 'If you are using a "manual" build mode for one or more of the' \
'languages you are analyzing, replace this with the commands to build' \
'your code, for example:'
echo ' make bootstrap'
echo ' make release'
exit 1
packs: ./.github/codeql

- name: Perform CodeQL Analysis
uses: github/codeql-action/analyze@v3
Expand Down

0 comments on commit 3768349

Please sign in to comment.