usr/src/cmd/mdb/common/modules/libumem/proc_kludges.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License, Version 1.0 only
   6  * (the "License").  You may not use this file except in compliance
   7  * with the License.
   8  *
   9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
  10  * or http://www.opensolaris.org/os/licensing.
  11  * See the License for the specific language governing permissions
  12  * and limitations under the License.
  13  *
  14  * When distributing Covered Code, include this CDDL HEADER in each
  15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  16  * If applicable, add the following below this CDDL HEADER, with the
  17  * fields enclosed by brackets "[]" replaced with your own identifying
  18  * information: Portions Copyright [yyyy] [name of copyright owner]
  19  *
  20  * CDDL HEADER END
  21  */
  22 /*
  23  * Copyright 2000-2002 Sun Microsystems, Inc.  All rights reserved.
  24  * Use is subject to license terms.
  25  */
  26
  27 #pragma ident   "%Z%%M% %I%     %E% SMI"
  28
  29 #include <mdb/mdb_modapi.h>
  30 #include <sys/machelf.h>
  31
  32 #include <libproc.h>
  33 #include <stdio.h>
  34
  35 #include "proc_kludges.h"
  36
  37 typedef struct prockuldge_mappings {
  38         struct ps_prochandle *pkm_Pr;
  39
  40         uint_t pkm_idx;
  41
  42         uint_t pkm_count;
  43         uint_t pkm_max;
  44
  45         prmap_t *pkm_mappings;
  46
  47         uint_t pkm_old_max;
  48         prmap_t *pkm_old_mappings;
  49 } prockludge_mappings_t;
  50
  51 /* ARGSUSED */
  52 static int
  53 prockludge_mappings_iter(prockludge_mappings_t *pkm, const prmap_t *pmp,
  54     const char *object_name)
  55 {
  56         if (pkm->pkm_count >= pkm->pkm_max) {
  57                 int s = pkm->pkm_max ? pkm->pkm_max * 2 : 16;
  58
  59                 pkm->pkm_old_max = pkm->pkm_max;
  60                 pkm->pkm_old_mappings = pkm->pkm_mappings;
  61                 pkm->pkm_max = s;
  62                 pkm->pkm_mappings = mdb_alloc(sizeof (prmap_t) * s, UM_SLEEP);
  63
  64                 bcopy(pkm->pkm_old_mappings, pkm->pkm_mappings,
  65                     sizeof (prmap_t) * pkm->pkm_old_max);
  66
  67                 mdb_free(pkm->pkm_old_mappings,
  68                     sizeof (prmap_t) * pkm->pkm_old_max);
  69
  70                 pkm->pkm_old_mappings = NULL;
  71                 pkm->pkm_old_max = 0;
  72         }
  73         bcopy(pmp, &pkm->pkm_mappings[pkm->pkm_count++], sizeof (prmap_t));
  74
  75         return (0);
  76 }
  77
  78 int
  79 prockludge_mappings_walk_init(mdb_walk_state_t *mws)
  80 {
  81         struct ps_prochandle *Pr;
  82         int rc;
  83
  84         prockludge_mappings_t *pkm;
  85
  86         if (mdb_get_xdata("pshandle", &Pr, sizeof (Pr)) == -1) {
  87                 mdb_warn("couldn't read pshandle xdata");
  88                 return (WALK_ERR);
  89         }
  90
  91         pkm = mdb_zalloc(sizeof (prockludge_mappings_t), UM_SLEEP);
  92         pkm->pkm_Pr = Pr;
  93         mws->walk_data = pkm;
  94
  95         rc = Pmapping_iter(Pr, (proc_map_f *)prockludge_mappings_iter, pkm);
  96         if (rc != 0) {
  97                 mdb_warn("Pmapping_iter failed");
  98                 /* clean up */
  99                 prockludge_mappings_walk_fini(mws);
 100                 return (WALK_ERR);
 101         }
 102         return (WALK_NEXT);
 103 }
 104
 105 int
 106 prockludge_mappings_walk_step(mdb_walk_state_t *wsp)
 107 {
 108         prockludge_mappings_t *pkm = wsp->walk_data;
 109         int status;
 110
 111         if (pkm->pkm_idx >= pkm->pkm_count)
 112                 return (WALK_DONE);
 113
 114         status = wsp->walk_callback(0, &pkm->pkm_mappings[pkm->pkm_idx++],
 115             wsp->walk_cbdata);
 116
 117         return (status);
 118 }
 119
 120 void
 121 prockludge_mappings_walk_fini(mdb_walk_state_t *wsp)
 122 {
 123         prockludge_mappings_t *pkm = wsp->walk_data;
 124         if (pkm != NULL) {
 125                 if (pkm->pkm_old_mappings != NULL) {
 126                         mdb_free(pkm->pkm_old_mappings,
 127                             sizeof (prmap_t) * pkm->pkm_old_max);
 128                 }
 129                 if (pkm->pkm_mappings &&
 130                     pkm->pkm_mappings != pkm->pkm_old_mappings) {
 131                         mdb_free(pkm->pkm_mappings,
 132                             sizeof (prmap_t) * pkm->pkm_max);
 133                 }
 134                 mdb_free(pkm, sizeof (prockludge_mappings_t));
 135         }
 136 }
 137
 138 static int add_count = 0;
 139
 140 void
 141 prockludge_add_walkers(void)
 142 {
 143         mdb_walker_t w;
 144
 145         if (add_count++ == 0) {
 146                 w.walk_name = KLUDGE_MAPWALK_NAME;
 147                 w.walk_descr = "kludge: walk the process' prmap_ts";
 148                 w.walk_init = prockludge_mappings_walk_init;
 149                 w.walk_step = prockludge_mappings_walk_step;
 150                 w.walk_fini = prockludge_mappings_walk_fini;
 151                 w.walk_init_arg = NULL;
 152
 153                 if (mdb_add_walker(&w) == -1) {
 154                         mdb_warn("unable to add walker "KLUDGE_MAPWALK_NAME);
 155                 }
 156         }
 157 }
 158
 159 void
 160 prockludge_remove_walkers(void)
 161 {
 162         if (--add_count == 0) {
 163                 mdb_remove_walker(KLUDGE_MAPWALK_NAME);
 164         }
 165 }