process.cc revision 12431
111723Sar4jc@virginia.edu/* 211723Sar4jc@virginia.edu * Copyright (c) 2004-2005 The Regents of The University of Michigan 311723Sar4jc@virginia.edu * Copyright (c) 2016 The University of Virginia 411723Sar4jc@virginia.edu * All rights reserved. 511723Sar4jc@virginia.edu * 611723Sar4jc@virginia.edu * Redistribution and use in source and binary forms, with or without 711723Sar4jc@virginia.edu * modification, are permitted provided that the following conditions are 811723Sar4jc@virginia.edu * met: redistributions of source code must retain the above copyright 911723Sar4jc@virginia.edu * notice, this list of conditions and the following disclaimer; 1011723Sar4jc@virginia.edu * redistributions in binary form must reproduce the above copyright 1111723Sar4jc@virginia.edu * notice, this list of conditions and the following disclaimer in the 1211723Sar4jc@virginia.edu * documentation and/or other materials provided with the distribution; 1311723Sar4jc@virginia.edu * neither the name of the copyright holders nor the names of its 1411723Sar4jc@virginia.edu * contributors may be used to endorse or promote products derived from 1511723Sar4jc@virginia.edu * this software without specific prior written permission. 1611723Sar4jc@virginia.edu * 1711723Sar4jc@virginia.edu * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 1811723Sar4jc@virginia.edu * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 1911723Sar4jc@virginia.edu * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 2011723Sar4jc@virginia.edu * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 2111723Sar4jc@virginia.edu * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2211723Sar4jc@virginia.edu * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 2311723Sar4jc@virginia.edu * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 2411723Sar4jc@virginia.edu * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 2511723Sar4jc@virginia.edu * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 2611723Sar4jc@virginia.edu * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 2711723Sar4jc@virginia.edu * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 2811723Sar4jc@virginia.edu * 2911723Sar4jc@virginia.edu * Authors: Gabe Black 3011723Sar4jc@virginia.edu * Ali Saidi 3111723Sar4jc@virginia.edu * Korey Sewell 3211723Sar4jc@virginia.edu * Alec Roelke 3311723Sar4jc@virginia.edu */ 3411723Sar4jc@virginia.edu#include "arch/riscv/process.hh" 3511723Sar4jc@virginia.edu 3611964Sar4jc@virginia.edu#include <algorithm> 3711964Sar4jc@virginia.edu#include <cstddef> 3811964Sar4jc@virginia.edu#include <iostream> 3912394Sar4jc@virginia.edu#include <iterator> 4011964Sar4jc@virginia.edu#include <map> 4111964Sar4jc@virginia.edu#include <string> 4211723Sar4jc@virginia.edu#include <vector> 4311723Sar4jc@virginia.edu 4411723Sar4jc@virginia.edu#include "arch/riscv/isa_traits.hh" 4511723Sar4jc@virginia.edu#include "base/loader/elf_object.hh" 4611723Sar4jc@virginia.edu#include "base/loader/object_file.hh" 4712334Sgabeblack@google.com#include "base/logging.hh" 4812394Sar4jc@virginia.edu#include "base/random.hh" 4911723Sar4jc@virginia.edu#include "cpu/thread_context.hh" 5011964Sar4jc@virginia.edu#include "debug/Stack.hh" 5111723Sar4jc@virginia.edu#include "mem/page_table.hh" 5211964Sar4jc@virginia.edu#include "params/Process.hh" 5311854Sbrandon.potter@amd.com#include "sim/aux_vector.hh" 5411723Sar4jc@virginia.edu#include "sim/process.hh" 5511723Sar4jc@virginia.edu#include "sim/process_impl.hh" 5611800Sbrandon.potter@amd.com#include "sim/syscall_return.hh" 5711723Sar4jc@virginia.edu#include "sim/system.hh" 5811723Sar4jc@virginia.edu 5911723Sar4jc@virginia.eduusing namespace std; 6011723Sar4jc@virginia.eduusing namespace RiscvISA; 6111723Sar4jc@virginia.edu 6212431Sgabeblack@google.comRiscvProcess::RiscvProcess(ProcessParams *params, ObjectFile *objFile) : 6312431Sgabeblack@google.com Process(params, new FuncPageTable(params->name, params->pid), objFile) 6411723Sar4jc@virginia.edu{ 6512431Sgabeblack@google.com fatal_if(!params->useArchPT, "Arch page tables not implemented."); 6611970Sar4jc@virginia.edu const Addr stack_base = 0x7FFFFFFFFFFFFFFFL; 6712393Sar4jc@virginia.edu const Addr max_stack_size = 8 * 1024 * 1024; 6811964Sar4jc@virginia.edu const Addr next_thread_stack_base = stack_base - max_stack_size; 6911964Sar4jc@virginia.edu const Addr brk_point = roundUp(objFile->bssBase() + objFile->bssSize(), 7011964Sar4jc@virginia.edu PageBytes); 7111970Sar4jc@virginia.edu const Addr mmap_end = 0x4000000000000000L; 7211905SBrandon.Potter@amd.com memState = make_shared<MemState>(brk_point, stack_base, max_stack_size, 7311964Sar4jc@virginia.edu next_thread_stack_base, mmap_end); 7411723Sar4jc@virginia.edu} 7511723Sar4jc@virginia.edu 7611723Sar4jc@virginia.eduvoid 7711851Sbrandon.potter@amd.comRiscvProcess::initState() 7811723Sar4jc@virginia.edu{ 7911851Sbrandon.potter@amd.com Process::initState(); 8011723Sar4jc@virginia.edu 8111723Sar4jc@virginia.edu argsInit<uint64_t>(PageBytes); 8211723Sar4jc@virginia.edu} 8311723Sar4jc@virginia.edu 8411723Sar4jc@virginia.edutemplate<class IntType> void 8511851Sbrandon.potter@amd.comRiscvProcess::argsInit(int pageSize) 8611723Sar4jc@virginia.edu{ 8712394Sar4jc@virginia.edu const int RandomBytes = 16; 8812394Sar4jc@virginia.edu 8911723Sar4jc@virginia.edu updateBias(); 9011964Sar4jc@virginia.edu objFile->loadSections(initVirtMem); 9111964Sar4jc@virginia.edu ElfObject* elfObject = dynamic_cast<ElfObject*>(objFile); 9211964Sar4jc@virginia.edu memState->setStackMin(memState->getStackBase()); 9311723Sar4jc@virginia.edu 9411964Sar4jc@virginia.edu // Determine stack size and populate auxv 9511964Sar4jc@virginia.edu Addr stack_top = memState->getStackMin(); 9612394Sar4jc@virginia.edu stack_top -= RandomBytes; 9711964Sar4jc@virginia.edu for (const string& arg: argv) 9811964Sar4jc@virginia.edu stack_top -= arg.size() + 1; 9911964Sar4jc@virginia.edu for (const string& env: envp) 10011964Sar4jc@virginia.edu stack_top -= env.size() + 1; 10111964Sar4jc@virginia.edu stack_top &= -sizeof(Addr); 10211723Sar4jc@virginia.edu 10311964Sar4jc@virginia.edu vector<AuxVector<IntType>> auxv; 10411964Sar4jc@virginia.edu if (elfObject != nullptr) { 10511964Sar4jc@virginia.edu auxv.push_back({M5_AT_ENTRY, objFile->entryPoint()}); 10611964Sar4jc@virginia.edu auxv.push_back({M5_AT_PHNUM, elfObject->programHeaderCount()}); 10711964Sar4jc@virginia.edu auxv.push_back({M5_AT_PHENT, elfObject->programHeaderSize()}); 10811964Sar4jc@virginia.edu auxv.push_back({M5_AT_PHDR, elfObject->programHeaderTable()}); 10911964Sar4jc@virginia.edu auxv.push_back({M5_AT_PAGESZ, PageBytes}); 11011964Sar4jc@virginia.edu auxv.push_back({M5_AT_SECURE, 0}); 11111964Sar4jc@virginia.edu auxv.push_back({M5_AT_RANDOM, stack_top}); 11211964Sar4jc@virginia.edu auxv.push_back({M5_AT_NULL, 0}); 11311964Sar4jc@virginia.edu } 11411964Sar4jc@virginia.edu stack_top -= (1 + argv.size()) * sizeof(Addr) + 11511964Sar4jc@virginia.edu (1 + envp.size()) * sizeof(Addr) + 11611964Sar4jc@virginia.edu sizeof(Addr) + 2 * sizeof(IntType) * auxv.size(); 11711964Sar4jc@virginia.edu stack_top &= -2*sizeof(Addr); 11811964Sar4jc@virginia.edu memState->setStackSize(memState->getStackBase() - stack_top); 11911964Sar4jc@virginia.edu allocateMem(roundDown(stack_top, pageSize), 12011964Sar4jc@virginia.edu roundUp(memState->getStackSize(), pageSize)); 12111964Sar4jc@virginia.edu 12212394Sar4jc@virginia.edu // Copy random bytes (for AT_RANDOM) to stack 12312394Sar4jc@virginia.edu memState->setStackMin(memState->getStackMin() - RandomBytes); 12412394Sar4jc@virginia.edu uint8_t at_random[RandomBytes]; 12512394Sar4jc@virginia.edu generate(begin(at_random), end(at_random), 12612394Sar4jc@virginia.edu [&]{ return random_mt.random(0, 0xFF); }); 12712394Sar4jc@virginia.edu initVirtMem.writeBlob(memState->getStackMin(), at_random, RandomBytes); 12812039Sar4jc@virginia.edu 12911964Sar4jc@virginia.edu // Copy argv to stack 13011964Sar4jc@virginia.edu vector<Addr> argPointers; 13111964Sar4jc@virginia.edu for (const string& arg: argv) { 13211964Sar4jc@virginia.edu memState->setStackMin(memState->getStackMin() - (arg.size() + 1)); 13311964Sar4jc@virginia.edu initVirtMem.writeString(memState->getStackMin(), arg.c_str()); 13411964Sar4jc@virginia.edu argPointers.push_back(memState->getStackMin()); 13511964Sar4jc@virginia.edu if (DTRACE(Stack)) { 13611964Sar4jc@virginia.edu string wrote; 13711964Sar4jc@virginia.edu initVirtMem.readString(wrote, argPointers.back()); 13811964Sar4jc@virginia.edu DPRINTFN("Wrote arg \"%s\" to address %p\n", 13911964Sar4jc@virginia.edu wrote, (void*)memState->getStackMin()); 14011964Sar4jc@virginia.edu } 14111964Sar4jc@virginia.edu } 14211964Sar4jc@virginia.edu argPointers.push_back(0); 14311964Sar4jc@virginia.edu 14411964Sar4jc@virginia.edu // Copy envp to stack 14511964Sar4jc@virginia.edu vector<Addr> envPointers; 14611964Sar4jc@virginia.edu for (const string& env: envp) { 14711964Sar4jc@virginia.edu memState->setStackMin(memState->getStackMin() - (env.size() + 1)); 14811964Sar4jc@virginia.edu initVirtMem.writeString(memState->getStackMin(), env.c_str()); 14911964Sar4jc@virginia.edu envPointers.push_back(memState->getStackMin()); 15011964Sar4jc@virginia.edu DPRINTF(Stack, "Wrote env \"%s\" to address %p\n", 15111964Sar4jc@virginia.edu env, (void*)memState->getStackMin()); 15211964Sar4jc@virginia.edu } 15311964Sar4jc@virginia.edu envPointers.push_back(0); 15411964Sar4jc@virginia.edu 15511964Sar4jc@virginia.edu // Align stack 15611964Sar4jc@virginia.edu memState->setStackMin(memState->getStackMin() & -sizeof(Addr)); 15711964Sar4jc@virginia.edu 15811964Sar4jc@virginia.edu // Calculate bottom of stack 15911964Sar4jc@virginia.edu memState->setStackMin(memState->getStackMin() - 16011964Sar4jc@virginia.edu ((1 + argv.size()) * sizeof(Addr) + 16111964Sar4jc@virginia.edu (1 + envp.size()) * sizeof(Addr) + 16211964Sar4jc@virginia.edu sizeof(Addr) + 2 * sizeof(IntType) * auxv.size())); 16311964Sar4jc@virginia.edu memState->setStackMin(memState->getStackMin() & -2*sizeof(Addr)); 16411964Sar4jc@virginia.edu Addr sp = memState->getStackMin(); 16511964Sar4jc@virginia.edu const auto pushOntoStack = 16611964Sar4jc@virginia.edu [this, &sp](const uint8_t* data, const size_t size) { 16711964Sar4jc@virginia.edu initVirtMem.writeBlob(sp, data, size); 16811964Sar4jc@virginia.edu sp += size; 16911964Sar4jc@virginia.edu }; 17011964Sar4jc@virginia.edu 17111964Sar4jc@virginia.edu // Push argc and argv pointers onto stack 17211964Sar4jc@virginia.edu IntType argc = htog((IntType)argv.size()); 17311964Sar4jc@virginia.edu DPRINTF(Stack, "Wrote argc %d to address %p\n", 17411964Sar4jc@virginia.edu argv.size(), (void*)sp); 17511964Sar4jc@virginia.edu pushOntoStack((uint8_t*)&argc, sizeof(IntType)); 17611964Sar4jc@virginia.edu for (const Addr& argPointer: argPointers) { 17711964Sar4jc@virginia.edu DPRINTF(Stack, "Wrote argv pointer %p to address %p\n", 17811964Sar4jc@virginia.edu (void*)argPointer, (void*)sp); 17911964Sar4jc@virginia.edu pushOntoStack((uint8_t*)&argPointer, sizeof(Addr)); 18011723Sar4jc@virginia.edu } 18111723Sar4jc@virginia.edu 18211964Sar4jc@virginia.edu // Push env pointers onto stack 18311964Sar4jc@virginia.edu for (const Addr& envPointer: envPointers) { 18411964Sar4jc@virginia.edu DPRINTF(Stack, "Wrote envp pointer %p to address %p\n", 18511964Sar4jc@virginia.edu (void*)envPointer, (void*)sp); 18611964Sar4jc@virginia.edu pushOntoStack((uint8_t*)&envPointer, sizeof(Addr)); 18711723Sar4jc@virginia.edu } 18811723Sar4jc@virginia.edu 18911964Sar4jc@virginia.edu // Push aux vector onto stack 19011964Sar4jc@virginia.edu std::map<IntType, string> aux_keys = { 19111964Sar4jc@virginia.edu {M5_AT_ENTRY, "M5_AT_ENTRY"}, 19211964Sar4jc@virginia.edu {M5_AT_PHNUM, "M5_AT_PHNUM"}, 19311964Sar4jc@virginia.edu {M5_AT_PHENT, "M5_AT_PHENT"}, 19411964Sar4jc@virginia.edu {M5_AT_PHDR, "M5_AT_PHDR"}, 19511964Sar4jc@virginia.edu {M5_AT_PAGESZ, "M5_AT_PAGESZ"}, 19611964Sar4jc@virginia.edu {M5_AT_SECURE, "M5_AT_SECURE"}, 19711964Sar4jc@virginia.edu {M5_AT_RANDOM, "M5_AT_RANDOM"}, 19811964Sar4jc@virginia.edu {M5_AT_NULL, "M5_AT_NULL"} 19911964Sar4jc@virginia.edu }; 20011964Sar4jc@virginia.edu for (const AuxVector<IntType>& aux: auxv) { 20111964Sar4jc@virginia.edu DPRINTF(Stack, "Wrote aux key %s to address %p\n", 20211964Sar4jc@virginia.edu aux_keys[aux.a_type], (void*)sp); 20311964Sar4jc@virginia.edu pushOntoStack((uint8_t*)&aux.a_type, sizeof(IntType)); 20411964Sar4jc@virginia.edu DPRINTF(Stack, "Wrote aux value %x to address %p\n", 20511964Sar4jc@virginia.edu aux.a_val, (void*)sp); 20611964Sar4jc@virginia.edu pushOntoStack((uint8_t*)&aux.a_val, sizeof(IntType)); 20711723Sar4jc@virginia.edu } 20811723Sar4jc@virginia.edu 20911723Sar4jc@virginia.edu ThreadContext *tc = system->getThreadContext(contextIds[0]); 21011905SBrandon.Potter@amd.com tc->setIntReg(StackPointerReg, memState->getStackMin()); 21111723Sar4jc@virginia.edu tc->pcState(getStartPC()); 21211964Sar4jc@virginia.edu 21311964Sar4jc@virginia.edu memState->setStackMin(roundDown(memState->getStackMin(), pageSize)); 21411723Sar4jc@virginia.edu} 21511723Sar4jc@virginia.edu 21611723Sar4jc@virginia.eduRiscvISA::IntReg 21711851Sbrandon.potter@amd.comRiscvProcess::getSyscallArg(ThreadContext *tc, int &i) 21811723Sar4jc@virginia.edu{ 21912412Sar4jc@virginia.edu // If a larger index is requested than there are syscall argument 22012412Sar4jc@virginia.edu // registers, return 0 22111806Sar4jc@virginia.edu RiscvISA::IntReg retval = 0; 22212412Sar4jc@virginia.edu if (i < SyscallArgumentRegs.size()) 22311806Sar4jc@virginia.edu retval = tc->readIntReg(SyscallArgumentRegs[i]); 22411806Sar4jc@virginia.edu i++; 22511806Sar4jc@virginia.edu return retval; 22611723Sar4jc@virginia.edu} 22711723Sar4jc@virginia.edu 22811723Sar4jc@virginia.eduvoid 22911851Sbrandon.potter@amd.comRiscvProcess::setSyscallArg(ThreadContext *tc, int i, RiscvISA::IntReg val) 23011723Sar4jc@virginia.edu{ 23111723Sar4jc@virginia.edu tc->setIntReg(SyscallArgumentRegs[i], val); 23211723Sar4jc@virginia.edu} 23311723Sar4jc@virginia.edu 23411723Sar4jc@virginia.eduvoid 23511851Sbrandon.potter@amd.comRiscvProcess::setSyscallReturn(ThreadContext *tc, SyscallReturn sysret) 23611723Sar4jc@virginia.edu{ 23711723Sar4jc@virginia.edu if (sysret.successful()) { 23811723Sar4jc@virginia.edu // no error 23911723Sar4jc@virginia.edu tc->setIntReg(SyscallPseudoReturnReg, sysret.returnValue()); 24011723Sar4jc@virginia.edu } else { 24111723Sar4jc@virginia.edu // got an error, return details 24211723Sar4jc@virginia.edu tc->setIntReg(SyscallPseudoReturnReg, sysret.errnoValue()); 24311723Sar4jc@virginia.edu } 24411723Sar4jc@virginia.edu} 245