statetrace.cc revision 7414 
1/*
2 * Copyright (c) 2006-2007 The Regents of The University of Michigan
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are
7 * met: redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer;
9 * redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution;
12 * neither the name of the copyright holders nor the names of its
13 * contributors may be used to endorse or promote products derived from
14 * this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
20 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 *
28 * Authors: Gabe Black
29 */
30
31#include <cstring>
32#include <errno.h>
33#include <fstream>
34#include <iostream>
35#include <netdb.h>
36#include <netinet/in.h>
37#include <stdio.h>
38#include <string>
39#include <sys/ptrace.h>
40#include <sys/socket.h>
41#include <sys/types.h>
42#include <sys/wait.h>
43#include <unistd.h>
44
45#include "printer.hh"
46#include "tracechild.hh"
47
48using namespace std;
49
50void printUsage(const char * execName)
51{
52    cout << execName << " <options> -- <command> <arguments>" << endl;
53    cout << "options:" << endl;
54    cout << "         -h          print this help" << endl;
55    cout << "         --host      remote m5 host to connect to" << endl;
56    cout << "         -r          print register names" << endl;
57    cout << "         -i          print initial stack state" << endl;
58    cout << "         -nt         don't print an instruction trace" << endl;
59}
60
61int main(int argc, char * argv[], char * envp[])
62{
63    TraceChild * child = genTraceChild();
64    string args;
65    int startProgramArgs;
66
67    //Parse the command line arguments
68    bool printInitial = false;
69    bool printTrace = true;
70    string host = "localhost";
71
72    if (argc == 1) {
73        printUsage(argv[0]);
74        return 0;
75    }
76    for(int x = 1; x < argc; x++)
77    {
78        if(!strcmp(argv[x], "-h"))
79        {
80            printUsage(argv[0]);
81            return 0;
82        }
83        if(!strcmp(argv[x], "--host"))
84        {
85            x++;
86            if(x >= argc)
87            {
88                cerr << "Incorrect usage.\n" << endl;
89                printUsage(argv[0]);
90                return 1;
91            }
92            host = argv[x];
93        }
94        else if(!strcmp(argv[x], "-r"))
95        {
96            cout << "Legal register names:" << endl;
97            int numRegs = child->getNumRegs();
98            for(unsigned int x = 0; x < numRegs; x++)
99            {
100                cout << "\t" << child->getRegName(x) << endl;
101            }
102            return 0;
103        }
104        else if(!strcmp(argv[x], "-i"))
105        {
106            printInitial = true;
107        }
108        else if(!strcmp(argv[x], "-nt"))
109        {
110            printTrace = false;
111        }
112        else if(!strcmp(argv[x], "--"))
113        {
114            x++;
115            if(x >= argc)
116            {
117                cerr << "Incorrect usage.\n" << endl;
118                printUsage(argv[0]);
119                return 1;
120            }
121            startProgramArgs = x;
122            break;
123        }
124        else
125        {
126            cerr << "Incorrect usage.\n" << endl;
127            printUsage(argv[0]);
128            return 1;
129        }
130    }
131    if(!child->startTracing(argv[startProgramArgs],
132                argv + startProgramArgs))
133    {
134        cerr << "Couldn't start target program" << endl;
135        return 1;
136    }
137    child->step();
138    if(printInitial)
139    {
140        child->outputStartState(cout);
141    }
142    if(printTrace)
143    {
144        // Connect to m5
145        bool portSet = false;
146        int port;
147        int sock = socket(AF_INET, SOCK_STREAM, 0);
148        if(sock < 0)
149        {
150            cerr << "Error opening socket! " << strerror(errno) << endl;
151            return 1;
152        }
153        struct hostent *server;
154        server = gethostbyname(host.c_str());
155        if(!server)
156        {
157            cerr << "Couldn't get host ip! " << strerror(errno) << endl;
158            return 1;
159        }
160        struct sockaddr_in serv_addr;
161        bzero((char *)&serv_addr, sizeof(serv_addr));
162        serv_addr.sin_family = AF_INET;
163        bcopy((char *)server->h_addr,
164                (char *)&serv_addr.sin_addr.s_addr,
165                server->h_length);
166        serv_addr.sin_port = htons(8000);
167        if(connect(sock, (sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
168        {
169            cerr << "Couldn't connect to server! " << strerror(errno) << endl;
170            return 1;
171        }
172        while(child->isTracing())
173        {
174                if(!child->sendState(sock))
175                    break;
176                child->step();
177        }
178    }
179    if(!child->stopTracing())
180    {
181            cerr << "Couldn't stop child" << endl;
182            return 1;
183    }
184    return 0;
185}
186
187