Taint-track package expvar

Author: gagliardetto

ql/src/semmle/go/frameworks/Stdlib.qll

@@ -30,6 +30,7 @@ import semmle.go.frameworks.stdlib.EncodingXml
 import semmle.go.frameworks.stdlib.Html
 import semmle.go.frameworks.stdlib.HtmlTemplate
 import semmle.go.frameworks.stdlib.Errors
+import semmle.go.frameworks.stdlib.Expvar
 import semmle.go.frameworks.stdlib.Path
 import semmle.go.frameworks.stdlib.PathFilepath
 import semmle.go.frameworks.stdlib.Reflect

ql/src/semmle/go/frameworks/stdlib/Expvar.qll

@@ -0,0 +1,51 @@
+/**
+ * Provides classes modeling security-relevant aspects of the `expvar` package.
+ */
+
+import go
+
+/** Provides models of commonly used functions in the `expvar` package. */
+module Expvar {
+  private class MethodModels extends TaintTracking::FunctionModel, Method {
+    FunctionInput inp;
+    FunctionOutput outp;
+
+    MethodModels() {
+      // signature: func (Func).Value() interface{}
+      this.hasQualifiedName("expvar", "Func", "Value") and
+      (inp.isReceiver() and outp.isResult())
+      or
+      // signature: func (*Map).Get(key string) Var
+      this.hasQualifiedName("expvar", "Map", "Get") and
+      (inp.isReceiver() and outp.isResult())
+      or
+      // signature: func (*Map).Set(key string, av Var)
+      this.hasQualifiedName("expvar", "Map", "Set") and
+      (inp.isParameter(_) and outp.isReceiver())
+      or
+      // signature: func (*Map).String() string
+      this.hasQualifiedName("expvar", "Map", "String") and
+      (inp.isReceiver() and outp.isResult())
+      or
+      // signature: func (*String).Set(value string)
+      this.hasQualifiedName("expvar", "String", "Set") and
+      (inp.isParameter(0) and outp.isReceiver())
+      or
+      // signature: func (*String).String() string
+      this.hasQualifiedName("expvar", "String", "String") and
+      (inp.isReceiver() and outp.isResult())
+      or
+      // signature: func (*String).Value() string
+      this.hasQualifiedName("expvar", "String", "Value") and
+      (inp.isReceiver() and outp.isResult())
+      or
+      // signature: func (Var).String() string
+      this.implements("expvar", "Var", "String") and
+      (inp.isReceiver() and outp.isResult())
+    }
+
+    override predicate hasTaintFlow(FunctionInput input, FunctionOutput output) {
+      input = inp and output = outp
+    }
+  }
+}