/* TEMPLATE GENERATED TESTCASE FILE Filename: CWE191_Integer_Underflow__short_rand_postdec_42.java Label Definition File: CWE191_Integer_Underflow.label.xml Template File: sources-sinks-42.tmpl.java */ /* * @description * CWE: 191 Integer Underflow * BadSource: rand Set data to result of rand() * GoodSource: A hardcoded non-zero, non-min, non-max, even number * Sinks: decrement * GoodSink: Ensure there will not be an underflow before decrementing data * BadSink : Decrement data, which can cause an Underflow * Flow Variant: 42 Data flow: data returned from one method to another in the same class * * */ package testcases.CWE191_Integer_Underflow.s04; import testcasesupport.*; public class CWE191_Integer_Underflow__short_rand_postdec_42 extends AbstractTestCase { private short badSource() throws Throwable { short data; /* POTENTIAL FLAW: Use a random value */ data = (short)((new java.security.SecureRandom()).nextInt(1+Short.MAX_VALUE-Short.MIN_VALUE)+Short.MIN_VALUE); return data; } public void bad() throws Throwable { short data = badSource(); /* POTENTIAL FLAW: if data == Short.MIN_VALUE, this will overflow */ data--; short result = (short)(data); IO.writeLine("result: " + result); } /* goodG2B() - use goodsource and badsink */ private short goodG2BSource() throws Throwable { short data; /* FIX: Use a hardcoded number that won't cause underflow, overflow, divide by zero, or loss-of-precision issues */ data = 2; return data; } private void goodG2B() throws Throwable { short data = goodG2BSource(); /* POTENTIAL FLAW: if data == Short.MIN_VALUE, this will overflow */ data--; short result = (short)(data); IO.writeLine("result: " + result); } /* goodB2G() - use badsource and goodsink */ private short goodB2GSource() throws Throwable { short data; /* POTENTIAL FLAW: Use a random value */ data = (short)((new java.security.SecureRandom()).nextInt(1+Short.MAX_VALUE-Short.MIN_VALUE)+Short.MIN_VALUE); return data; } private void goodB2G() throws Throwable { short data = goodB2GSource(); /* FIX: Add a check to prevent an underflow from occurring */ if (data > Short.MIN_VALUE) { data--; short result = (short)(data); IO.writeLine("result: " + result); } else { IO.writeLine("data value is too small to decrement."); } } public void good() throws Throwable { goodG2B(); goodB2G(); } /* Below is the main(). It is only used when building this testcase on * its own for testing or for building a binary to use in testing binary * analysis tools. It is not used when compiling all the testcases as one * application, which is how source code analysis tools are tested. */ public static void main(String[] args) throws ClassNotFoundException, InstantiationException, IllegalAccessException { mainFromParent(args); } }