xref/libmicro/pthread_create.c

/*
 * 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
 * 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 2005 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <pthread.h>
#include <errno.h>

#include "libmicro.h"

typedef struct {
    pthread_t       *ts_threads;
    pthread_attr_t      *ts_attr;
    pthread_mutex_t     ts_lock;
} tsd_t;

static int              opts = 0;

int
benchmark_init()
{
    lm_defN = "pthread";

    lm_tsdsize = sizeof (tsd_t);

    (void) sprintf(lm_usage,
        "       [-s stacksize] (specify stacksize)\n"
        "notes: measures pthread_create\n");

    (void) sprintf(lm_optstr, "s:");

    return (0);
}

int
benchmark_optswitch(int opt, char *optarg)
{
    switch (opt) {
    case 's':
        opts = sizetoll(optarg);
        break;
    default:
        return (-1);
    }

    return (0);
}

int
benchmark_initworker(void *tsd)
{
    tsd_t           *ts = (tsd_t *)tsd;
    int errors = 0;

    ts->ts_threads = calloc(lm_optB, sizeof (pthread_t));
    (void) pthread_mutex_init(&ts->ts_lock, NULL);

    if (opts) {
        ts->ts_attr = malloc(sizeof (pthread_attr_t));
        (void) pthread_attr_init(ts->ts_attr);
        if ((errors = pthread_attr_setstacksize(ts->ts_attr, opts))
            != 0) {
            errno = errors;
            perror("pthread_attr_setstacksize");
        }
    } else
        ts->ts_attr = NULL;

    return (errors?1:0);
}

int
benchmark_initbatch(void *tsd)
{
    tsd_t           *ts = (tsd_t *)tsd;

    (void) pthread_mutex_lock(&ts->ts_lock);

    return (0);
}


void *
func(void *tsd)
{
    tsd_t           *ts = (tsd_t *)tsd;

    (void) pthread_mutex_lock(&ts->ts_lock);
    (void) pthread_mutex_unlock(&ts->ts_lock);

    return (tsd);
}

int
benchmark(void *tsd, result_t *res)
{
    int         i;
    tsd_t           *ts = (tsd_t *)tsd;
    int error;

    for (i = 0; i < lm_optB; i++) {
        if ((error = pthread_create(ts->ts_threads + i,
            ts->ts_attr, func, tsd)) != 0) {
            errno = error;
            perror("pthread_create");
            ts->ts_threads[i] = 0;
            res->re_errors++;
            return (0);
        }
    }
    res->re_count = lm_optB;

    return (0);
}

int
benchmark_finibatch(void *tsd)
{
    tsd_t           *ts = (tsd_t *)tsd;
    int i;
    int errors = 0;

    (void) pthread_mutex_unlock(&ts->ts_lock);

    for (i = 0; i < lm_optB; i++)
        if (ts->ts_threads[i] == 0 ||
            pthread_join(ts->ts_threads[i], NULL) < 0) {
            errors++;
        }
    return (errors);
}