testcases/kernel/containers/sysvipc/mesgq_nstest.c

/* *************************************************************************
* Copyright (c) International Business Machines Corp., 2009
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
* the GNU General Public License for more details.
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* Author: Veerendra C <vechandr@in.ibm.com>
*
* In Parent Process , create mesgq with key 154326L
* Now create container by passing 1 of the flag values..
* 	Flag = clone, clone(CLONE_NEWIPC), or unshare(CLONE_NEWIPC)
* In cloned process, try to access the created mesgq
* Test PASS: If the mesgq is readable when flag is None.
* Test FAIL: If the mesgq is readable when flag is Unshare or Clone.
***************************************************************************/

#define _GNU_SOURCE 1
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <libclone.h>
#include "test.h"

#define KEY_VAL		154326L
#define UNSHARESTR	"unshare"
#define CLONESTR	"clone"
#define NONESTR		"none"

char *TCID = "mesgq_nstest";
int TST_TOTAL = 1;
int p1[2];
int p2[2];
struct msg_buf {
		long int mtype;	 /* type of received/sent message */
		char mtext[80];	 /* text of the message */
	} msg;

void mesgq_read(id)
{
	int READMAX = 80;
	int n;
	/* read msg type 5 on the Q; msgtype, flags are last 2 params..*/

	n = msgrcv(id, &msg, READMAX, 5, 0);
	if (n == -1)
		perror("msgrcv"), tst_exit();

	tst_resm(TINFO, "Mesg read of %d bytes; Type %ld: Msg: %.*s\n",
					n, msg.mtype, n, msg.mtext);
}

int check_mesgq(void *vtest)
{
	char buf[3];
	int id;

	close(p1[1]);
	close(p2[0]);

	read(p1[0], buf, 3);
	id = msgget(KEY_VAL, 0);
	if (id == -1)
		write(p2[1], "notfnd", 7);
	else {
		write(p2[1], "exists", 7);
		mesgq_read(id);
	}
	tst_exit();

	return 0;
}

int main(int argc, char *argv[])
{
	int ret, use_clone = T_NONE, id, n;
	char *tsttype = NONESTR;
	char buf[7];

	if (argc != 2) {
		tst_resm(TFAIL, "Usage: %s <clone|unshare|none>\n", argv[0]);
		tst_resm(TFAIL, " where clone, unshare, or fork specifies"
				" unshare method.");
		tst_exit();
	}

	/* Using PIPE's to sync between container and Parent */
	if (pipe(p1) == -1) { perror("pipe"); exit(EXIT_FAILURE); }
	if (pipe(p2) == -1) { perror("pipe"); exit(EXIT_FAILURE); }

	tsttype = NONESTR;

	if (strcmp(argv[1], "clone") == 0) {
		use_clone = T_CLONE;
		tsttype = CLONESTR;
	} else if (strcmp(argv[1], "unshare") == 0) {
		use_clone = T_UNSHARE;
		tsttype = UNSHARESTR;
	}

	id = msgget(KEY_VAL, IPC_CREAT | IPC_EXCL | 0600);
	if (id == -1) {
		perror("msgget");
		/* Retry without attempting to create the MQ */
		id = msgget(KEY_VAL, 0);
		if (id == -1) perror("msgget failure"), exit(1);
	}

	msg.mtype = 5;
	strcpy(msg.mtext, "Message of type 5!");
	n = msgsnd(id, &msg, strlen( msg.mtext), 0);
	if (n == -1)
		perror("msgsnd"), tst_exit();

	tst_resm(TINFO, "mesgq namespaces test : %s\n", tsttype);
	/* fire off the test */
	ret = do_clone_unshare_test(use_clone, CLONE_NEWIPC, check_mesgq, NULL);
	if (ret < 0) {
		tst_resm(TFAIL, "%s failed\n", tsttype);
		tst_exit();
	}

	close(p1[0]);
	close(p2[1]);
	write(p1[1], "go", 3);

	read(p2[0], buf, 7);
	if (strcmp(buf, "exists") == 0) {
		if (use_clone == T_NONE)
			tst_resm(TPASS, "Plain cloned process found mesgq "
				  "inside container\n");
		else
			tst_resm(TFAIL, "%s: Container init process found mesgq\n",
				tsttype);
	} else {
		if (use_clone == T_NONE)
			tst_resm(TFAIL, "Plain cloned process didn't find mesgq\n");
		else
			tst_resm(TPASS, "%s: Container didn't find mesgq", tsttype);
	}

	/* Delete the mesgQ */
	id = msgget(KEY_VAL, 0);
	msgctl(id, IPC_RMID, NULL);

	tst_exit();

	tst_exit();
}