static int
shmrmall(void)
{
	int mib[4];
	struct shm_sysctl_info *shmsi;
	size_t i, len;
	int result = 0;

	mib[0] = CTL_KERN;
	mib[1] = KERN_SYSVIPC;
	mib[2] = KERN_SYSVIPC_INFO;
	mib[3] = KERN_SYSVIPC_SHM_INFO;

	if (sysctl(mib, 4, NULL, &len, NULL, 0) == -1)
		err(1, "sysctl(KERN_SYSVIPC_SHM_INFO)");

	if ((shmsi = malloc(len)) == NULL)
		err(1, "malloc");
	if (sysctl(mib, 4, shmsi, &len, NULL, 0) == -1) {
		free(shmsi);
		err(1, "sysctl(KERN_SYSVIPC_SHM_INFO)");
	}

	for (i = 0; i < shmsi->shminfo.shmmni; i++) {
		struct shmid_ds_sysctl *shmptr = &shmsi->shmids[i];
		if (shmptr->shm_perm.mode & 0x0800)
			result -= shmrm((key_t)0,
			    (int)IXSEQ_TO_IPCID(i, shmptr->shm_perm));
	}
	free(shmsi);
	return result;
}
Ejemplo n.º 2
0
int
main(int argc, char **argv)
{
    int c, result, errflg, target_id;
    key_t target_key;

    errflg = 0;
    signal(SIGSYS, not_configured);
    while ((c = getopt(argc, argv, ":q:m:s:Q:M:S:")) != -1) {

	signaled = 0;
	switch (c) {
	case 'q':
	case 'm':
	case 's':
	    target_id = atoi(optarg);
	    if (c == 'q')
		result = msgrm(0, target_id);
	    else if (c == 'm')
		result = shmrm(0, target_id);
	    else
		result = semrm(0, target_id);
	    if (result < 0) {
		errflg++;
		if (!signaled)
		    warn("%sid(%d)", IPC_TO_STR(toupper(c)), target_id);
		else
		    warnx("%ss are not configured in the running kernel",
			  IPC_TO_STRING(toupper(c)));
	    }
	    break;
	case 'Q':
	case 'M':
	case 'S':
	    target_key = atol(optarg);
	    if (target_key == IPC_PRIVATE) {
		warnx("can't remove private %ss", IPC_TO_STRING(c));
		continue;
	    }
	    if (c == 'Q')
		result = msgrm(target_key, 0);
	    else if (c == 'M')
		result = shmrm(target_key, 0);
	    else
		result = semrm(target_key, 0);
	    if (result < 0) {
		errflg++;
		if (!signaled)
		    warn("%s key %ld aborted", IPC_TO_STR(c), target_key);
		else
		    warnx("%ss are not configured in the running kernel",
			  IPC_TO_STRING(c));
	    }
	    break;
	case ':':
	    warnx("option -%c requires an argument\n", optopt);
	    usage();
	case '?':
	    warnx("unrecognized option: -%c\n", optopt);
	    usage();
	}
    }

    if (optind != argc) {
	    warnx("unknown argument: %s\n", argv[optind]);
	    usage();
    }
    exit(errflg);
}
Ejemplo n.º 3
0
int
main(int argc, char *argv[])
{
	int c, result, target_id;
	key_t target_key;

	while ((c = getopt(argc, argv, "q:m:s:Q:M:S:vWy")) != -1) {

		signaled = 0;
		switch (c) {
		case 'v':
			rmverbose++;
			break;
		case 'y':
			use_sysctl = 0;
			break;
		}
	}

	optind = 1;
	errflg = 0;
	signal(SIGSYS, not_configured);
	while ((c = getopt(argc, argv, "q:m:s:Q:M:S:vWy")) != -1) {

		signaled = 0;
		switch (c) {
		case 'q':
		case 'm':
		case 's':
			target_id = atoi(optarg);
			if (c == 'q')
				result = msgrm(0, target_id);
			else if (c == 'm')
				result = shmrm(0, target_id);
			else
				result = semrm(0, target_id);
			if (result < 0) {
				errflg++;
				if (!signaled)
					warn("%sid(%d): ",
					    IPC_TO_STR(toupper(c)), target_id);
				else
					warnx(
					    "%ss are not configured "
					    "in the running kernel",
					    IPC_TO_STRING(toupper(c)));
			}
			break;
		case 'Q':
		case 'M':
		case 'S':
			target_key = atol(optarg);
			if (target_key == IPC_PRIVATE) {
				warnx("can't remove private %ss",
				    IPC_TO_STRING(c));
				continue;
			}
			if (c == 'Q')
				result = msgrm(target_key, 0);
			else if (c == 'M')
				result = shmrm(target_key, 0);
			else
				result = semrm(target_key, 0);
			if (result < 0) {
				errflg++;
				if (!signaled)
					warn("%ss(%ld): ",
					    IPC_TO_STR(c), target_key);
				else
					warnx("%ss are not configured "
					    "in the running kernel",
					    IPC_TO_STRING(c));
			}
			break;
		case 'v':
		case 'y':
			/* Handled in other getopt() loop */
			break;
		case 'W':
			msgrm(-1, 0);
			shmrm(-1, 0);
			semrm(-1, 0);
			break;
		case ':':
			fprintf(stderr,
			    "option -%c requires an argument\n", optopt);
			usage();
		case '?':
			fprintf(stderr, "unrecognized option: -%c\n", optopt);
			usage();
		}
	}

	if (optind != argc) {
		fprintf(stderr, "unknown argument: %s\n", argv[optind]);
		usage();
	}
	exit(errflg);
}
Ejemplo n.º 4
0
// Application entry point.
//
// Gets a list of SHMs on the system.
// Then, removes them.
// Gets a list of sems on the system.
// Then, removes them.
// Prints out totals of removed objects.
int main(int argc, char **argv)
{
  int  procs,rank;
  int  shm_rm=0,sem_rm=0,shm_fl=0;
  int *rbuf=NULL;
  int  i;


  // Initialize MPI
  if( MPI_Init(&argc,&argv) != MPI_SUCCESS ) {
    fprintf(stderr,"Error starting MPI. Terminating.\n");
    return 1;
  }
  MPI_Comm_size(MPI_COMM_WORLD,&procs);
  MPI_Comm_rank(MPI_COMM_WORLD,&rank);

  // Remove all SHMs
  {
    struct shm_info shm_info;
    struct shmid_ds shmseg;
    int             maxid, shmid, id;

    // Get the largest SHM id on the system
    maxid = shmctl(0, SHM_INFO, (struct shmid_ds *) (void *) &shm_info);
    if( maxid < 0 ) {
      fprintf(stderr,"shmctl() says shared memory is not supported.\n");
      MPI_Abort(1,MPI_COMM_WORLD);
    }
    
    // Check each id less than the max
    for(id=shm_rm=shm_fl=0; id <= maxid; id++) {
      // Get info on the SHM
      shmid = shmctl(id, SHM_STAT, &shmseg);
      if( shmid < 0 ) {
	// No SHM with this id exists; keep trying the others.
	continue;
      }
      // Remove the SHM if it is not the system SHM (id == 0)
      if( shmseg.shm_perm.__key ) {
	if( shmrm(shmseg.shm_perm.__key) < 0 ) {
	  // The SHM could not be removed for some reason...
	  shm_fl++;
	  continue;
	}
	// The SHM was removed
	shm_rm++;
      }
    }
  }

  // Remove all semaphores
  {
    struct semid_ds semary;
    struct seminfo  seminfo;
    union  semun    arg;
    int             maxid, semid, id;

    // Get the largest sem id on the system
    arg.array = (unsigned short *) (void *) &seminfo;
    maxid = semctl(0, 0, SEM_INFO, arg);
    if( maxid < 0 ) {
      fprintf(stderr,"semctl() says semaphores are not supported.\n");
      MPI_Abort(1,MPI_COMM_WORLD);
    }

    // Check each id less than the max
    for(id=sem_rm=0; id <= maxid; id++) {
      // Get info on the sem
      arg.buf = (struct semid_ds *) &semary;
      semid = semctl(id, 0, SEM_STAT, arg);
      if( semid < 0 ) {
        // No semaphore with this id exists; keep trying the others.
        continue;
      }
      // Remove the semaphore set
      if( semrm(semary.sem_perm.__key) < 0 ) {
	// The semaphore set could not be removed for some reason...
	continue;
      }
      // The semaphore set was removed
      sem_rm++;
    }

  }

  // Only root needs the array (entry per node)
  if( !rank ) {
    if( !(rbuf=(int*)malloc(procs*sizeof(int))) ) {
      // Malloc failed
      MPI_Abort(1,MPI_COMM_WORLD);
    }
  }
   
  // Gather all removed SHM counts
  MPI_Gather(&shm_rm, 1, MPI_INT, rbuf, 1, MPI_INT, 0, MPI_COMM_WORLD);
  
  // Only root sums for total count
  if( !rank ) {
    for(i=shm_rm=0; i<procs; i++) {
      shm_rm += rbuf[i];
    }
  }

  // Gather all removed sem counts
  MPI_Gather(&sem_rm, 1, MPI_INT, rbuf, 1, MPI_INT, 0, MPI_COMM_WORLD);

  // Only root sums for total count
  if( !rank ) {
    for(i=sem_rm=0; i<procs; i++) {
      sem_rm += rbuf[i];
    }
  }

  // Gather all not removed shm counts
  MPI_Gather(&shm_fl, 1, MPI_INT, rbuf, 1, MPI_INT, 0, MPI_COMM_WORLD);

  // Only root sums for total count
  if( !rank ) {
    for(i=shm_fl=0; i<procs; i++) {
      shm_fl += rbuf[i];
    }
  }

  // Only root needs to report stats
  if( !rank ) {
    // Print out number of removed segments
    printf("Removed a total of %d SHMs for %d nodes.\n",shm_rm,procs);
    printf("Removed a total of %d sems for %d nodes.\n",sem_rm,procs);

    // Print out any failures
    if( shm_fl ) {
      printf("\n!! Failed to remove %d SHMs for %d nodes:\n",shm_fl,procs);
      for(i=0; i<procs; i++) {
	if( rbuf[i] ) {
	  printf("Rank %d still has %d SHMs.\n",i,rbuf[i]);
	}
      }
    }

    // Cleanup
    free(rbuf);
  }

  // Mark that we have cleaned
  if( !rank ) {
    FILE *f;

    if( !(f=fopen("cleaned","w")) ) {
      MPI_Abort(1,MPI_COMM_WORLD);
    }
    fclose(f);
  }

  // Done with MPI
  MPI_Finalize();

  // Return success
  return 0;
}
int 
main(int argc, char *argv[])
{
	int     c, result, errflg, target_id;
	key_t   target_key;

	setprogname(argv[0]);
	errflg = 0;
	(void)signal(SIGSYS, not_configured);
	while ((c = getopt(argc, argv, "q:m:s:Q:M:S:")) != -1) {
		signaled = 0;
		target_id = 0;
		target_key = 0;
		result = 0;

		if (optarg != NULL && strcmp(optarg, "all") == 0) {
			switch (c) {
			case 'm':
			case 'M':
				result = shmrmall();
				break;
			case 'q':
			case 'Q':
				result = msgrmall();
				break;
			case 's':
			case 'S':
				result = semrmall();
				break;
			default:
				usage();
			}
		} else {
			switch (c) {
			case 'q':
			case 'm':
			case 's':
				target_id = atoi(optarg);
				break;
			case 'Q':
			case 'M':
			case 'S':
				target_key = atol(optarg);
				if (target_key == IPC_PRIVATE) {
					warnx("can't remove private %ss",
					    IPC_TO_STRING(c));
					continue;
				}
				break;
			default:
				usage();
			}
			switch (c) {
			case 'q':
				result = msgrm((key_t)0, target_id);
				break;
			case 'm':
				result = shmrm((key_t)0, target_id);
				break;
			case 's':
				result = semrm((key_t)0, target_id);
				break;
			case 'Q':
				result = msgrm(target_key, 0);
				break;
			case 'M':
				result = shmrm(target_key, 0);
				break;
			case 'S':
				result = semrm(target_key, 0);
				break;
			}
		}
		if (result < 0) {
			if (!signaled) {
				if (target_id) {
					warn("%sid(%d): ",
					    IPC_TO_STR(toupper(c)), target_id);
					errflg++;
				} else if (target_key) {
					warn("%skey(%ld): ", IPC_TO_STR(c),
					    (long)target_key);
					errflg++;
				}
			} else {
				errflg++;
				warnx("%ss are not configured in "
				    "the running kernel",
				    IPC_TO_STRING(toupper(c)));
			}
		}
	}

	if (optind != argc) {
		warnx("Unknown argument: %s", argv[optind]);
		usage();
	}
	return errflg;
}