githubmirror/kmod

test/test-mod-double-ref.c
78 строк2 KB
Lucas De Marchi
Improve test of double references
15 дек 2011, 19:49
15 дек 2011, 19:4922907a1
#include <stdio.h>
#include <stdlib.h>
#include <stddef.h>
#include <errno.h>
#include <unistd.h>
#include <inttypes.h>
#include <string.h>
#include <libkmod.h>


int main(int argc, char *argv[])
{
	struct kmod_ctx *ctx;
	struct kmod_module *mod1, *mod2;
	const char *modname;
	int err;

	if (argc < 2) {
		fprintf(stderr, "ERR: Provide an alias name\n");
		return EXIT_FAILURE;
	}

	modname = argv[1];

	ctx = kmod_new(NULL, NULL);
	if (ctx == NULL)
		exit(EXIT_FAILURE);

	printf("libkmod version %s\n", VERSION);

	err = kmod_module_new_from_name(ctx, modname, &mod1);
	if (err < 0) {
		fprintf(stderr, "error creating module: '%s'\n", strerror(-err));
		goto fail;
	}

	printf("modname='%s' obj=%p\n", modname, mod1);

	err = kmod_module_new_from_name(ctx, modname, &mod2);
	if (err < 0) {
		fprintf(stderr, "error creating module: '%s'\n", strerror(-err));
		goto fail;
	}

	printf("modname='%s' obj=%p\n", modname, mod2);

	kmod_module_unref(mod1);
	kmod_module_unref(mod2);

	/* same thing, but now unref the first module */

	err = kmod_module_new_from_name(ctx, modname, &mod1);
	if (err < 0) {
		fprintf(stderr, "error creating module: '%s'\n", strerror(-err));
		goto fail;
	}

	printf("modname='%s' obj=%p\n", modname, mod1);

	kmod_module_unref(mod1);

	err = kmod_module_new_from_name(ctx, modname, &mod2);
	if (err < 0) {
		fprintf(stderr, "error creating module: '%s'\n", strerror(-err));
		goto fail;
	}

	printf("modname='%s' obj=%p\n", modname, mod2);

	kmod_module_unref(mod2);
	kmod_unref(ctx);

	return EXIT_SUCCESS;

fail:
	kmod_unref(ctx);
	return EXIT_FAILURE;
}