hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2025-12-16 16:53:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Util: Refactor DenseRank implementation

Browse Source
This commit is contained in:
Tom Hvitved
2024-11-20 13:09:59 +01:00
parent 6a3e34cc4c
commit 42e0d7ce10
2 changed files with 21 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rust/ql/lib/codeql/rust/elements/internal/VariableImpl.qll
View File

@@ -378,7 +378,7 @@ module Impl {
}
}
private module DenseRankInput implements DenseRankInputSig3 {
private module DenseRankInput implements DenseRankInputSig2 {
class C1 = VariableScope;
class C2 = string;
@@ -401,7 +401,7 @@ module Impl {
* to a variable named `name` in the variable scope `scope`.
*/
private int rankVariableOrAccess(VariableScope scope, string name, VariableOrAccessCand v) {
result = DenseRank3<DenseRankInput>::denseRank(scope, name, v) - 1
v = DenseRank2<DenseRankInput>::denseRank(scope, name, result + 1)
}
/**

40
shared/util/codeql/util/DenseRank.qll
View File

@@ -58,12 +58,12 @@ module DenseRank<DenseRankInputSig Input> {
rnk = rank[result](int rnk0 | rnk0 = getRank(_) | rnk0)
}
/** Gets the dense rank of `r`. */
int denseRank(Ranked r) { result = rankRank(r, getRank(r)) }
/** Gets the `Ranked` value for which the dense rank is `rnk`. */
Ranked denseRank(int rnk) { rnk = rankRank(result, getRank(result)) }
}
/** Provides the input to `DenseRank2`. */
signature module DenseRankInputSig2 {
/** Provides the input to `DenseRank1`. */
signature module DenseRankInputSig1 {
/** A ranking context. */
bindingset[this]
class C;
@@ -77,7 +77,7 @@ signature module DenseRankInputSig2 {
}
/** Same as `DenseRank`, but allows for a context consisting of one element. */
module DenseRank2<DenseRankInputSig2 Input> {
module DenseRank1<DenseRankInputSig1 Input> {
private import Input
private int rankRank(C c, Ranked r, int rnk) {
@@ -85,17 +85,15 @@ module DenseRank2<DenseRankInputSig2 Input> {
rnk = rank[result](int rnk0 | rnk0 = getRank(c, _) | rnk0)
}
/** Gets the dense rank of `r` in the context provided by `c`. */
int denseRank(C c, Ranked r) {
exists(int rnk |
result = rankRank(c, r, rnk) and
rnk = getRank(c, r)
)
}
/**
* Gets the `Ranked` value for which the dense rank in the context provided by
* `c` is `rnk`.
*/
Ranked denseRank(C c, int rnk) { rnk = rankRank(c, result, getRank(c, result)) }
}
/** Provides the input to `DenseRank3`. */
signature module DenseRankInputSig3 {
/** Provides the input to `DenseRank2`. */
signature module DenseRankInputSig2 {
/** A ranking context. */
bindingset[this]
class C1;
@@ -113,7 +111,7 @@ signature module DenseRankInputSig3 {
}
/** Same as `DenseRank`, but allows for a context consisting of two elements. */
module DenseRank3<DenseRankInputSig3 Input> {
module DenseRank2<DenseRankInputSig2 Input> {
private import Input
private int rankRank(C1 c1, C2 c2, Ranked r, int rnk) {
@@ -121,11 +119,11 @@ module DenseRank3<DenseRankInputSig3 Input> {
rnk = rank[result](int rnk0 | rnk0 = getRank(c1, c2, _) | rnk0)
}
/** Gets the dense rank of `r` in the context provided by `c1` and `c2`. */
int denseRank(C1 c1, C2 c2, Ranked r) {
exists(int rnk |
result = rankRank(c1, c2, r, rnk) and
rnk = getRank(c1, c2, r)
)
/**
* Gets the `Ranked` value for which the dense rank in the context provided by
* `c1` and `c2` is `rnk`.
*/
Ranked denseRank(C1 c1, C2 c2, int rnk) {
rnk = rankRank(c1, c2, result, getRank(c1, c2, result))
}
}