#!/usr/bin/env python3
# Copyright (C) 2019 The Android Open Source Project
#
# 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.

# This synthetic trace tests a short lived process which is forked, performs
# a rename (to simulate an exec) then dies before the /proc scraper finds the
# real process name

from os import sys, path

from synth_common import CLONE_THREAD
import synth_common

trace = synth_common.create_trace()

# Create a parent process which  will be forked below.
trace.add_packet(ts=1)
trace.add_process(10, 0, "parent")
trace.add_process(11, 0, "short_lived")

# Make the short lived thread emit a begin/end pair then exit.
trace.add_ftrace_packet(0)
trace.add_atrace_begin(ts=10, tid=11, pid=11, buf='test')
trace.add_atrace_end(ts=11, tid=11, pid=11)
trace.add_process_free(ts=12, tid=11, comm='short_lived', prio=0)

# Now, reuse tid 11 and emit a begin on the parent and an end on the child.
# This is an abuse of sync events (which technically should always be associated
# with a single thread) but this happens all the time in places like ART so
# we should support it.
trace.add_ftrace_packet(0)
trace.add_newtask(
    ts=15, tid=10, new_tid=11, new_comm='child', flags=CLONE_THREAD)
trace.add_rename(
    ts=17, tid=11, old_comm='child', new_comm='true_name', oom_score_adj=1000)
trace.add_atrace_begin(ts=18, tid=10, pid=10, buf='test')
trace.add_atrace_end(ts=19, tid=11, pid=10)

sys.stdout.buffer.write(trace.trace.SerializeToString())