15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef SANDBOX_LINUX_SECCOMP_BPF_SYSCALL_ITERATOR_H__
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define SANDBOX_LINUX_SECCOMP_BPF_SYSCALL_ITERATOR_H__
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdint.h>
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace playground2 {
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Iterates over the entire system call range from 0..0xFFFFFFFFu. This
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// iterator is aware of how system calls look like and will skip quickly
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// over ranges that can't contain system calls. It iterates more slowly
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// whenever it reaches a range that is potentially problematic, returning
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the last invalid value before a valid range of system calls, and the
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// first invalid value after a valid range of syscalls. It iterates over
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// individual values whenever it is in the normal range for system calls
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// (typically MIN_SYSCALL..MAX_SYSCALL).
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// If |invalid_only| is true, this iterator will only return invalid
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// syscall numbers, but will still skip quickly over invalid ranges,
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// returning the first invalid value in the range and then skipping
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// to the last invalid value in the range.
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Example usage:
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//   for (SyscallIterator iter(false); !iter.Done(); ) {
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//     uint32_t sysnum = iter.Next();
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//     // Do something with sysnum.
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//   }
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(markus): Make this a classic C++ iterator.
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SyscallIterator {
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public:
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  explicit SyscallIterator(bool invalid_only)
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      : invalid_only_(invalid_only),
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        done_(false),
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        num_(0) {}
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool Done() const { return done_; }
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint32_t Next();
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  static bool IsValid(uint32_t num);
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private:
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  static bool IsArmPrivate(uint32_t num);
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool     invalid_only_;
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool     done_;
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint32_t num_;
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DISALLOW_IMPLICIT_CONSTRUCTORS(SyscallIterator);
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace playground2
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif  // SANDBOX_LINUX_SECCOMP_BPF_SYSCALL_ITERATOR_H__
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
57