1*ec63e07aSXin Li // Copyright 2020 Google LLC 2*ec63e07aSXin Li // 3*ec63e07aSXin Li // Licensed under the Apache License, Version 2.0 (the "License"); 4*ec63e07aSXin Li // you may not use this file except in compliance with the License. 5*ec63e07aSXin Li // You may obtain a copy of the License at 6*ec63e07aSXin Li // 7*ec63e07aSXin Li // https://www.apache.org/licenses/LICENSE-2.0 8*ec63e07aSXin Li // 9*ec63e07aSXin Li // Unless required by applicable law or agreed to in writing, software 10*ec63e07aSXin Li // distributed under the License is distributed on an "AS IS" BASIS, 11*ec63e07aSXin Li // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12*ec63e07aSXin Li // See the License for the specific language governing permissions and 13*ec63e07aSXin Li // limitations under the License. 14*ec63e07aSXin Li 15*ec63e07aSXin Li #ifndef CONTRIB_LODEPNG_EXAMPLES_SANDBOX_H_ 16*ec63e07aSXin Li #define CONTRIB_LODEPNG_EXAMPLES_SANDBOX_H_ 17*ec63e07aSXin Li 18*ec63e07aSXin Li #include <syscall.h> 19*ec63e07aSXin Li 20*ec63e07aSXin Li #include "lodepng_sapi.sapi.h" // NOLINT(build/include) 21*ec63e07aSXin Li 22*ec63e07aSXin Li class SapiLodepngSandbox : public LodepngSandbox { 23*ec63e07aSXin Li public: SapiLodepngSandbox(const std::string & images_path)24*ec63e07aSXin Li explicit SapiLodepngSandbox(const std::string& images_path) 25*ec63e07aSXin Li : images_path_(images_path) {} 26*ec63e07aSXin Li 27*ec63e07aSXin Li private: ModifyPolicy(sandbox2::PolicyBuilder *)28*ec63e07aSXin Li std::unique_ptr<sandbox2::Policy> ModifyPolicy( 29*ec63e07aSXin Li sandbox2::PolicyBuilder*) override { 30*ec63e07aSXin Li return sandbox2::PolicyBuilder() 31*ec63e07aSXin Li .AllowRead() 32*ec63e07aSXin Li .AllowWrite() 33*ec63e07aSXin Li .AllowOpen() 34*ec63e07aSXin Li .AllowSystemMalloc() 35*ec63e07aSXin Li .AllowExit() 36*ec63e07aSXin Li .AllowStat() 37*ec63e07aSXin Li .AllowGetPIDs() 38*ec63e07aSXin Li .AddDirectoryAt(images_path_, "/output/", /*is_ro=*/false) 39*ec63e07aSXin Li .AllowSyscalls({ 40*ec63e07aSXin Li __NR_futex, 41*ec63e07aSXin Li __NR_lseek, 42*ec63e07aSXin Li __NR_close, 43*ec63e07aSXin Li }) 44*ec63e07aSXin Li .BuildOrDie(); 45*ec63e07aSXin Li } 46*ec63e07aSXin Li 47*ec63e07aSXin Li const std::string images_path_; 48*ec63e07aSXin Li }; 49*ec63e07aSXin Li 50*ec63e07aSXin Li #endif // CONTRIB_LODEPNG_EXAMPLES_SANDBOX__ 51