pwntester/codeql.nvim

github github
github plugin
star 47
stars
alert-circle 0
open issues
users 2
subscribers
git-branch 4
forks
CREATED

2019-12-12

UPDATED

14 days ago

packer

require('packer').startup(function()
  use 'pwntester/codeql.nvim'
end)

paq

require "paq" { 
  'pwntester/codeql.nvim'
}

CodeQL.nvim

Neovim plugin to help writing and testing CodeQL queries.

asciicast

Features

  • Syntax highlighting for CodeQL query language
  • Query execution
  • Quick query evaluation
  • Query history
  • Result browser

Requirements

  • Neovim 0.5+

Usage

Query pack

Create qlpack.yaml (see QL packs). E.g:

name: test 
version: 0.0.1
libraryPathDependencies: [codeql-java]

Query

Create .ql file with query

Database

Use SetDatabase <path to db> to let the plugin know what DB to work with. Use UnsetDatabase to deregister the current registered database.

Run query or eval predicates

Use RunQuery or QuickEval commands or qr, qe shortcuts respectively to run the query or evaluate the predicate under the cursor.

Configuration options

g:codeql_search_path: List of codeql search paths g:codeql_max_ram: Max RAM memory to be used by CodeQL

eg:

let g:codeql_max_ram = 32000
let g:codeql_search_path = ['/Users/pwntester/codeql-home/codeql-repo', '/Users/pwntester/codeql-home/codeql-go-repo']

Commands

  • SetDatabase <path to db>: Required before running any query.
  • RunQuery: Runs the query on the current buffer. Requires a DB to be set first.
  • QuickEval: Quick evals the predicate or selection under cursor. Requires a DB to be set first.
  • History: Shows a menu to render results of previous queries (on the same nvim session).
  • StopServer: Stops the query server associated with the query buffer. A new one will be started upon query evaluation.
  • PrintAST: On a codeql:/ buffer, prints the AST of the current file.
  • LoadSarif: Loads the issues of a SARIF file. To browse the results, use SetDatabase before.

Mappings

  • gd: On a codeql:/ file, jumps to the symbol definition.
  • gr: On a codeql:/ file, show symbol references in the QuickFix window.
  • qr: Runs the current query.
  • qe: QuickEvals the selected predicate.

Result Browser

After running a query or quick evaluating a predicate, results will be rendered in a special panel.

  • o: collapses/Expands result
  • Enter (on a visitable result node): opens node file in nvim and moves cursor to window with source code file
  • p: similar to Enter but does not keep cursor on the results panel
  • n: (on a Paths node): change to next path
  • P: (on a Paths node): change to previous path
  • f: filter issues by issue label
  • F: filter issues
  • c: clear filter
  • t: open all folds
  • T: closes all folds
  • q: closes result panel

Language Server Protocol

This plugin does not provide any support for the Language Server Protocol (LSP). But in order to have the best CodeQL writing experience it is recommended to configure a LSP client to connect to the CodeQL Language Server.

There are many LSP clients in the NeoVim ecosystem. The following clients have been tested with CodeQL Language Server:

Neovim Built-In LSP

It is possible to configure the built-in LSP client without any additional plugins, but a default configuration for the CodeQL Language Server has been added to Nvim-LSP. If you are using vim-plug, it is a matter of adding following line to you vim config:

Plug 'neovim/nvim-lsp'

Using this client, it is only required to configure the client with:

local nvim_lsp = require 'nvim_lsp'

nvim_lsp.codeqlls.setup{
    on_attach = on_attach_callback;
    settings = {
        search_path = {'~/codeql-home/codeql-repo'};
    };
}

NOTE: change search_path to the path where the CodeQL repo has been installed to.

It is also recommended to add an on_attach callback to define LSP mappings. E.g:

local function on_attach_callback(client, bufnr)
    api.nvim_buf_set_keymap(bufnr, "n", "gD", "<Cmd>lua show_diagnostics_details()<CR>", { silent = true; })
    api.nvim_buf_set_keymap(bufnr, "n", "gd", "<Cmd>lua vim.lsp.buf.definition()<CR>", { silent = true; })
    api.nvim_buf_set_keymap(bufnr, "n", "gi", "<Cmd>lua vim.lsp.buf.implementation()<CR>", { silent = true; })
    api.nvim_buf_set_keymap(bufnr, "n", "gK", "<Cmd>lua vim.lsp.buf.hover()<CR>", { silent = true; })
    api.nvim_buf_set_keymap(bufnr, "n", "gh", "<Cmd>lua vim.lsp.buf.signature_help()<CR>", { silent = true; })
    api.nvim_buf_set_keymap(bufnr, "n", "gr", "<Cmd>lua vim.lsp.buf.references()<CR>", { silent = true; })
    api.nvim_buf_set_keymap(bufnr, "n", "gF", "<Cmd>lua vim.lsp.buf.formatting()<CR>", { silent = true; })
    api.nvim_command [[autocmd CursorHold  <buffer> lua vim.lsp.buf.document_highlight()]]
    api.nvim_command [[autocmd CursorHoldI <buffer> lua vim.lsp.buf.document_highlight()]]
    api.nvim_command [[autocmd CursorMoved <buffer> lua vim.lsp.util.buf_clear_references()]] 
end

local nvim_lsp = require 'nvim_lsp'

nvim_lsp.codeqlls.setup{
    on_attach = on_attach_callback;
    settings = {
        search_path = {'~/codeql-home/codeql-repo'};
    };
}

Check my dotfiles for examples on how to configure the NVIM LSP client.

Coc.nvim

It is possible to add codeql language server to coc.nvim using coc-settings.json as an executable language server

{
    "languageserver": {
        "codeql" : {
            "command": "<path to codeql binary>",
            "args": [
                "execute",
                "language-server",
                "--check-errors=ON_CHANGE",
                "--search-path=./:<path to semmle/ql repo maybe>:<any more paths>",
                "-q"
            ],
            "filetypes": ["ql"],
            "rootPatterns": ["qlpack.yml"],
            "requireRootPattern": true
        }
    }
}