#!/bin/bash -eu
# Copyright 2023 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
##########################################################################
# Retrieve JDK-11
wget https://download.java.net/openjdk/jdk11.0.0.1/ri/openjdk-11.0.0.1_linux-x64_bin.tar.gz -O openjdk-11.tar.gz
tar -zxf openjdk-11.tar.gz
rm -f openjdk-11.tar.gz
cp -r jdk-11.0.0.1 $OUT/
JAVA_HOME=$OUT/jdk-11.0.0.1
PATH=$JAVA_HOME/bin:$PATH

# Add task for copy dependency jars
echo "
task copyToLib(type: Copy) {
    into \"\${buildDir}/dependencies\"
    from configurations.runtimeClasspath
}" >> ./build.gradle

# Gradle build with gradle wrapper
rm -rf $HOME/.gradle/caches/
./gradlew clean build shadowJar copyToLib -x test -x javadoc -x sources
./gradlew --stop

cp "./build/libs/$(basename ./build/tmp/jar/*.jar)" $OUT/graphql-java.jar

ALL_JARS="graphql-java.jar"

# Copy dependency jars
for JARFILE in $(ls ./build/dependencies/*.jar)
do
  cp $JARFILE $OUT/
  ALL_JARS=$ALL_JARS" $(basename $JARFILE)"
done

# The classpath at build-time includes the project jars in $OUT as well as the
# Jazzer API.
BUILD_CLASSPATH=$(echo $ALL_JARS | xargs printf -- "$OUT/%s:"):$JAZZER_API_PATH

# All .jar and .class files lie in the same directory as the fuzzer at runtime.
RUNTIME_CLASSPATH=$(echo $ALL_JARS | xargs printf -- "\$this_dir/%s:"):\$this_dir

for fuzzer in $(find $SRC -name '*Fuzzer.java')
do
  fuzzer_basename=$(basename -s .java $fuzzer)
  javac -cp $BUILD_CLASSPATH $fuzzer
  cp $SRC/$fuzzer_basename.class $OUT/

  # Create an execution wrapper that executes Jazzer with the correct arguments.
  echo "#!/bin/bash

  # LLVMFuzzerTestOneInput for fuzzer detection.
  this_dir=\$(dirname "\$0")
  if [[ "\$@" =~ (^| )-runs=[0-9]+($| ) ]]
  then
    mem_settings='-Xmx1900m:-Xss900k'
  else
    mem_settings='-Xmx2048m:-Xss1024k'
  fi

  LD_LIBRARY_PATH="$JVM_LD_LIBRARY_PATH":\$this_dir \
    \$this_dir/jazzer_driver                        \
    --agent_path=\$this_dir/jazzer_agent_deploy.jar \
    --cp=$RUNTIME_CLASSPATH                         \
    --target_class=$fuzzer_basename                 \
    --jvm_args="\$mem_settings"                     \
    \$@" > $OUT/$fuzzer_basename

    chmod u+x $OUT/$fuzzer_basename
done