/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 * ident        "%Z%%M% %I%     %E% SMI"
 */

import java.io.File;
import java.io.IOException;
import java.util.List;
import org.opensolaris.os.dtrace.*;

/**
 * Regression for multi-aggregation printa() corner cases.
 */
public class TestMultiAggPrinta {
    static int exitStatus;

    // Gets a string representation of the given PrintaRecord minus the
    // timestamp of the aggregate snapshot, so that the output is
    // comparable across multiple test runs.
    static String
    printaRecordString(PrintaRecord rec)
    {
        StringBuffer buf = new StringBuffer();
        buf.append(PrintaRecord.class.getName());
        buf.append("[aggregations = ");
        buf.append(rec.getAggregations());
        buf.append(", formattedStrings = ");
        buf.append(rec.getFormattedStrings());
        buf.append(", tuples = ");
        buf.append(rec.getTuples());
        buf.append(", output = ");
        buf.append(rec.getOutput());
        buf.append(']');
        return buf.toString();
    }

    static String
    probeDataString(ProbeData data)
    {
        StringBuffer buf = new StringBuffer();
        buf.append(ProbeData.class.getName());
        buf.append("[epid = ");
        buf.append(data.getEnabledProbeID());
        // Do not include cpu, since it can change across test runs
        buf.append(", enabledProbeDescription = ");
        buf.append(data.getEnabledProbeDescription());
        buf.append(", flow = ");
        buf.append(data.getFlow());
        buf.append(", records = ");

        List <org.opensolaris.os.dtrace.Record> records = data.getRecords();
        org.opensolaris.os.dtrace.Record record;
        Object value;
        buf.append('[');
        for (int i = 0; i < records.size(); ++i) {
            if (i > 0) {
                buf.append(", ");
            }
            record = records.get(i);
            if (record instanceof ValueRecord) {
                value = ((ValueRecord)record).getValue();
                if (value instanceof String) {
                    buf.append("\"");
                    buf.append((String)value);
                    buf.append("\"");
                } else {
                    buf.append(record);
                }
            } else if (record instanceof PrintaRecord) {
                PrintaRecord printa = (PrintaRecord)record;
                buf.append(printaRecordString(printa));
            } else {
                buf.append(record);
            }
        }
        buf.append(']');
        return buf.toString();
    }

    public static void
    main(String[] args)
    {
        if (args.length != 1) {
            System.err.println("usage: java TestMultiAggPrinta <script>");
            System.exit(2);
        }

        final Consumer consumer = new LocalConsumer();
        consumer.addConsumerListener(new ConsumerAdapter() {
            public void dataReceived(DataEvent e) {
                ProbeData data = e.getProbeData();
                List <org.opensolaris.os.dtrace.Record> records =
                    data.getRecords();
                for (org.opensolaris.os.dtrace.Record r : records) {
                    if (r instanceof ExitRecord) {
                        ExitRecord exitRecord = (ExitRecord)r;
                        exitStatus = exitRecord.getStatus();
                    }
                }
                System.out.println(probeDataString(e.getProbeData()));
            }
            public void consumerStopped(ConsumerEvent e) {
                consumer.close();
                System.exit(exitStatus);
            }
        });

        File file = new File(args[0]);
        try {
            consumer.open();
            consumer.compile(file);
            consumer.enable();
            consumer.go();
        } catch (DTraceException e) {
            e.printStackTrace();
            System.exit(1);
        } catch (IOException e) {
            e.printStackTrace();
            System.exit(1);
        }
    }
}