Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 3a08cc02 authored by Steven Moreland's avatar Steven Moreland Committed by Automerger Merge Worker
Browse files

Merge "binder_parcel_fuzzer: rlimit mem" am: b5024193 

Original change: https://android-review.googlesource.com/c/platform/frameworks/native/+/1520011

MUST ONLY BE SUBMITTED BY AUTOMERGER

Change-Id: Ifc58aebccd496d7bcff450411405eb7ed8e2f63a
parents 698b09c0 b5024193

libs/binder/parcel_fuzzer/main.cpp

+24 −0
Original line number Diff line number Diff line
@@ -20,12 +20,16 @@ 
#include "hwbinder.h"

#include "util.h"



#include <iostream>



#include <android-base/logging.h>

#include <fuzzbinder/random_parcel.h>

#include <fuzzer/FuzzedDataProvider.h>



#include <cstdlib>

#include <ctime>

#include <sys/resource.h>

#include <sys/time.h>



using android::fillRandomParcel;
@@ -77,7 +81,25 @@ void doFuzz(const char* backend, const std::vector<ParcelRead<P>>& reads, 
    }

}



size_t getHardMemoryLimit() {

    struct rlimit limit;

    CHECK(0 == getrlimit(RLIMIT_AS, &limit)) << errno;

    return limit.rlim_max;

}



void setMemoryLimit(size_t cur, size_t max) {

    const struct rlimit kLimit = {

       .rlim_cur = cur,

       .rlim_max = max,

    };

    CHECK(0 == setrlimit(RLIMIT_AS, &kLimit)) << errno;

}



extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {

    static constexpr size_t kMemLimit = 1 * 1024 * 1024;

    size_t hardLimit = getHardMemoryLimit();

    setMemoryLimit(std::min(kMemLimit, hardLimit), hardLimit);



    if (size <= 1) return 0;  // no use



    // avoid timeouts, see b/142617274, b/142473153
@@ -102,5 +124,7 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { 


    provider.PickValueInArray(fuzzBackend)(std::move(provider));



    setMemoryLimit(hardLimit, hardLimit);



    return 0;

}