remove put_cpu_no_resched()
[sfrench/cifs-2.6.git] / arch / ia64 / kernel / perfmon.c
1 /*
2  * This file implements the perfmon-2 subsystem which is used
3  * to program the IA-64 Performance Monitoring Unit (PMU).
4  *
5  * The initial version of perfmon.c was written by
6  * Ganesh Venkitachalam, IBM Corp.
7  *
8  * Then it was modified for perfmon-1.x by Stephane Eranian and
9  * David Mosberger, Hewlett Packard Co.
10  *
11  * Version Perfmon-2.x is a rewrite of perfmon-1.x
12  * by Stephane Eranian, Hewlett Packard Co.
13  *
14  * Copyright (C) 1999-2005  Hewlett Packard Co
15  *               Stephane Eranian <eranian@hpl.hp.com>
16  *               David Mosberger-Tang <davidm@hpl.hp.com>
17  *
18  * More information about perfmon available at:
19  *      http://www.hpl.hp.com/research/linux/perfmon
20  */
21
22 #include <linux/module.h>
23 #include <linux/kernel.h>
24 #include <linux/sched.h>
25 #include <linux/interrupt.h>
26 #include <linux/proc_fs.h>
27 #include <linux/seq_file.h>
28 #include <linux/init.h>
29 #include <linux/vmalloc.h>
30 #include <linux/mm.h>
31 #include <linux/sysctl.h>
32 #include <linux/list.h>
33 #include <linux/file.h>
34 #include <linux/poll.h>
35 #include <linux/vfs.h>
36 #include <linux/smp.h>
37 #include <linux/pagemap.h>
38 #include <linux/mount.h>
39 #include <linux/bitops.h>
40 #include <linux/capability.h>
41 #include <linux/rcupdate.h>
42 #include <linux/completion.h>
43 #include <linux/tracehook.h>
44
45 #include <asm/errno.h>
46 #include <asm/intrinsics.h>
47 #include <asm/page.h>
48 #include <asm/perfmon.h>
49 #include <asm/processor.h>
50 #include <asm/signal.h>
51 #include <asm/system.h>
52 #include <asm/uaccess.h>
53 #include <asm/delay.h>
54
55 #ifdef CONFIG_PERFMON
56 /*
57  * perfmon context state
58  */
59 #define PFM_CTX_UNLOADED        1       /* context is not loaded onto any task */
60 #define PFM_CTX_LOADED          2       /* context is loaded onto a task */
61 #define PFM_CTX_MASKED          3       /* context is loaded but monitoring is masked due to overflow */
62 #define PFM_CTX_ZOMBIE          4       /* owner of the context is closing it */
63
64 #define PFM_INVALID_ACTIVATION  (~0UL)
65
66 #define PFM_NUM_PMC_REGS        64      /* PMC save area for ctxsw */
67 #define PFM_NUM_PMD_REGS        64      /* PMD save area for ctxsw */
68
69 /*
70  * depth of message queue
71  */
72 #define PFM_MAX_MSGS            32
73 #define PFM_CTXQ_EMPTY(g)       ((g)->ctx_msgq_head == (g)->ctx_msgq_tail)
74
75 /*
76  * type of a PMU register (bitmask).
77  * bitmask structure:
78  *      bit0   : register implemented
79  *      bit1   : end marker
80  *      bit2-3 : reserved
81  *      bit4   : pmc has pmc.pm
82  *      bit5   : pmc controls a counter (has pmc.oi), pmd is used as counter
83  *      bit6-7 : register type
84  *      bit8-31: reserved
85  */
86 #define PFM_REG_NOTIMPL         0x0 /* not implemented at all */
87 #define PFM_REG_IMPL            0x1 /* register implemented */
88 #define PFM_REG_END             0x2 /* end marker */
89 #define PFM_REG_MONITOR         (0x1<<4|PFM_REG_IMPL) /* a PMC with a pmc.pm field only */
90 #define PFM_REG_COUNTING        (0x2<<4|PFM_REG_MONITOR) /* a monitor + pmc.oi+ PMD used as a counter */
91 #define PFM_REG_CONTROL         (0x4<<4|PFM_REG_IMPL) /* PMU control register */
92 #define PFM_REG_CONFIG          (0x8<<4|PFM_REG_IMPL) /* configuration register */
93 #define PFM_REG_BUFFER          (0xc<<4|PFM_REG_IMPL) /* PMD used as buffer */
94
95 #define PMC_IS_LAST(i)  (pmu_conf->pmc_desc[i].type & PFM_REG_END)
96 #define PMD_IS_LAST(i)  (pmu_conf->pmd_desc[i].type & PFM_REG_END)
97
98 #define PMC_OVFL_NOTIFY(ctx, i) ((ctx)->ctx_pmds[i].flags &  PFM_REGFL_OVFL_NOTIFY)
99
100 /* i assumed unsigned */
101 #define PMC_IS_IMPL(i)    (i< PMU_MAX_PMCS && (pmu_conf->pmc_desc[i].type & PFM_REG_IMPL))
102 #define PMD_IS_IMPL(i)    (i< PMU_MAX_PMDS && (pmu_conf->pmd_desc[i].type & PFM_REG_IMPL))
103
104 /* XXX: these assume that register i is implemented */
105 #define PMD_IS_COUNTING(i) ((pmu_conf->pmd_desc[i].type & PFM_REG_COUNTING) == PFM_REG_COUNTING)
106 #define PMC_IS_COUNTING(i) ((pmu_conf->pmc_desc[i].type & PFM_REG_COUNTING) == PFM_REG_COUNTING)
107 #define PMC_IS_MONITOR(i)  ((pmu_conf->pmc_desc[i].type & PFM_REG_MONITOR)  == PFM_REG_MONITOR)
108 #define PMC_IS_CONTROL(i)  ((pmu_conf->pmc_desc[i].type & PFM_REG_CONTROL)  == PFM_REG_CONTROL)
109
110 #define PMC_DFL_VAL(i)     pmu_conf->pmc_desc[i].default_value
111 #define PMC_RSVD_MASK(i)   pmu_conf->pmc_desc[i].reserved_mask
112 #define PMD_PMD_DEP(i)     pmu_conf->pmd_desc[i].dep_pmd[0]
113 #define PMC_PMD_DEP(i)     pmu_conf->pmc_desc[i].dep_pmd[0]
114
115 #define PFM_NUM_IBRS      IA64_NUM_DBG_REGS
116 #define PFM_NUM_DBRS      IA64_NUM_DBG_REGS
117
118 #define CTX_OVFL_NOBLOCK(c)     ((c)->ctx_fl_block == 0)
119 #define CTX_HAS_SMPL(c)         ((c)->ctx_fl_is_sampling)
120 #define PFM_CTX_TASK(h)         (h)->ctx_task
121
122 #define PMU_PMC_OI              5 /* position of pmc.oi bit */
123
124 /* XXX: does not support more than 64 PMDs */
125 #define CTX_USED_PMD(ctx, mask) (ctx)->ctx_used_pmds[0] |= (mask)
126 #define CTX_IS_USED_PMD(ctx, c) (((ctx)->ctx_used_pmds[0] & (1UL << (c))) != 0UL)
127
128 #define CTX_USED_MONITOR(ctx, mask) (ctx)->ctx_used_monitors[0] |= (mask)
129
130 #define CTX_USED_IBR(ctx,n)     (ctx)->ctx_used_ibrs[(n)>>6] |= 1UL<< ((n) % 64)
131 #define CTX_USED_DBR(ctx,n)     (ctx)->ctx_used_dbrs[(n)>>6] |= 1UL<< ((n) % 64)
132 #define CTX_USES_DBREGS(ctx)    (((pfm_context_t *)(ctx))->ctx_fl_using_dbreg==1)
133 #define PFM_CODE_RR     0       /* requesting code range restriction */
134 #define PFM_DATA_RR     1       /* requestion data range restriction */
135
136 #define PFM_CPUINFO_CLEAR(v)    pfm_get_cpu_var(pfm_syst_info) &= ~(v)
137 #define PFM_CPUINFO_SET(v)      pfm_get_cpu_var(pfm_syst_info) |= (v)
138 #define PFM_CPUINFO_GET()       pfm_get_cpu_var(pfm_syst_info)
139
140 #define RDEP(x) (1UL<<(x))
141
142 /*
143  * context protection macros
144  * in SMP:
145  *      - we need to protect against CPU concurrency (spin_lock)
146  *      - we need to protect against PMU overflow interrupts (local_irq_disable)
147  * in UP:
148  *      - we need to protect against PMU overflow interrupts (local_irq_disable)
149  *
150  * spin_lock_irqsave()/spin_unlock_irqrestore():
151  *      in SMP: local_irq_disable + spin_lock
152  *      in UP : local_irq_disable
153  *
154  * spin_lock()/spin_lock():
155  *      in UP : removed automatically
156  *      in SMP: protect against context accesses from other CPU. interrupts
157  *              are not masked. This is useful for the PMU interrupt handler
158  *              because we know we will not get PMU concurrency in that code.
159  */
160 #define PROTECT_CTX(c, f) \
161         do {  \
162                 DPRINT(("spinlock_irq_save ctx %p by [%d]\n", c, task_pid_nr(current))); \
163                 spin_lock_irqsave(&(c)->ctx_lock, f); \
164                 DPRINT(("spinlocked ctx %p  by [%d]\n", c, task_pid_nr(current))); \
165         } while(0)
166
167 #define UNPROTECT_CTX(c, f) \
168         do { \
169                 DPRINT(("spinlock_irq_restore ctx %p by [%d]\n", c, task_pid_nr(current))); \
170                 spin_unlock_irqrestore(&(c)->ctx_lock, f); \
171         } while(0)
172
173 #define PROTECT_CTX_NOPRINT(c, f) \
174         do {  \
175                 spin_lock_irqsave(&(c)->ctx_lock, f); \
176         } while(0)
177
178
179 #define UNPROTECT_CTX_NOPRINT(c, f) \
180         do { \
181                 spin_unlock_irqrestore(&(c)->ctx_lock, f); \
182         } while(0)
183
184
185 #define PROTECT_CTX_NOIRQ(c) \
186         do {  \
187                 spin_lock(&(c)->ctx_lock); \
188         } while(0)
189
190 #define UNPROTECT_CTX_NOIRQ(c) \
191         do { \
192                 spin_unlock(&(c)->ctx_lock); \
193         } while(0)
194
195
196 #ifdef CONFIG_SMP
197
198 #define GET_ACTIVATION()        pfm_get_cpu_var(pmu_activation_number)
199 #define INC_ACTIVATION()        pfm_get_cpu_var(pmu_activation_number)++
200 #define SET_ACTIVATION(c)       (c)->ctx_last_activation = GET_ACTIVATION()
201
202 #else /* !CONFIG_SMP */
203 #define SET_ACTIVATION(t)       do {} while(0)
204 #define GET_ACTIVATION(t)       do {} while(0)
205 #define INC_ACTIVATION(t)       do {} while(0)
206 #endif /* CONFIG_SMP */
207
208 #define SET_PMU_OWNER(t, c)     do { pfm_get_cpu_var(pmu_owner) = (t); pfm_get_cpu_var(pmu_ctx) = (c); } while(0)
209 #define GET_PMU_OWNER()         pfm_get_cpu_var(pmu_owner)
210 #define GET_PMU_CTX()           pfm_get_cpu_var(pmu_ctx)
211
212 #define LOCK_PFS(g)             spin_lock_irqsave(&pfm_sessions.pfs_lock, g)
213 #define UNLOCK_PFS(g)           spin_unlock_irqrestore(&pfm_sessions.pfs_lock, g)
214
215 #define PFM_REG_RETFLAG_SET(flags, val) do { flags &= ~PFM_REG_RETFL_MASK; flags |= (val); } while(0)
216
217 /*
218  * cmp0 must be the value of pmc0
219  */
220 #define PMC0_HAS_OVFL(cmp0)  (cmp0 & ~0x1UL)
221
222 #define PFMFS_MAGIC 0xa0b4d889
223
224 /*
225  * debugging
226  */
227 #define PFM_DEBUGGING 1
228 #ifdef PFM_DEBUGGING
229 #define DPRINT(a) \
230         do { \
231                 if (unlikely(pfm_sysctl.debug >0)) { printk("%s.%d: CPU%d [%d] ", __func__, __LINE__, smp_processor_id(), task_pid_nr(current)); printk a; } \
232         } while (0)
233
234 #define DPRINT_ovfl(a) \
235         do { \
236                 if (unlikely(pfm_sysctl.debug > 0 && pfm_sysctl.debug_ovfl >0)) { printk("%s.%d: CPU%d [%d] ", __func__, __LINE__, smp_processor_id(), task_pid_nr(current)); printk a; } \
237         } while (0)
238 #endif
239
240 /*
241  * 64-bit software counter structure
242  *
243  * the next_reset_type is applied to the next call to pfm_reset_regs()
244  */
245 typedef struct {
246         unsigned long   val;            /* virtual 64bit counter value */
247         unsigned long   lval;           /* last reset value */
248         unsigned long   long_reset;     /* reset value on sampling overflow */
249         unsigned long   short_reset;    /* reset value on overflow */
250         unsigned long   reset_pmds[4];  /* which other pmds to reset when this counter overflows */
251         unsigned long   smpl_pmds[4];   /* which pmds are accessed when counter overflow */
252         unsigned long   seed;           /* seed for random-number generator */
253         unsigned long   mask;           /* mask for random-number generator */
254         unsigned int    flags;          /* notify/do not notify */
255         unsigned long   eventid;        /* overflow event identifier */
256 } pfm_counter_t;
257
258 /*
259  * context flags
260  */
261 typedef struct {
262         unsigned int block:1;           /* when 1, task will blocked on user notifications */
263         unsigned int system:1;          /* do system wide monitoring */
264         unsigned int using_dbreg:1;     /* using range restrictions (debug registers) */
265         unsigned int is_sampling:1;     /* true if using a custom format */
266         unsigned int excl_idle:1;       /* exclude idle task in system wide session */
267         unsigned int going_zombie:1;    /* context is zombie (MASKED+blocking) */
268         unsigned int trap_reason:2;     /* reason for going into pfm_handle_work() */
269         unsigned int no_msg:1;          /* no message sent on overflow */
270         unsigned int can_restart:1;     /* allowed to issue a PFM_RESTART */
271         unsigned int reserved:22;
272 } pfm_context_flags_t;
273
274 #define PFM_TRAP_REASON_NONE            0x0     /* default value */
275 #define PFM_TRAP_REASON_BLOCK           0x1     /* we need to block on overflow */
276 #define PFM_TRAP_REASON_RESET           0x2     /* we need to reset PMDs */
277
278
279 /*
280  * perfmon context: encapsulates all the state of a monitoring session
281  */
282
283 typedef struct pfm_context {
284         spinlock_t              ctx_lock;               /* context protection */
285
286         pfm_context_flags_t     ctx_flags;              /* bitmask of flags  (block reason incl.) */
287         unsigned int            ctx_state;              /* state: active/inactive (no bitfield) */
288
289         struct task_struct      *ctx_task;              /* task to which context is attached */
290
291         unsigned long           ctx_ovfl_regs[4];       /* which registers overflowed (notification) */
292
293         struct completion       ctx_restart_done;       /* use for blocking notification mode */
294
295         unsigned long           ctx_used_pmds[4];       /* bitmask of PMD used            */
296         unsigned long           ctx_all_pmds[4];        /* bitmask of all accessible PMDs */
297         unsigned long           ctx_reload_pmds[4];     /* bitmask of force reload PMD on ctxsw in */
298
299         unsigned long           ctx_all_pmcs[4];        /* bitmask of all accessible PMCs */
300         unsigned long           ctx_reload_pmcs[4];     /* bitmask of force reload PMC on ctxsw in */
301         unsigned long           ctx_used_monitors[4];   /* bitmask of monitor PMC being used */
302
303         unsigned long           ctx_pmcs[PFM_NUM_PMC_REGS];     /*  saved copies of PMC values */
304
305         unsigned int            ctx_used_ibrs[1];               /* bitmask of used IBR (speedup ctxsw in) */
306         unsigned int            ctx_used_dbrs[1];               /* bitmask of used DBR (speedup ctxsw in) */
307         unsigned long           ctx_dbrs[IA64_NUM_DBG_REGS];    /* DBR values (cache) when not loaded */
308         unsigned long           ctx_ibrs[IA64_NUM_DBG_REGS];    /* IBR values (cache) when not loaded */
309
310         pfm_counter_t           ctx_pmds[PFM_NUM_PMD_REGS]; /* software state for PMDS */
311
312         unsigned long           th_pmcs[PFM_NUM_PMC_REGS];      /* PMC thread save state */
313         unsigned long           th_pmds[PFM_NUM_PMD_REGS];      /* PMD thread save state */
314
315         u64                     ctx_saved_psr_up;       /* only contains psr.up value */
316
317         unsigned long           ctx_last_activation;    /* context last activation number for last_cpu */
318         unsigned int            ctx_last_cpu;           /* CPU id of current or last CPU used (SMP only) */
319         unsigned int            ctx_cpu;                /* cpu to which perfmon is applied (system wide) */
320
321         int                     ctx_fd;                 /* file descriptor used my this context */
322         pfm_ovfl_arg_t          ctx_ovfl_arg;           /* argument to custom buffer format handler */
323
324         pfm_buffer_fmt_t        *ctx_buf_fmt;           /* buffer format callbacks */
325         void                    *ctx_smpl_hdr;          /* points to sampling buffer header kernel vaddr */
326         unsigned long           ctx_smpl_size;          /* size of sampling buffer */
327         void                    *ctx_smpl_vaddr;        /* user level virtual address of smpl buffer */
328
329         wait_queue_head_t       ctx_msgq_wait;
330         pfm_msg_t               ctx_msgq[PFM_MAX_MSGS];
331         int                     ctx_msgq_head;
332         int                     ctx_msgq_tail;
333         struct fasync_struct    *ctx_async_queue;
334
335         wait_queue_head_t       ctx_zombieq;            /* termination cleanup wait queue */
336 } pfm_context_t;
337
338 /*
339  * magic number used to verify that structure is really
340  * a perfmon context
341  */
342 #define PFM_IS_FILE(f)          ((f)->f_op == &pfm_file_ops)
343
344 #define PFM_GET_CTX(t)          ((pfm_context_t *)(t)->thread.pfm_context)
345
346 #ifdef CONFIG_SMP
347 #define SET_LAST_CPU(ctx, v)    (ctx)->ctx_last_cpu = (v)
348 #define GET_LAST_CPU(ctx)       (ctx)->ctx_last_cpu
349 #else
350 #define SET_LAST_CPU(ctx, v)    do {} while(0)
351 #define GET_LAST_CPU(ctx)       do {} while(0)
352 #endif
353
354
355 #define ctx_fl_block            ctx_flags.block
356 #define ctx_fl_system           ctx_flags.system
357 #define ctx_fl_using_dbreg      ctx_flags.using_dbreg
358 #define ctx_fl_is_sampling      ctx_flags.is_sampling
359 #define ctx_fl_excl_idle        ctx_flags.excl_idle
360 #define ctx_fl_going_zombie     ctx_flags.going_zombie
361 #define ctx_fl_trap_reason      ctx_flags.trap_reason
362 #define ctx_fl_no_msg           ctx_flags.no_msg
363 #define ctx_fl_can_restart      ctx_flags.can_restart
364
365 #define PFM_SET_WORK_PENDING(t, v)      do { (t)->thread.pfm_needs_checking = v; } while(0);
366 #define PFM_GET_WORK_PENDING(t)         (t)->thread.pfm_needs_checking
367
368 /*
369  * global information about all sessions
370  * mostly used to synchronize between system wide and per-process
371  */
372 typedef struct {
373         spinlock_t              pfs_lock;                  /* lock the structure */
374
375         unsigned int            pfs_task_sessions;         /* number of per task sessions */
376         unsigned int            pfs_sys_sessions;          /* number of per system wide sessions */
377         unsigned int            pfs_sys_use_dbregs;        /* incremented when a system wide session uses debug regs */
378         unsigned int            pfs_ptrace_use_dbregs;     /* incremented when a process uses debug regs */
379         struct task_struct      *pfs_sys_session[NR_CPUS]; /* point to task owning a system-wide session */
380 } pfm_session_t;
381
382 /*
383  * information about a PMC or PMD.
384  * dep_pmd[]: a bitmask of dependent PMD registers
385  * dep_pmc[]: a bitmask of dependent PMC registers
386  */
387 typedef int (*pfm_reg_check_t)(struct task_struct *task, pfm_context_t *ctx, unsigned int cnum, unsigned long *val, struct pt_regs *regs);
388 typedef struct {
389         unsigned int            type;
390         int                     pm_pos;
391         unsigned long           default_value;  /* power-on default value */
392         unsigned long           reserved_mask;  /* bitmask of reserved bits */
393         pfm_reg_check_t         read_check;
394         pfm_reg_check_t         write_check;
395         unsigned long           dep_pmd[4];
396         unsigned long           dep_pmc[4];
397 } pfm_reg_desc_t;
398
399 /* assume cnum is a valid monitor */
400 #define PMC_PM(cnum, val)       (((val) >> (pmu_conf->pmc_desc[cnum].pm_pos)) & 0x1)
401
402 /*
403  * This structure is initialized at boot time and contains
404  * a description of the PMU main characteristics.
405  *
406  * If the probe function is defined, detection is based
407  * on its return value: 
408  *      - 0 means recognized PMU
409  *      - anything else means not supported
410  * When the probe function is not defined, then the pmu_family field
411  * is used and it must match the host CPU family such that:
412  *      - cpu->family & config->pmu_family != 0
413  */
414 typedef struct {
415         unsigned long  ovfl_val;        /* overflow value for counters */
416
417         pfm_reg_desc_t *pmc_desc;       /* detailed PMC register dependencies descriptions */
418         pfm_reg_desc_t *pmd_desc;       /* detailed PMD register dependencies descriptions */
419
420         unsigned int   num_pmcs;        /* number of PMCS: computed at init time */
421         unsigned int   num_pmds;        /* number of PMDS: computed at init time */
422         unsigned long  impl_pmcs[4];    /* bitmask of implemented PMCS */
423         unsigned long  impl_pmds[4];    /* bitmask of implemented PMDS */
424
425         char          *pmu_name;        /* PMU family name */
426         unsigned int  pmu_family;       /* cpuid family pattern used to identify pmu */
427         unsigned int  flags;            /* pmu specific flags */
428         unsigned int  num_ibrs;         /* number of IBRS: computed at init time */
429         unsigned int  num_dbrs;         /* number of DBRS: computed at init time */
430         unsigned int  num_counters;     /* PMC/PMD counting pairs : computed at init time */
431         int           (*probe)(void);   /* customized probe routine */
432         unsigned int  use_rr_dbregs:1;  /* set if debug registers used for range restriction */
433 } pmu_config_t;
434 /*
435  * PMU specific flags
436  */
437 #define PFM_PMU_IRQ_RESEND      1       /* PMU needs explicit IRQ resend */
438
439 /*
440  * debug register related type definitions
441  */
442 typedef struct {
443         unsigned long ibr_mask:56;
444         unsigned long ibr_plm:4;
445         unsigned long ibr_ig:3;
446         unsigned long ibr_x:1;
447 } ibr_mask_reg_t;
448
449 typedef struct {
450         unsigned long dbr_mask:56;
451         unsigned long dbr_plm:4;
452         unsigned long dbr_ig:2;
453         unsigned long dbr_w:1;
454         unsigned long dbr_r:1;
455 } dbr_mask_reg_t;
456
457 typedef union {
458         unsigned long  val;
459         ibr_mask_reg_t ibr;
460         dbr_mask_reg_t dbr;
461 } dbreg_t;
462
463
464 /*
465  * perfmon command descriptions
466  */
467 typedef struct {
468         int             (*cmd_func)(pfm_context_t *ctx, void *arg, int count, struct pt_regs *regs);
469         char            *cmd_name;
470         int             cmd_flags;
471         unsigned int    cmd_narg;
472         size_t          cmd_argsize;
473         int             (*cmd_getsize)(void *arg, size_t *sz);
474 } pfm_cmd_desc_t;
475
476 #define PFM_CMD_FD              0x01    /* command requires a file descriptor */
477 #define PFM_CMD_ARG_READ        0x02    /* command must read argument(s) */
478 #define PFM_CMD_ARG_RW          0x04    /* command must read/write argument(s) */
479 #define PFM_CMD_STOP            0x08    /* command does not work on zombie context */
480
481
482 #define PFM_CMD_NAME(cmd)       pfm_cmd_tab[(cmd)].cmd_name
483 #define PFM_CMD_READ_ARG(cmd)   (pfm_cmd_tab[(cmd)].cmd_flags & PFM_CMD_ARG_READ)
484 #define PFM_CMD_RW_ARG(cmd)     (pfm_cmd_tab[(cmd)].cmd_flags & PFM_CMD_ARG_RW)
485 #define PFM_CMD_USE_FD(cmd)     (pfm_cmd_tab[(cmd)].cmd_flags & PFM_CMD_FD)
486 #define PFM_CMD_STOPPED(cmd)    (pfm_cmd_tab[(cmd)].cmd_flags & PFM_CMD_STOP)
487
488 #define PFM_CMD_ARG_MANY        -1 /* cannot be zero */
489
490 typedef struct {
491         unsigned long pfm_spurious_ovfl_intr_count;     /* keep track of spurious ovfl interrupts */
492         unsigned long pfm_replay_ovfl_intr_count;       /* keep track of replayed ovfl interrupts */
493         unsigned long pfm_ovfl_intr_count;              /* keep track of ovfl interrupts */
494         unsigned long pfm_ovfl_intr_cycles;             /* cycles spent processing ovfl interrupts */
495         unsigned long pfm_ovfl_intr_cycles_min;         /* min cycles spent processing ovfl interrupts */
496         unsigned long pfm_ovfl_intr_cycles_max;         /* max cycles spent processing ovfl interrupts */
497         unsigned long pfm_smpl_handler_calls;
498         unsigned long pfm_smpl_handler_cycles;
499         char pad[SMP_CACHE_BYTES] ____cacheline_aligned;
500 } pfm_stats_t;
501
502 /*
503  * perfmon internal variables
504  */
505 static pfm_stats_t              pfm_stats[NR_CPUS];
506 static pfm_session_t            pfm_sessions;   /* global sessions information */
507
508 static DEFINE_SPINLOCK(pfm_alt_install_check);
509 static pfm_intr_handler_desc_t  *pfm_alt_intr_handler;
510
511 static struct proc_dir_entry    *perfmon_dir;
512 static pfm_uuid_t               pfm_null_uuid = {0,};
513
514 static spinlock_t               pfm_buffer_fmt_lock;
515 static LIST_HEAD(pfm_buffer_fmt_list);
516
517 static pmu_config_t             *pmu_conf;
518
519 /* sysctl() controls */
520 pfm_sysctl_t pfm_sysctl;
521 EXPORT_SYMBOL(pfm_sysctl);
522
523 static ctl_table pfm_ctl_table[]={
524         {
525                 .ctl_name       = CTL_UNNUMBERED,
526                 .procname       = "debug",
527                 .data           = &pfm_sysctl.debug,
528                 .maxlen         = sizeof(int),
529                 .mode           = 0666,
530                 .proc_handler   = &proc_dointvec,
531         },
532         {
533                 .ctl_name       = CTL_UNNUMBERED,
534                 .procname       = "debug_ovfl",
535                 .data           = &pfm_sysctl.debug_ovfl,
536                 .maxlen         = sizeof(int),
537                 .mode           = 0666,
538                 .proc_handler   = &proc_dointvec,
539         },
540         {
541                 .ctl_name       = CTL_UNNUMBERED,
542                 .procname       = "fastctxsw",
543                 .data           = &pfm_sysctl.fastctxsw,
544                 .maxlen         = sizeof(int),
545                 .mode           = 0600,
546                 .proc_handler   =  &proc_dointvec,
547         },
548         {
549                 .ctl_name       = CTL_UNNUMBERED,
550                 .procname       = "expert_mode",
551                 .data           = &pfm_sysctl.expert_mode,
552                 .maxlen         = sizeof(int),
553                 .mode           = 0600,
554                 .proc_handler   = &proc_dointvec,
555         },
556         {}
557 };
558 static ctl_table pfm_sysctl_dir[] = {
559         {
560                 .ctl_name       = CTL_UNNUMBERED,
561                 .procname       = "perfmon",
562                 .mode           = 0555,
563                 .child          = pfm_ctl_table,
564         },
565         {}
566 };
567 static ctl_table pfm_sysctl_root[] = {
568         {
569                 .ctl_name       = CTL_KERN,
570                 .procname       = "kernel",
571                 .mode           = 0555,
572                 .child          = pfm_sysctl_dir,
573         },
574         {}
575 };
576 static struct ctl_table_header *pfm_sysctl_header;
577
578 static int pfm_context_unload(pfm_context_t *ctx, void *arg, int count, struct pt_regs *regs);
579
580 #define pfm_get_cpu_var(v)              __ia64_per_cpu_var(v)
581 #define pfm_get_cpu_data(a,b)           per_cpu(a, b)
582
583 static inline void
584 pfm_put_task(struct task_struct *task)
585 {
586         if (task != current) put_task_struct(task);
587 }
588
589 static inline void
590 pfm_reserve_page(unsigned long a)
591 {
592         SetPageReserved(vmalloc_to_page((void *)a));
593 }
594 static inline void
595 pfm_unreserve_page(unsigned long a)
596 {
597         ClearPageReserved(vmalloc_to_page((void*)a));
598 }
599
600 static inline unsigned long
601 pfm_protect_ctx_ctxsw(pfm_context_t *x)
602 {
603         spin_lock(&(x)->ctx_lock);
604         return 0UL;
605 }
606
607 static inline void
608 pfm_unprotect_ctx_ctxsw(pfm_context_t *x, unsigned long f)
609 {
610         spin_unlock(&(x)->ctx_lock);
611 }
612
613 static inline unsigned int
614 pfm_do_munmap(struct mm_struct *mm, unsigned long addr, size_t len, int acct)
615 {
616         return do_munmap(mm, addr, len);
617 }
618
619 static inline unsigned long 
620 pfm_get_unmapped_area(struct file *file, unsigned long addr, unsigned long len, unsigned long pgoff, unsigned long flags, unsigned long exec)
621 {
622         return get_unmapped_area(file, addr, len, pgoff, flags);
623 }
624
625
626 static int
627 pfmfs_get_sb(struct file_system_type *fs_type, int flags, const char *dev_name, void *data,
628              struct vfsmount *mnt)
629 {
630         return get_sb_pseudo(fs_type, "pfm:", NULL, PFMFS_MAGIC, mnt);
631 }
632
633 static struct file_system_type pfm_fs_type = {
634         .name     = "pfmfs",
635         .get_sb   = pfmfs_get_sb,
636         .kill_sb  = kill_anon_super,
637 };
638
639 DEFINE_PER_CPU(unsigned long, pfm_syst_info);
640 DEFINE_PER_CPU(struct task_struct *, pmu_owner);
641 DEFINE_PER_CPU(pfm_context_t  *, pmu_ctx);
642 DEFINE_PER_CPU(unsigned long, pmu_activation_number);
643 EXPORT_PER_CPU_SYMBOL_GPL(pfm_syst_info);
644
645
646 /* forward declaration */
647 static const struct file_operations pfm_file_ops;
648
649 /*
650  * forward declarations
651  */
652 #ifndef CONFIG_SMP
653 static void pfm_lazy_save_regs (struct task_struct *ta);
654 #endif
655
656 void dump_pmu_state(const char *);
657 static int pfm_write_ibr_dbr(int mode, pfm_context_t *ctx, void *arg, int count, struct pt_regs *regs);
658
659 #include "perfmon_itanium.h"
660 #include "perfmon_mckinley.h"
661 #include "perfmon_montecito.h"
662 #include "perfmon_generic.h"
663
664 static pmu_config_t *pmu_confs[]={
665         &pmu_conf_mont,
666         &pmu_conf_mck,
667         &pmu_conf_ita,
668         &pmu_conf_gen, /* must be last */
669         NULL
670 };
671
672
673 static int pfm_end_notify_user(pfm_context_t *ctx);
674
675 static inline void
676 pfm_clear_psr_pp(void)
677 {
678         ia64_rsm(IA64_PSR_PP);
679         ia64_srlz_i();
680 }
681
682 static inline void
683 pfm_set_psr_pp(void)
684 {
685         ia64_ssm(IA64_PSR_PP);
686         ia64_srlz_i();
687 }
688
689 static inline void
690 pfm_clear_psr_up(void)
691 {
692         ia64_rsm(IA64_PSR_UP);
693         ia64_srlz_i();
694 }
695
696 static inline void
697 pfm_set_psr_up(void)
698 {
699         ia64_ssm(IA64_PSR_UP);
700         ia64_srlz_i();
701 }
702
703 static inline unsigned long
704 pfm_get_psr(void)
705 {
706         unsigned long tmp;
707         tmp = ia64_getreg(_IA64_REG_PSR);
708         ia64_srlz_i();
709         return tmp;
710 }
711
712 static inline void
713 pfm_set_psr_l(unsigned long val)
714 {
715         ia64_setreg(_IA64_REG_PSR_L, val);
716         ia64_srlz_i();
717 }
718
719 static inline void
720 pfm_freeze_pmu(void)
721 {
722         ia64_set_pmc(0,1UL);
723         ia64_srlz_d();
724 }
725
726 static inline void
727 pfm_unfreeze_pmu(void)
728 {
729         ia64_set_pmc(0,0UL);
730         ia64_srlz_d();
731 }
732
733 static inline void
734 pfm_restore_ibrs(unsigned long *ibrs, unsigned int nibrs)
735 {
736         int i;
737
738         for (i=0; i < nibrs; i++) {
739                 ia64_set_ibr(i, ibrs[i]);
740                 ia64_dv_serialize_instruction();
741         }
742         ia64_srlz_i();
743 }
744
745 static inline void
746 pfm_restore_dbrs(unsigned long *dbrs, unsigned int ndbrs)
747 {
748         int i;
749
750         for (i=0; i < ndbrs; i++) {
751                 ia64_set_dbr(i, dbrs[i]);
752                 ia64_dv_serialize_data();
753         }
754         ia64_srlz_d();
755 }
756
757 /*
758  * PMD[i] must be a counter. no check is made
759  */
760 static inline unsigned long
761 pfm_read_soft_counter(pfm_context_t *ctx, int i)
762 {
763         return ctx->ctx_pmds[i].val + (ia64_get_pmd(i) & pmu_conf->ovfl_val);
764 }
765
766 /*
767  * PMD[i] must be a counter. no check is made
768  */
769 static inline void
770 pfm_write_soft_counter(pfm_context_t *ctx, int i, unsigned long val)
771 {
772         unsigned long ovfl_val = pmu_conf->ovfl_val;
773
774         ctx->ctx_pmds[i].val = val  & ~ovfl_val;
775         /*
776          * writing to unimplemented part is ignore, so we do not need to
777          * mask off top part
778          */
779         ia64_set_pmd(i, val & ovfl_val);
780 }
781
782 static pfm_msg_t *
783 pfm_get_new_msg(pfm_context_t *ctx)
784 {
785         int idx, next;
786
787         next = (ctx->ctx_msgq_tail+1) % PFM_MAX_MSGS;
788
789         DPRINT(("ctx_fd=%p head=%d tail=%d\n", ctx, ctx->ctx_msgq_head, ctx->ctx_msgq_tail));
790         if (next == ctx->ctx_msgq_head) return NULL;
791
792         idx =   ctx->ctx_msgq_tail;
793         ctx->ctx_msgq_tail = next;
794
795         DPRINT(("ctx=%p head=%d tail=%d msg=%d\n", ctx, ctx->ctx_msgq_head, ctx->ctx_msgq_tail, idx));
796
797         return ctx->ctx_msgq+idx;
798 }
799
800 static pfm_msg_t *
801 pfm_get_next_msg(pfm_context_t *ctx)
802 {
803         pfm_msg_t *msg;
804
805         DPRINT(("ctx=%p head=%d tail=%d\n", ctx, ctx->ctx_msgq_head, ctx->ctx_msgq_tail));
806
807         if (PFM_CTXQ_EMPTY(ctx)) return NULL;
808
809         /*
810          * get oldest message
811          */
812         msg = ctx->ctx_msgq+ctx->ctx_msgq_head;
813
814         /*
815          * and move forward
816          */
817         ctx->ctx_msgq_head = (ctx->ctx_msgq_head+1) % PFM_MAX_MSGS;
818
819         DPRINT(("ctx=%p head=%d tail=%d type=%d\n", ctx, ctx->ctx_msgq_head, ctx->ctx_msgq_tail, msg->pfm_gen_msg.msg_type));
820
821         return msg;
822 }
823
824 static void
825 pfm_reset_msgq(pfm_context_t *ctx)
826 {
827         ctx->ctx_msgq_head = ctx->ctx_msgq_tail = 0;
828         DPRINT(("ctx=%p msgq reset\n", ctx));
829 }
830
831 static void *
832 pfm_rvmalloc(unsigned long size)
833 {
834         void *mem;
835         unsigned long addr;
836
837         size = PAGE_ALIGN(size);
838         mem  = vmalloc(size);
839         if (mem) {
840                 //printk("perfmon: CPU%d pfm_rvmalloc(%ld)=%p\n", smp_processor_id(), size, mem);
841                 memset(mem, 0, size);
842                 addr = (unsigned long)mem;
843                 while (size > 0) {
844                         pfm_reserve_page(addr);
845                         addr+=PAGE_SIZE;
846                         size-=PAGE_SIZE;
847                 }
848         }
849         return mem;
850 }
851
852 static void
853 pfm_rvfree(void *mem, unsigned long size)
854 {
855         unsigned long addr;
856
857         if (mem) {
858                 DPRINT(("freeing physical buffer @%p size=%lu\n", mem, size));
859                 addr = (unsigned long) mem;
860                 while ((long) size > 0) {
861                         pfm_unreserve_page(addr);
862                         addr+=PAGE_SIZE;
863                         size-=PAGE_SIZE;
864                 }
865                 vfree(mem);
866         }
867         return;
868 }
869
870 static pfm_context_t *
871 pfm_context_alloc(int ctx_flags)
872 {
873         pfm_context_t *ctx;
874
875         /* 
876          * allocate context descriptor 
877          * must be able to free with interrupts disabled
878          */
879         ctx = kzalloc(sizeof(pfm_context_t), GFP_KERNEL);
880         if (ctx) {
881                 DPRINT(("alloc ctx @%p\n", ctx));
882
883                 /*
884                  * init context protection lock
885                  */
886                 spin_lock_init(&ctx->ctx_lock);
887
888                 /*
889                  * context is unloaded
890                  */
891                 ctx->ctx_state = PFM_CTX_UNLOADED;
892
893                 /*
894                  * initialization of context's flags
895                  */
896                 ctx->ctx_fl_block       = (ctx_flags & PFM_FL_NOTIFY_BLOCK) ? 1 : 0;
897                 ctx->ctx_fl_system      = (ctx_flags & PFM_FL_SYSTEM_WIDE) ? 1: 0;
898                 ctx->ctx_fl_no_msg      = (ctx_flags & PFM_FL_OVFL_NO_MSG) ? 1: 0;
899                 /*
900                  * will move to set properties
901                  * ctx->ctx_fl_excl_idle   = (ctx_flags & PFM_FL_EXCL_IDLE) ? 1: 0;
902                  */
903
904                 /*
905                  * init restart semaphore to locked
906                  */
907                 init_completion(&ctx->ctx_restart_done);
908
909                 /*
910                  * activation is used in SMP only
911                  */
912                 ctx->ctx_last_activation = PFM_INVALID_ACTIVATION;
913                 SET_LAST_CPU(ctx, -1);
914
915                 /*
916                  * initialize notification message queue
917                  */
918                 ctx->ctx_msgq_head = ctx->ctx_msgq_tail = 0;
919                 init_waitqueue_head(&ctx->ctx_msgq_wait);
920                 init_waitqueue_head(&ctx->ctx_zombieq);
921
922         }
923         return ctx;
924 }
925
926 static void
927 pfm_context_free(pfm_context_t *ctx)
928 {
929         if (ctx) {
930                 DPRINT(("free ctx @%p\n", ctx));
931                 kfree(ctx);
932         }
933 }
934
935 static void
936 pfm_mask_monitoring(struct task_struct *task)
937 {
938         pfm_context_t *ctx = PFM_GET_CTX(task);
939         unsigned long mask, val, ovfl_mask;
940         int i;
941
942         DPRINT_ovfl(("masking monitoring for [%d]\n", task_pid_nr(task)));
943
944         ovfl_mask = pmu_conf->ovfl_val;
945         /*
946          * monitoring can only be masked as a result of a valid
947          * counter overflow. In UP, it means that the PMU still
948          * has an owner. Note that the owner can be different
949          * from the current task. However the PMU state belongs
950          * to the owner.
951          * In SMP, a valid overflow only happens when task is
952          * current. Therefore if we come here, we know that
953          * the PMU state belongs to the current task, therefore
954          * we can access the live registers.
955          *
956          * So in both cases, the live register contains the owner's
957          * state. We can ONLY touch the PMU registers and NOT the PSR.
958          *
959          * As a consequence to this call, the ctx->th_pmds[] array
960          * contains stale information which must be ignored
961          * when context is reloaded AND monitoring is active (see
962          * pfm_restart).
963          */
964         mask = ctx->ctx_used_pmds[0];
965         for (i = 0; mask; i++, mask>>=1) {
966                 /* skip non used pmds */
967                 if ((mask & 0x1) == 0) continue;
968                 val = ia64_get_pmd(i);
969
970                 if (PMD_IS_COUNTING(i)) {
971                         /*
972                          * we rebuild the full 64 bit value of the counter
973                          */
974                         ctx->ctx_pmds[i].val += (val & ovfl_mask);
975                 } else {
976                         ctx->ctx_pmds[i].val = val;
977                 }
978                 DPRINT_ovfl(("pmd[%d]=0x%lx hw_pmd=0x%lx\n",
979                         i,
980                         ctx->ctx_pmds[i].val,
981                         val & ovfl_mask));
982         }
983         /*
984          * mask monitoring by setting the privilege level to 0
985          * we cannot use psr.pp/psr.up for this, it is controlled by
986          * the user
987          *
988          * if task is current, modify actual registers, otherwise modify
989          * thread save state, i.e., what will be restored in pfm_load_regs()
990          */
991         mask = ctx->ctx_used_monitors[0] >> PMU_FIRST_COUNTER;
992         for(i= PMU_FIRST_COUNTER; mask; i++, mask>>=1) {
993                 if ((mask & 0x1) == 0UL) continue;
994                 ia64_set_pmc(i, ctx->th_pmcs[i] & ~0xfUL);
995                 ctx->th_pmcs[i] &= ~0xfUL;
996                 DPRINT_ovfl(("pmc[%d]=0x%lx\n", i, ctx->th_pmcs[i]));
997         }
998         /*
999          * make all of this visible
1000          */
1001         ia64_srlz_d();
1002 }
1003
1004 /*
1005  * must always be done with task == current
1006  *
1007  * context must be in MASKED state when calling
1008  */
1009 static void
1010 pfm_restore_monitoring(struct task_struct *task)
1011 {
1012         pfm_context_t *ctx = PFM_GET_CTX(task);
1013         unsigned long mask, ovfl_mask;
1014         unsigned long psr, val;
1015         int i, is_system;
1016
1017         is_system = ctx->ctx_fl_system;
1018         ovfl_mask = pmu_conf->ovfl_val;
1019
1020         if (task != current) {
1021                 printk(KERN_ERR "perfmon.%d: invalid task[%d] current[%d]\n", __LINE__, task_pid_nr(task), task_pid_nr(current));
1022                 return;
1023         }
1024         if (ctx->ctx_state != PFM_CTX_MASKED) {
1025                 printk(KERN_ERR "perfmon.%d: task[%d] current[%d] invalid state=%d\n", __LINE__,
1026                         task_pid_nr(task), task_pid_nr(current), ctx->ctx_state);
1027                 return;
1028         }
1029         psr = pfm_get_psr();
1030         /*
1031          * monitoring is masked via the PMC.
1032          * As we restore their value, we do not want each counter to
1033          * restart right away. We stop monitoring using the PSR,
1034          * restore the PMC (and PMD) and then re-establish the psr
1035          * as it was. Note that there can be no pending overflow at
1036          * this point, because monitoring was MASKED.
1037          *
1038          * system-wide session are pinned and self-monitoring
1039          */
1040         if (is_system && (PFM_CPUINFO_GET() & PFM_CPUINFO_DCR_PP)) {
1041                 /* disable dcr pp */
1042                 ia64_setreg(_IA64_REG_CR_DCR, ia64_getreg(_IA64_REG_CR_DCR) & ~IA64_DCR_PP);
1043                 pfm_clear_psr_pp();
1044         } else {
1045                 pfm_clear_psr_up();
1046         }
1047         /*
1048          * first, we restore the PMD
1049          */
1050         mask = ctx->ctx_used_pmds[0];
1051         for (i = 0; mask; i++, mask>>=1) {
1052                 /* skip non used pmds */
1053                 if ((mask & 0x1) == 0) continue;
1054
1055                 if (PMD_IS_COUNTING(i)) {
1056                         /*
1057                          * we split the 64bit value according to
1058                          * counter width
1059                          */
1060                         val = ctx->ctx_pmds[i].val & ovfl_mask;
1061                         ctx->ctx_pmds[i].val &= ~ovfl_mask;
1062                 } else {
1063                         val = ctx->ctx_pmds[i].val;
1064                 }
1065                 ia64_set_pmd(i, val);
1066
1067                 DPRINT(("pmd[%d]=0x%lx hw_pmd=0x%lx\n",
1068                         i,
1069                         ctx->ctx_pmds[i].val,
1070                         val));
1071         }
1072         /*
1073          * restore the PMCs
1074          */
1075         mask = ctx->ctx_used_monitors[0] >> PMU_FIRST_COUNTER;
1076         for(i= PMU_FIRST_COUNTER; mask; i++, mask>>=1) {
1077                 if ((mask & 0x1) == 0UL) continue;
1078                 ctx->th_pmcs[i] = ctx->ctx_pmcs[i];
1079                 ia64_set_pmc(i, ctx->th_pmcs[i]);
1080                 DPRINT(("[%d] pmc[%d]=0x%lx\n",
1081                                         task_pid_nr(task), i, ctx->th_pmcs[i]));
1082         }
1083         ia64_srlz_d();
1084
1085         /*
1086          * must restore DBR/IBR because could be modified while masked
1087          * XXX: need to optimize 
1088          */
1089         if (ctx->ctx_fl_using_dbreg) {
1090                 pfm_restore_ibrs(ctx->ctx_ibrs, pmu_conf->num_ibrs);
1091                 pfm_restore_dbrs(ctx->ctx_dbrs, pmu_conf->num_dbrs);
1092         }
1093
1094         /*
1095          * now restore PSR
1096          */
1097         if (is_system && (PFM_CPUINFO_GET() & PFM_CPUINFO_DCR_PP)) {
1098                 /* enable dcr pp */
1099                 ia64_setreg(_IA64_REG_CR_DCR, ia64_getreg(_IA64_REG_CR_DCR) | IA64_DCR_PP);
1100                 ia64_srlz_i();
1101         }
1102         pfm_set_psr_l(psr);
1103 }
1104
1105 static inline void
1106 pfm_save_pmds(unsigned long *pmds, unsigned long mask)
1107 {
1108         int i;
1109
1110         ia64_srlz_d();
1111
1112         for (i=0; mask; i++, mask>>=1) {
1113                 if (mask & 0x1) pmds[i] = ia64_get_pmd(i);
1114         }
1115 }
1116
1117 /*
1118  * reload from thread state (used for ctxw only)
1119  */
1120 static inline void
1121 pfm_restore_pmds(unsigned long *pmds, unsigned long mask)
1122 {
1123         int i;
1124         unsigned long val, ovfl_val = pmu_conf->ovfl_val;
1125
1126         for (i=0; mask; i++, mask>>=1) {
1127                 if ((mask & 0x1) == 0) continue;
1128                 val = PMD_IS_COUNTING(i) ? pmds[i] & ovfl_val : pmds[i];
1129                 ia64_set_pmd(i, val);
1130         }
1131         ia64_srlz_d();
1132 }
1133
1134 /*
1135  * propagate PMD from context to thread-state
1136  */
1137 static inline void
1138 pfm_copy_pmds(struct task_struct *task, pfm_context_t *ctx)
1139 {
1140         unsigned long ovfl_val = pmu_conf->ovfl_val;
1141         unsigned long mask = ctx->ctx_all_pmds[0];
1142         unsigned long val;
1143         int i;
1144
1145         DPRINT(("mask=0x%lx\n", mask));
1146
1147         for (i=0; mask; i++, mask>>=1) {
1148
1149                 val = ctx->ctx_pmds[i].val;
1150
1151                 /*
1152                  * We break up the 64 bit value into 2 pieces
1153                  * the lower bits go to the machine state in the
1154                  * thread (will be reloaded on ctxsw in).
1155                  * The upper part stays in the soft-counter.
1156                  */
1157                 if (PMD_IS_COUNTING(i)) {
1158                         ctx->ctx_pmds[i].val = val & ~ovfl_val;
1159                          val &= ovfl_val;
1160                 }
1161                 ctx->th_pmds[i] = val;
1162
1163                 DPRINT(("pmd[%d]=0x%lx soft_val=0x%lx\n",
1164                         i,
1165                         ctx->th_pmds[i],
1166                         ctx->ctx_pmds[i].val));
1167         }
1168 }
1169
1170 /*
1171  * propagate PMC from context to thread-state
1172  */
1173 static inline void
1174 pfm_copy_pmcs(struct task_struct *task, pfm_context_t *ctx)
1175 {
1176         unsigned long mask = ctx->ctx_all_pmcs[0];
1177         int i;
1178
1179         DPRINT(("mask=0x%lx\n", mask));
1180
1181         for (i=0; mask; i++, mask>>=1) {
1182                 /* masking 0 with ovfl_val yields 0 */
1183                 ctx->th_pmcs[i] = ctx->ctx_pmcs[i];
1184                 DPRINT(("pmc[%d]=0x%lx\n", i, ctx->th_pmcs[i]));
1185         }
1186 }
1187
1188
1189
1190 static inline void
1191 pfm_restore_pmcs(unsigned long *pmcs, unsigned long mask)
1192 {
1193         int i;
1194
1195         for (i=0; mask; i++, mask>>=1) {
1196                 if ((mask & 0x1) == 0) continue;
1197                 ia64_set_pmc(i, pmcs[i]);
1198         }
1199         ia64_srlz_d();
1200 }
1201
1202 static inline int
1203 pfm_uuid_cmp(pfm_uuid_t a, pfm_uuid_t b)
1204 {
1205         return memcmp(a, b, sizeof(pfm_uuid_t));
1206 }
1207
1208 static inline int
1209 pfm_buf_fmt_exit(pfm_buffer_fmt_t *fmt, struct task_struct *task, void *buf, struct pt_regs *regs)
1210 {
1211         int ret = 0;
1212         if (fmt->fmt_exit) ret = (*fmt->fmt_exit)(task, buf, regs);
1213         return ret;
1214 }
1215
1216 static inline int
1217 pfm_buf_fmt_getsize(pfm_buffer_fmt_t *fmt, struct task_struct *task, unsigned int flags, int cpu, void *arg, unsigned long *size)
1218 {
1219         int ret = 0;
1220         if (fmt->fmt_getsize) ret = (*fmt->fmt_getsize)(task, flags, cpu, arg, size);
1221         return ret;
1222 }
1223
1224
1225 static inline int
1226 pfm_buf_fmt_validate(pfm_buffer_fmt_t *fmt, struct task_struct *task, unsigned int flags,
1227                      int cpu, void *arg)
1228 {
1229         int ret = 0;
1230         if (fmt->fmt_validate) ret = (*fmt->fmt_validate)(task, flags, cpu, arg);
1231         return ret;
1232 }
1233
1234 static inline int
1235 pfm_buf_fmt_init(pfm_buffer_fmt_t *fmt, struct task_struct *task, void *buf, unsigned int flags,
1236                      int cpu, void *arg)
1237 {
1238         int ret = 0;
1239         if (fmt->fmt_init) ret = (*fmt->fmt_init)(task, buf, flags, cpu, arg);
1240         return ret;
1241 }
1242
1243 static inline int
1244 pfm_buf_fmt_restart(pfm_buffer_fmt_t *fmt, struct task_struct *task, pfm_ovfl_ctrl_t *ctrl, void *buf, struct pt_regs *regs)
1245 {
1246         int ret = 0;
1247         if (fmt->fmt_restart) ret = (*fmt->fmt_restart)(task, ctrl, buf, regs);
1248         return ret;
1249 }
1250
1251 static inline int
1252 pfm_buf_fmt_restart_active(pfm_buffer_fmt_t *fmt, struct task_struct *task, pfm_ovfl_ctrl_t *ctrl, void *buf, struct pt_regs *regs)
1253 {
1254         int ret = 0;
1255         if (fmt->fmt_restart_active) ret = (*fmt->fmt_restart_active)(task, ctrl, buf, regs);
1256         return ret;
1257 }
1258
1259 static pfm_buffer_fmt_t *
1260 __pfm_find_buffer_fmt(pfm_uuid_t uuid)
1261 {
1262         struct list_head * pos;
1263         pfm_buffer_fmt_t * entry;
1264
1265         list_for_each(pos, &pfm_buffer_fmt_list) {
1266                 entry = list_entry(pos, pfm_buffer_fmt_t, fmt_list);
1267                 if (pfm_uuid_cmp(uuid, entry->fmt_uuid) == 0)
1268                         return entry;
1269         }
1270         return NULL;
1271 }
1272  
1273 /*
1274  * find a buffer format based on its uuid
1275  */
1276 static pfm_buffer_fmt_t *
1277 pfm_find_buffer_fmt(pfm_uuid_t uuid)
1278 {
1279         pfm_buffer_fmt_t * fmt;
1280         spin_lock(&pfm_buffer_fmt_lock);
1281         fmt = __pfm_find_buffer_fmt(uuid);
1282         spin_unlock(&pfm_buffer_fmt_lock);
1283         return fmt;
1284 }
1285  
1286 int
1287 pfm_register_buffer_fmt(pfm_buffer_fmt_t *fmt)
1288 {
1289         int ret = 0;
1290
1291         /* some sanity checks */
1292         if (fmt == NULL || fmt->fmt_name == NULL) return -EINVAL;
1293
1294         /* we need at least a handler */
1295         if (fmt->fmt_handler == NULL) return -EINVAL;
1296
1297         /*
1298          * XXX: need check validity of fmt_arg_size
1299          */
1300
1301         spin_lock(&pfm_buffer_fmt_lock);
1302
1303         if (__pfm_find_buffer_fmt(fmt->fmt_uuid)) {
1304                 printk(KERN_ERR "perfmon: duplicate sampling format: %s\n", fmt->fmt_name);
1305                 ret = -EBUSY;
1306                 goto out;
1307         } 
1308         list_add(&fmt->fmt_list, &pfm_buffer_fmt_list);
1309         printk(KERN_INFO "perfmon: added sampling format %s\n", fmt->fmt_name);
1310
1311 out:
1312         spin_unlock(&pfm_buffer_fmt_lock);
1313         return ret;
1314 }
1315 EXPORT_SYMBOL(pfm_register_buffer_fmt);
1316
1317 int
1318 pfm_unregister_buffer_fmt(pfm_uuid_t uuid)
1319 {
1320         pfm_buffer_fmt_t *fmt;
1321         int ret = 0;
1322
1323         spin_lock(&pfm_buffer_fmt_lock);
1324
1325         fmt = __pfm_find_buffer_fmt(uuid);
1326         if (!fmt) {
1327                 printk(KERN_ERR "perfmon: cannot unregister format, not found\n");
1328                 ret = -EINVAL;
1329                 goto out;
1330         }
1331         list_del_init(&fmt->fmt_list);
1332         printk(KERN_INFO "perfmon: removed sampling format: %s\n", fmt->fmt_name);
1333
1334 out:
1335         spin_unlock(&pfm_buffer_fmt_lock);
1336         return ret;
1337
1338 }
1339 EXPORT_SYMBOL(pfm_unregister_buffer_fmt);
1340
1341 extern void update_pal_halt_status(int);
1342
1343 static int
1344 pfm_reserve_session(struct task_struct *task, int is_syswide, unsigned int cpu)
1345 {
1346         unsigned long flags;
1347         /*
1348          * validity checks on cpu_mask have been done upstream
1349          */
1350         LOCK_PFS(flags);
1351
1352         DPRINT(("in sys_sessions=%u task_sessions=%u dbregs=%u syswide=%d cpu=%u\n",
1353                 pfm_sessions.pfs_sys_sessions,
1354                 pfm_sessions.pfs_task_sessions,
1355                 pfm_sessions.pfs_sys_use_dbregs,
1356                 is_syswide,
1357                 cpu));
1358
1359         if (is_syswide) {
1360                 /*
1361                  * cannot mix system wide and per-task sessions
1362                  */
1363                 if (pfm_sessions.pfs_task_sessions > 0UL) {
1364                         DPRINT(("system wide not possible, %u conflicting task_sessions\n",
1365                                 pfm_sessions.pfs_task_sessions));
1366                         goto abort;
1367                 }
1368
1369                 if (pfm_sessions.pfs_sys_session[cpu]) goto error_conflict;
1370
1371                 DPRINT(("reserving system wide session on CPU%u currently on CPU%u\n", cpu, smp_processor_id()));
1372
1373                 pfm_sessions.pfs_sys_session[cpu] = task;
1374
1375                 pfm_sessions.pfs_sys_sessions++ ;
1376
1377         } else {
1378                 if (pfm_sessions.pfs_sys_sessions) goto abort;
1379                 pfm_sessions.pfs_task_sessions++;
1380         }
1381
1382         DPRINT(("out sys_sessions=%u task_sessions=%u dbregs=%u syswide=%d cpu=%u\n",
1383                 pfm_sessions.pfs_sys_sessions,
1384                 pfm_sessions.pfs_task_sessions,
1385                 pfm_sessions.pfs_sys_use_dbregs,
1386                 is_syswide,
1387                 cpu));
1388
1389         /*
1390          * disable default_idle() to go to PAL_HALT
1391          */
1392         update_pal_halt_status(0);
1393
1394         UNLOCK_PFS(flags);
1395
1396         return 0;
1397
1398 error_conflict:
1399         DPRINT(("system wide not possible, conflicting session [%d] on CPU%d\n",
1400                 task_pid_nr(pfm_sessions.pfs_sys_session[cpu]),
1401                 cpu));
1402 abort:
1403         UNLOCK_PFS(flags);
1404
1405         return -EBUSY;
1406
1407 }
1408
1409 static int
1410 pfm_unreserve_session(pfm_context_t *ctx, int is_syswide, unsigned int cpu)
1411 {
1412         unsigned long flags;
1413         /*
1414          * validity checks on cpu_mask have been done upstream
1415          */
1416         LOCK_PFS(flags);
1417
1418         DPRINT(("in sys_sessions=%u task_sessions=%u dbregs=%u syswide=%d cpu=%u\n",
1419                 pfm_sessions.pfs_sys_sessions,
1420                 pfm_sessions.pfs_task_sessions,
1421                 pfm_sessions.pfs_sys_use_dbregs,
1422                 is_syswide,
1423                 cpu));
1424
1425
1426         if (is_syswide) {
1427                 pfm_sessions.pfs_sys_session[cpu] = NULL;
1428                 /*
1429                  * would not work with perfmon+more than one bit in cpu_mask
1430                  */
1431                 if (ctx && ctx->ctx_fl_using_dbreg) {
1432                         if (pfm_sessions.pfs_sys_use_dbregs == 0) {
1433                                 printk(KERN_ERR "perfmon: invalid release for ctx %p sys_use_dbregs=0\n", ctx);
1434                         } else {
1435                                 pfm_sessions.pfs_sys_use_dbregs--;
1436                         }
1437                 }
1438                 pfm_sessions.pfs_sys_sessions--;
1439         } else {
1440                 pfm_sessions.pfs_task_sessions--;
1441         }
1442         DPRINT(("out sys_sessions=%u task_sessions=%u dbregs=%u syswide=%d cpu=%u\n",
1443                 pfm_sessions.pfs_sys_sessions,
1444                 pfm_sessions.pfs_task_sessions,
1445                 pfm_sessions.pfs_sys_use_dbregs,
1446                 is_syswide,
1447                 cpu));
1448
1449         /*
1450          * if possible, enable default_idle() to go into PAL_HALT
1451          */
1452         if (pfm_sessions.pfs_task_sessions == 0 && pfm_sessions.pfs_sys_sessions == 0)
1453                 update_pal_halt_status(1);
1454
1455         UNLOCK_PFS(flags);
1456
1457         return 0;
1458 }
1459
1460 /*
1461  * removes virtual mapping of the sampling buffer.
1462  * IMPORTANT: cannot be called with interrupts disable, e.g. inside
1463  * a PROTECT_CTX() section.
1464  */
1465 static int
1466 pfm_remove_smpl_mapping(struct task_struct *task, void *vaddr, unsigned long size)
1467 {
1468         int r;
1469
1470         /* sanity checks */
1471         if (task->mm == NULL || size == 0UL || vaddr == NULL) {
1472                 printk(KERN_ERR "perfmon: pfm_remove_smpl_mapping [%d] invalid context mm=%p\n", task_pid_nr(task), task->mm);
1473                 return -EINVAL;
1474         }
1475
1476         DPRINT(("smpl_vaddr=%p size=%lu\n", vaddr, size));
1477
1478         /*
1479          * does the actual unmapping
1480          */
1481         down_write(&task->mm->mmap_sem);
1482
1483         DPRINT(("down_write done smpl_vaddr=%p size=%lu\n", vaddr, size));
1484
1485         r = pfm_do_munmap(task->mm, (unsigned long)vaddr, size, 0);
1486
1487         up_write(&task->mm->mmap_sem);
1488         if (r !=0) {
1489                 printk(KERN_ERR "perfmon: [%d] unable to unmap sampling buffer @%p size=%lu\n", task_pid_nr(task), vaddr, size);
1490         }
1491
1492         DPRINT(("do_unmap(%p, %lu)=%d\n", vaddr, size, r));
1493
1494         return 0;
1495 }
1496
1497 /*
1498  * free actual physical storage used by sampling buffer
1499  */
1500 #if 0
1501 static int
1502 pfm_free_smpl_buffer(pfm_context_t *ctx)
1503 {
1504         pfm_buffer_fmt_t *fmt;
1505
1506         if (ctx->ctx_smpl_hdr == NULL) goto invalid_free;
1507
1508         /*
1509          * we won't use the buffer format anymore
1510          */
1511         fmt = ctx->ctx_buf_fmt;
1512
1513         DPRINT(("sampling buffer @%p size %lu vaddr=%p\n",
1514                 ctx->ctx_smpl_hdr,
1515                 ctx->ctx_smpl_size,
1516                 ctx->ctx_smpl_vaddr));
1517
1518         pfm_buf_fmt_exit(fmt, current, NULL, NULL);
1519
1520         /*
1521          * free the buffer
1522          */
1523         pfm_rvfree(ctx->ctx_smpl_hdr, ctx->ctx_smpl_size);
1524
1525         ctx->ctx_smpl_hdr  = NULL;
1526         ctx->ctx_smpl_size = 0UL;
1527
1528         return 0;
1529
1530 invalid_free:
1531         printk(KERN_ERR "perfmon: pfm_free_smpl_buffer [%d] no buffer\n", task_pid_nr(current));
1532         return -EINVAL;
1533 }
1534 #endif
1535
1536 static inline void
1537 pfm_exit_smpl_buffer(pfm_buffer_fmt_t *fmt)
1538 {
1539         if (fmt == NULL) return;
1540
1541         pfm_buf_fmt_exit(fmt, current, NULL, NULL);
1542
1543 }
1544
1545 /*
1546  * pfmfs should _never_ be mounted by userland - too much of security hassle,
1547  * no real gain from having the whole whorehouse mounted. So we don't need
1548  * any operations on the root directory. However, we need a non-trivial
1549  * d_name - pfm: will go nicely and kill the special-casing in procfs.
1550  */
1551 static struct vfsmount *pfmfs_mnt;
1552
1553 static int __init
1554 init_pfm_fs(void)
1555 {
1556         int err = register_filesystem(&pfm_fs_type);
1557         if (!err) {
1558                 pfmfs_mnt = kern_mount(&pfm_fs_type);
1559                 err = PTR_ERR(pfmfs_mnt);
1560                 if (IS_ERR(pfmfs_mnt))
1561                         unregister_filesystem(&pfm_fs_type);
1562                 else
1563                         err = 0;
1564         }
1565         return err;
1566 }
1567
1568 static ssize_t
1569 pfm_read(struct file *filp, char __user *buf, size_t size, loff_t *ppos)
1570 {
1571         pfm_context_t *ctx;
1572         pfm_msg_t *msg;
1573         ssize_t ret;
1574         unsigned long flags;
1575         DECLARE_WAITQUEUE(wait, current);
1576         if (PFM_IS_FILE(filp) == 0) {
1577                 printk(KERN_ERR "perfmon: pfm_poll: bad magic [%d]\n", task_pid_nr(current));
1578                 return -EINVAL;
1579         }
1580
1581         ctx = (pfm_context_t *)filp->private_data;
1582         if (ctx == NULL) {
1583                 printk(KERN_ERR "perfmon: pfm_read: NULL ctx [%d]\n", task_pid_nr(current));
1584                 return -EINVAL;
1585         }
1586
1587         /*
1588          * check even when there is no message
1589          */
1590         if (size < sizeof(pfm_msg_t)) {
1591                 DPRINT(("message is too small ctx=%p (>=%ld)\n", ctx, sizeof(pfm_msg_t)));
1592                 return -EINVAL;
1593         }
1594
1595         PROTECT_CTX(ctx, flags);
1596
1597         /*
1598          * put ourselves on the wait queue
1599          */
1600         add_wait_queue(&ctx->ctx_msgq_wait, &wait);
1601
1602
1603         for(;;) {
1604                 /*
1605                  * check wait queue
1606                  */
1607
1608                 set_current_state(TASK_INTERRUPTIBLE);
1609
1610                 DPRINT(("head=%d tail=%d\n", ctx->ctx_msgq_head, ctx->ctx_msgq_tail));
1611
1612                 ret = 0;
1613                 if(PFM_CTXQ_EMPTY(ctx) == 0) break;
1614
1615                 UNPROTECT_CTX(ctx, flags);
1616
1617                 /*
1618                  * check non-blocking read
1619                  */
1620                 ret = -EAGAIN;
1621                 if(filp->f_flags & O_NONBLOCK) break;
1622
1623                 /*
1624                  * check pending signals
1625                  */
1626                 if(signal_pending(current)) {
1627                         ret = -EINTR;
1628                         break;
1629                 }
1630                 /*
1631                  * no message, so wait
1632                  */
1633                 schedule();
1634
1635                 PROTECT_CTX(ctx, flags);
1636         }
1637         DPRINT(("[%d] back to running ret=%ld\n", task_pid_nr(current), ret));
1638         set_current_state(TASK_RUNNING);
1639         remove_wait_queue(&ctx->ctx_msgq_wait, &wait);
1640
1641         if (ret < 0) goto abort;
1642
1643         ret = -EINVAL;
1644         msg = pfm_get_next_msg(ctx);
1645         if (msg == NULL) {
1646                 printk(KERN_ERR "perfmon: pfm_read no msg for ctx=%p [%d]\n", ctx, task_pid_nr(current));
1647                 goto abort_locked;
1648         }
1649
1650         DPRINT(("fd=%d type=%d\n", msg->pfm_gen_msg.msg_ctx_fd, msg->pfm_gen_msg.msg_type));
1651
1652         ret = -EFAULT;
1653         if(copy_to_user(buf, msg, sizeof(pfm_msg_t)) == 0) ret = sizeof(pfm_msg_t);
1654
1655 abort_locked:
1656         UNPROTECT_CTX(ctx, flags);
1657 abort:
1658         return ret;
1659 }
1660
1661 static ssize_t
1662 pfm_write(struct file *file, const char __user *ubuf,
1663                           size_t size, loff_t *ppos)
1664 {
1665         DPRINT(("pfm_write called\n"));
1666         return -EINVAL;
1667 }
1668
1669 static unsigned int
1670 pfm_poll(struct file *filp, poll_table * wait)
1671 {
1672         pfm_context_t *ctx;
1673         unsigned long flags;
1674         unsigned int mask = 0;
1675
1676         if (PFM_IS_FILE(filp) == 0) {
1677                 printk(KERN_ERR "perfmon: pfm_poll: bad magic [%d]\n", task_pid_nr(current));
1678                 return 0;
1679         }
1680
1681         ctx = (pfm_context_t *)filp->private_data;
1682         if (ctx == NULL) {
1683                 printk(KERN_ERR "perfmon: pfm_poll: NULL ctx [%d]\n", task_pid_nr(current));
1684                 return 0;
1685         }
1686
1687
1688         DPRINT(("pfm_poll ctx_fd=%d before poll_wait\n", ctx->ctx_fd));
1689
1690         poll_wait(filp, &ctx->ctx_msgq_wait, wait);
1691
1692         PROTECT_CTX(ctx, flags);
1693
1694         if (PFM_CTXQ_EMPTY(ctx) == 0)
1695                 mask =  POLLIN | POLLRDNORM;
1696
1697         UNPROTECT_CTX(ctx, flags);
1698
1699         DPRINT(("pfm_poll ctx_fd=%d mask=0x%x\n", ctx->ctx_fd, mask));
1700
1701         return mask;
1702 }
1703
1704 static int
1705 pfm_ioctl(struct inode *inode, struct file *file, unsigned int cmd, unsigned long arg)
1706 {
1707         DPRINT(("pfm_ioctl called\n"));
1708         return -EINVAL;
1709 }
1710
1711 /*
1712  * interrupt cannot be masked when coming here
1713  */
1714 static inline int
1715 pfm_do_fasync(int fd, struct file *filp, pfm_context_t *ctx, int on)
1716 {
1717         int ret;
1718
1719         ret = fasync_helper (fd, filp, on, &ctx->ctx_async_queue);
1720
1721         DPRINT(("pfm_fasync called by [%d] on ctx_fd=%d on=%d async_queue=%p ret=%d\n",
1722                 task_pid_nr(current),
1723                 fd,
1724                 on,
1725                 ctx->ctx_async_queue, ret));
1726
1727         return ret;
1728 }
1729
1730 static int
1731 pfm_fasync(int fd, struct file *filp, int on)
1732 {
1733         pfm_context_t *ctx;
1734         int ret;
1735
1736         if (PFM_IS_FILE(filp) == 0) {
1737                 printk(KERN_ERR "perfmon: pfm_fasync bad magic [%d]\n", task_pid_nr(current));
1738                 return -EBADF;
1739         }
1740
1741         ctx = (pfm_context_t *)filp->private_data;
1742         if (ctx == NULL) {
1743                 printk(KERN_ERR "perfmon: pfm_fasync NULL ctx [%d]\n", task_pid_nr(current));
1744                 return -EBADF;
1745         }
1746         /*
1747          * we cannot mask interrupts during this call because this may
1748          * may go to sleep if memory is not readily avalaible.
1749          *
1750          * We are protected from the conetxt disappearing by the get_fd()/put_fd()
1751          * done in caller. Serialization of this function is ensured by caller.
1752          */
1753         ret = pfm_do_fasync(fd, filp, ctx, on);
1754
1755
1756         DPRINT(("pfm_fasync called on ctx_fd=%d on=%d async_queue=%p ret=%d\n",
1757                 fd,
1758                 on,
1759                 ctx->ctx_async_queue, ret));
1760
1761         return ret;
1762 }
1763
1764 #ifdef CONFIG_SMP
1765 /*
1766  * this function is exclusively called from pfm_close().
1767  * The context is not protected at that time, nor are interrupts
1768  * on the remote CPU. That's necessary to avoid deadlocks.
1769  */
1770 static void
1771 pfm_syswide_force_stop(void *info)
1772 {
1773         pfm_context_t   *ctx = (pfm_context_t *)info;
1774         struct pt_regs *regs = task_pt_regs(current);
1775         struct task_struct *owner;
1776         unsigned long flags;
1777         int ret;
1778
1779         if (ctx->ctx_cpu != smp_processor_id()) {
1780                 printk(KERN_ERR "perfmon: pfm_syswide_force_stop for CPU%d  but on CPU%d\n",
1781                         ctx->ctx_cpu,
1782                         smp_processor_id());
1783                 return;
1784         }
1785         owner = GET_PMU_OWNER();
1786         if (owner != ctx->ctx_task) {
1787                 printk(KERN_ERR "perfmon: pfm_syswide_force_stop CPU%d unexpected owner [%d] instead of [%d]\n",
1788                         smp_processor_id(),
1789                         task_pid_nr(owner), task_pid_nr(ctx->ctx_task));
1790                 return;
1791         }
1792         if (GET_PMU_CTX() != ctx) {
1793                 printk(KERN_ERR "perfmon: pfm_syswide_force_stop CPU%d unexpected ctx %p instead of %p\n",
1794                         smp_processor_id(),
1795                         GET_PMU_CTX(), ctx);
1796                 return;
1797         }
1798
1799         DPRINT(("on CPU%d forcing system wide stop for [%d]\n", smp_processor_id(), task_pid_nr(ctx->ctx_task)));
1800         /*
1801          * the context is already protected in pfm_close(), we simply
1802          * need to mask interrupts to avoid a PMU interrupt race on
1803          * this CPU
1804          */
1805         local_irq_save(flags);
1806
1807         ret = pfm_context_unload(ctx, NULL, 0, regs);
1808         if (ret) {
1809                 DPRINT(("context_unload returned %d\n", ret));
1810         }
1811
1812         /*
1813          * unmask interrupts, PMU interrupts are now spurious here
1814          */
1815         local_irq_restore(flags);
1816 }
1817
1818 static void
1819 pfm_syswide_cleanup_other_cpu(pfm_context_t *ctx)
1820 {
1821         int ret;
1822
1823         DPRINT(("calling CPU%d for cleanup\n", ctx->ctx_cpu));
1824         ret = smp_call_function_single(ctx->ctx_cpu, pfm_syswide_force_stop, ctx, 1);
1825         DPRINT(("called CPU%d for cleanup ret=%d\n", ctx->ctx_cpu, ret));
1826 }
1827 #endif /* CONFIG_SMP */
1828
1829 /*
1830  * called for each close(). Partially free resources.
1831  * When caller is self-monitoring, the context is unloaded.
1832  */
1833 static int
1834 pfm_flush(struct file *filp, fl_owner_t id)
1835 {
1836         pfm_context_t *ctx;
1837         struct task_struct *task;
1838         struct pt_regs *regs;
1839         unsigned long flags;
1840         unsigned long smpl_buf_size = 0UL;
1841         void *smpl_buf_vaddr = NULL;
1842         int state, is_system;
1843
1844         if (PFM_IS_FILE(filp) == 0) {
1845                 DPRINT(("bad magic for\n"));
1846                 return -EBADF;
1847         }
1848
1849         ctx = (pfm_context_t *)filp->private_data;
1850         if (ctx == NULL) {
1851                 printk(KERN_ERR "perfmon: pfm_flush: NULL ctx [%d]\n", task_pid_nr(current));
1852                 return -EBADF;
1853         }
1854
1855         /*
1856          * remove our file from the async queue, if we use this mode.
1857          * This can be done without the context being protected. We come
1858          * here when the context has become unreachable by other tasks.
1859          *
1860          * We may still have active monitoring at this point and we may
1861          * end up in pfm_overflow_handler(). However, fasync_helper()
1862          * operates with interrupts disabled and it cleans up the
1863          * queue. If the PMU handler is called prior to entering
1864          * fasync_helper() then it will send a signal. If it is
1865          * invoked after, it will find an empty queue and no
1866          * signal will be sent. In both case, we are safe
1867          */
1868         PROTECT_CTX(ctx, flags);
1869
1870         state     = ctx->ctx_state;
1871         is_system = ctx->ctx_fl_system;
1872
1873         task = PFM_CTX_TASK(ctx);
1874         regs = task_pt_regs(task);
1875
1876         DPRINT(("ctx_state=%d is_current=%d\n",
1877                 state,
1878                 task == current ? 1 : 0));
1879
1880         /*
1881          * if state == UNLOADED, then task is NULL
1882          */
1883
1884         /*
1885          * we must stop and unload because we are losing access to the context.
1886          */
1887         if (task == current) {
1888 #ifdef CONFIG_SMP
1889                 /*
1890                  * the task IS the owner but it migrated to another CPU: that's bad
1891                  * but we must handle this cleanly. Unfortunately, the kernel does
1892                  * not provide a mechanism to block migration (while the context is loaded).
1893                  *
1894                  * We need to release the resource on the ORIGINAL cpu.
1895                  */
1896                 if (is_system && ctx->ctx_cpu != smp_processor_id()) {
1897
1898                         DPRINT(("should be running on CPU%d\n", ctx->ctx_cpu));
1899                         /*
1900                          * keep context protected but unmask interrupt for IPI
1901                          */
1902                         local_irq_restore(flags);
1903
1904                         pfm_syswide_cleanup_other_cpu(ctx);
1905
1906                         /*
1907                          * restore interrupt masking
1908                          */
1909                         local_irq_save(flags);
1910
1911                         /*
1912                          * context is unloaded at this point
1913                          */
1914                 } else
1915 #endif /* CONFIG_SMP */
1916                 {
1917
1918                         DPRINT(("forcing unload\n"));
1919                         /*
1920                         * stop and unload, returning with state UNLOADED
1921                         * and session unreserved.
1922                         */
1923                         pfm_context_unload(ctx, NULL, 0, regs);
1924
1925                         DPRINT(("ctx_state=%d\n", ctx->ctx_state));
1926                 }
1927         }
1928
1929         /*
1930          * remove virtual mapping, if any, for the calling task.
1931          * cannot reset ctx field until last user is calling close().
1932          *
1933          * ctx_smpl_vaddr must never be cleared because it is needed
1934          * by every task with access to the context
1935          *
1936          * When called from do_exit(), the mm context is gone already, therefore
1937          * mm is NULL, i.e., the VMA is already gone  and we do not have to
1938          * do anything here
1939          */
1940         if (ctx->ctx_smpl_vaddr && current->mm) {
1941                 smpl_buf_vaddr = ctx->ctx_smpl_vaddr;
1942                 smpl_buf_size  = ctx->ctx_smpl_size;
1943         }
1944
1945         UNPROTECT_CTX(ctx, flags);
1946
1947         /*
1948          * if there was a mapping, then we systematically remove it
1949          * at this point. Cannot be done inside critical section
1950          * because some VM function reenables interrupts.
1951          *
1952          */
1953         if (smpl_buf_vaddr) pfm_remove_smpl_mapping(current, smpl_buf_vaddr, smpl_buf_size);
1954
1955         return 0;
1956 }
1957 /*
1958  * called either on explicit close() or from exit_files(). 
1959  * Only the LAST user of the file gets to this point, i.e., it is
1960  * called only ONCE.
1961  *
1962  * IMPORTANT: we get called ONLY when the refcnt on the file gets to zero 
1963  * (fput()),i.e, last task to access the file. Nobody else can access the 
1964  * file at this point.
1965  *
1966  * When called from exit_files(), the VMA has been freed because exit_mm()
1967  * is executed before exit_files().
1968  *
1969  * When called from exit_files(), the current task is not yet ZOMBIE but we
1970  * flush the PMU state to the context. 
1971  */
1972 static int
1973 pfm_close(struct inode *inode, struct file *filp)
1974 {
1975         pfm_context_t *ctx;
1976         struct task_struct *task;
1977         struct pt_regs *regs;
1978         DECLARE_WAITQUEUE(wait, current);
1979         unsigned long flags;
1980         unsigned long smpl_buf_size = 0UL;
1981         void *smpl_buf_addr = NULL;
1982         int free_possible = 1;
1983         int state, is_system;
1984
1985         DPRINT(("pfm_close called private=%p\n", filp->private_data));
1986
1987         if (PFM_IS_FILE(filp) == 0) {
1988                 DPRINT(("bad magic\n"));
1989                 return -EBADF;
1990         }
1991         
1992         ctx = (pfm_context_t *)filp->private_data;
1993         if (ctx == NULL) {
1994                 printk(KERN_ERR "perfmon: pfm_close: NULL ctx [%d]\n", task_pid_nr(current));
1995                 return -EBADF;
1996         }
1997
1998         PROTECT_CTX(ctx, flags);
1999
2000         state     = ctx->ctx_state;
2001         is_system = ctx->ctx_fl_system;
2002
2003         task = PFM_CTX_TASK(ctx);
2004         regs = task_pt_regs(task);
2005
2006         DPRINT(("ctx_state=%d is_current=%d\n", 
2007                 state,
2008                 task == current ? 1 : 0));
2009
2010         /*
2011          * if task == current, then pfm_flush() unloaded the context
2012          */
2013         if (state == PFM_CTX_UNLOADED) goto doit;
2014
2015         /*
2016          * context is loaded/masked and task != current, we need to
2017          * either force an unload or go zombie
2018          */
2019
2020         /*
2021          * The task is currently blocked or will block after an overflow.
2022          * we must force it to wakeup to get out of the
2023          * MASKED state and transition to the unloaded state by itself.
2024          *
2025          * This situation is only possible for per-task mode
2026          */
2027         if (state == PFM_CTX_MASKED && CTX_OVFL_NOBLOCK(ctx) == 0) {
2028
2029                 /*
2030                  * set a "partial" zombie state to be checked
2031                  * upon return from down() in pfm_handle_work().
2032                  *
2033                  * We cannot use the ZOMBIE state, because it is checked
2034                  * by pfm_load_regs() which is called upon wakeup from down().
2035                  * In such case, it would free the context and then we would
2036                  * return to pfm_handle_work() which would access the
2037                  * stale context. Instead, we set a flag invisible to pfm_load_regs()
2038                  * but visible to pfm_handle_work().
2039                  *
2040                  * For some window of time, we have a zombie context with
2041                  * ctx_state = MASKED  and not ZOMBIE
2042                  */
2043                 ctx->ctx_fl_going_zombie = 1;
2044
2045                 /*
2046                  * force task to wake up from MASKED state
2047                  */
2048                 complete(&ctx->ctx_restart_done);
2049
2050                 DPRINT(("waking up ctx_state=%d\n", state));
2051
2052                 /*
2053                  * put ourself to sleep waiting for the other
2054                  * task to report completion
2055                  *
2056                  * the context is protected by mutex, therefore there
2057                  * is no risk of being notified of completion before
2058                  * begin actually on the waitq.
2059                  */
2060                 set_current_state(TASK_INTERRUPTIBLE);
2061                 add_wait_queue(&ctx->ctx_zombieq, &wait);
2062
2063                 UNPROTECT_CTX(ctx, flags);
2064
2065                 /*
2066                  * XXX: check for signals :
2067                  *      - ok for explicit close
2068                  *      - not ok when coming from exit_files()
2069                  */
2070                 schedule();
2071
2072
2073                 PROTECT_CTX(ctx, flags);
2074
2075
2076                 remove_wait_queue(&ctx->ctx_zombieq, &wait);
2077                 set_current_state(TASK_RUNNING);
2078
2079                 /*
2080                  * context is unloaded at this point
2081                  */
2082                 DPRINT(("after zombie wakeup ctx_state=%d for\n", state));
2083         }
2084         else if (task != current) {
2085 #ifdef CONFIG_SMP
2086                 /*
2087                  * switch context to zombie state
2088                  */
2089                 ctx->ctx_state = PFM_CTX_ZOMBIE;
2090
2091                 DPRINT(("zombie ctx for [%d]\n", task_pid_nr(task)));
2092                 /*
2093                  * cannot free the context on the spot. deferred until
2094                  * the task notices the ZOMBIE state
2095                  */
2096                 free_possible = 0;
2097 #else
2098                 pfm_context_unload(ctx, NULL, 0, regs);
2099 #endif
2100         }
2101
2102 doit:
2103         /* reload state, may have changed during  opening of critical section */
2104         state = ctx->ctx_state;
2105
2106         /*
2107          * the context is still attached to a task (possibly current)
2108          * we cannot destroy it right now
2109          */
2110
2111         /*
2112          * we must free the sampling buffer right here because
2113          * we cannot rely on it being cleaned up later by the
2114          * monitored task. It is not possible to free vmalloc'ed
2115          * memory in pfm_load_regs(). Instead, we remove the buffer
2116          * now. should there be subsequent PMU overflow originally
2117          * meant for sampling, the will be converted to spurious
2118          * and that's fine because the monitoring tools is gone anyway.
2119          */
2120         if (ctx->ctx_smpl_hdr) {
2121                 smpl_buf_addr = ctx->ctx_smpl_hdr;
2122                 smpl_buf_size = ctx->ctx_smpl_size;
2123                 /* no more sampling */
2124                 ctx->ctx_smpl_hdr = NULL;
2125                 ctx->ctx_fl_is_sampling = 0;
2126         }
2127
2128         DPRINT(("ctx_state=%d free_possible=%d addr=%p size=%lu\n",
2129                 state,
2130                 free_possible,
2131                 smpl_buf_addr,
2132                 smpl_buf_size));
2133
2134         if (smpl_buf_addr) pfm_exit_smpl_buffer(ctx->ctx_buf_fmt);
2135
2136         /*
2137          * UNLOADED that the session has already been unreserved.
2138          */
2139         if (state == PFM_CTX_ZOMBIE) {
2140                 pfm_unreserve_session(ctx, ctx->ctx_fl_system , ctx->ctx_cpu);
2141         }
2142
2143         /*
2144          * disconnect file descriptor from context must be done
2145          * before we unlock.
2146          */
2147         filp->private_data = NULL;
2148
2149         /*
2150          * if we free on the spot, the context is now completely unreachable
2151          * from the callers side. The monitored task side is also cut, so we
2152          * can freely cut.
2153          *
2154          * If we have a deferred free, only the caller side is disconnected.
2155          */
2156         UNPROTECT_CTX(ctx, flags);
2157
2158         /*
2159          * All memory free operations (especially for vmalloc'ed memory)
2160          * MUST be done with interrupts ENABLED.
2161          */
2162         if (smpl_buf_addr)  pfm_rvfree(smpl_buf_addr, smpl_buf_size);
2163
2164         /*
2165          * return the memory used by the context
2166          */
2167         if (free_possible) pfm_context_free(ctx);
2168
2169         return 0;
2170 }
2171
2172 static int
2173 pfm_no_open(struct inode *irrelevant, struct file *dontcare)
2174 {
2175         DPRINT(("pfm_no_open called\n"));
2176         return -ENXIO;
2177 }
2178
2179
2180
2181 static const struct file_operations pfm_file_ops = {
2182         .llseek   = no_llseek,
2183         .read     = pfm_read,
2184         .write    = pfm_write,
2185         .poll     = pfm_poll,
2186         .ioctl    = pfm_ioctl,
2187         .open     = pfm_no_open,        /* special open code to disallow open via /proc */
2188         .fasync   = pfm_fasync,
2189         .release  = pfm_close,
2190         .flush    = pfm_flush
2191 };
2192
2193 static int
2194 pfmfs_delete_dentry(struct dentry *dentry)
2195 {
2196         return 1;
2197 }
2198
2199 static const struct dentry_operations pfmfs_dentry_operations = {
2200         .d_delete = pfmfs_delete_dentry,
2201 };
2202
2203
2204 static struct file *
2205 pfm_alloc_file(pfm_context_t *ctx)
2206 {
2207         struct file *file;
2208         struct inode *inode;
2209         struct dentry *dentry;
2210         char name[32];
2211         struct qstr this;
2212
2213         /*
2214          * allocate a new inode
2215          */
2216         inode = new_inode(pfmfs_mnt->mnt_sb);
2217         if (!inode)
2218                 return ERR_PTR(-ENOMEM);
2219
2220         DPRINT(("new inode ino=%ld @%p\n", inode->i_ino, inode));
2221
2222         inode->i_mode = S_IFCHR|S_IRUGO;
2223         inode->i_uid  = current_fsuid();
2224         inode->i_gid  = current_fsgid();
2225
2226         sprintf(name, "[%lu]", inode->i_ino);
2227         this.name = name;
2228         this.len  = strlen(name);
2229         this.hash = inode->i_ino;
2230
2231         /*
2232          * allocate a new dcache entry
2233          */
2234         dentry = d_alloc(pfmfs_mnt->mnt_sb->s_root, &this);
2235         if (!dentry) {
2236                 iput(inode);
2237                 return ERR_PTR(-ENOMEM);
2238         }
2239
2240         dentry->d_op = &pfmfs_dentry_operations;
2241         d_add(dentry, inode);
2242
2243         file = alloc_file(pfmfs_mnt, dentry, FMODE_READ, &pfm_file_ops);
2244         if (!file) {
2245                 dput(dentry);
2246                 return ERR_PTR(-ENFILE);
2247         }
2248
2249         file->f_flags = O_RDONLY;
2250         file->private_data = ctx;
2251
2252         return file;
2253 }
2254
2255 static int
2256 pfm_remap_buffer(struct vm_area_struct *vma, unsigned long buf, unsigned long addr, unsigned long size)
2257 {
2258         DPRINT(("CPU%d buf=0x%lx addr=0x%lx size=%ld\n", smp_processor_id(), buf, addr, size));
2259
2260         while (size > 0) {
2261                 unsigned long pfn = ia64_tpa(buf) >> PAGE_SHIFT;
2262
2263
2264                 if (remap_pfn_range(vma, addr, pfn, PAGE_SIZE, PAGE_READONLY))
2265                         return -ENOMEM;
2266
2267                 addr  += PAGE_SIZE;
2268                 buf   += PAGE_SIZE;
2269                 size  -= PAGE_SIZE;
2270         }
2271         return 0;
2272 }
2273
2274 /*
2275  * allocate a sampling buffer and remaps it into the user address space of the task
2276  */
2277 static int
2278 pfm_smpl_buffer_alloc(struct task_struct *task, struct file *filp, pfm_context_t *ctx, unsigned long rsize, void **user_vaddr)
2279 {
2280         struct mm_struct *mm = task->mm;
2281         struct vm_area_struct *vma = NULL;
2282         unsigned long size;
2283         void *smpl_buf;
2284
2285
2286         /*
2287          * the fixed header + requested size and align to page boundary
2288          */
2289         size = PAGE_ALIGN(rsize);
2290
2291         DPRINT(("sampling buffer rsize=%lu size=%lu bytes\n", rsize, size));
2292
2293         /*
2294          * check requested size to avoid Denial-of-service attacks
2295          * XXX: may have to refine this test
2296          * Check against address space limit.
2297          *
2298          * if ((mm->total_vm << PAGE_SHIFT) + len> task->rlim[RLIMIT_AS].rlim_cur)
2299          *      return -ENOMEM;
2300          */
2301         if (size > task->signal->rlim[RLIMIT_MEMLOCK].rlim_cur)
2302                 return -ENOMEM;
2303
2304         /*
2305          * We do the easy to undo allocations first.
2306          *
2307          * pfm_rvmalloc(), clears the buffer, so there is no leak
2308          */
2309         smpl_buf = pfm_rvmalloc(size);
2310         if (smpl_buf == NULL) {
2311                 DPRINT(("Can't allocate sampling buffer\n"));
2312                 return -ENOMEM;
2313         }
2314
2315         DPRINT(("smpl_buf @%p\n", smpl_buf));
2316
2317         /* allocate vma */
2318         vma = kmem_cache_zalloc(vm_area_cachep, GFP_KERNEL);
2319         if (!vma) {
2320                 DPRINT(("Cannot allocate vma\n"));
2321                 goto error_kmem;
2322         }
2323
2324         /*
2325          * partially initialize the vma for the sampling buffer
2326          */
2327         vma->vm_mm           = mm;
2328         vma->vm_file         = filp;
2329         vma->vm_flags        = VM_READ| VM_MAYREAD |VM_RESERVED;
2330         vma->vm_page_prot    = PAGE_READONLY; /* XXX may need to change */
2331
2332         /*
2333          * Now we have everything we need and we can initialize
2334          * and connect all the data structures
2335          */
2336
2337         ctx->ctx_smpl_hdr   = smpl_buf;
2338         ctx->ctx_smpl_size  = size; /* aligned size */
2339
2340         /*
2341          * Let's do the difficult operations next.
2342          *
2343          * now we atomically find some area in the address space and
2344          * remap the buffer in it.
2345          */
2346         down_write(&task->mm->mmap_sem);
2347
2348         /* find some free area in address space, must have mmap sem held */
2349         vma->vm_start = pfm_get_unmapped_area(NULL, 0, size, 0, MAP_PRIVATE|MAP_ANONYMOUS, 0);
2350         if (vma->vm_start == 0UL) {
2351                 DPRINT(("Cannot find unmapped area for size %ld\n", size));
2352                 up_write(&task->mm->mmap_sem);
2353                 goto error;
2354         }
2355         vma->vm_end = vma->vm_start + size;
2356         vma->vm_pgoff = vma->vm_start >> PAGE_SHIFT;
2357
2358         DPRINT(("aligned size=%ld, hdr=%p mapped @0x%lx\n", size, ctx->ctx_smpl_hdr, vma->vm_start));
2359
2360         /* can only be applied to current task, need to have the mm semaphore held when called */
2361         if (pfm_remap_buffer(vma, (unsigned long)smpl_buf, vma->vm_start, size)) {
2362                 DPRINT(("Can't remap buffer\n"));
2363                 up_write(&task->mm->mmap_sem);
2364                 goto error;
2365         }
2366
2367         get_file(filp);
2368
2369         /*
2370          * now insert the vma in the vm list for the process, must be
2371          * done with mmap lock held
2372          */
2373         insert_vm_struct(mm, vma);
2374
2375         mm->total_vm  += size >> PAGE_SHIFT;
2376         vm_stat_account(vma->vm_mm, vma->vm_flags, vma->vm_file,
2377                                                         vma_pages(vma));
2378         up_write(&task->mm->mmap_sem);
2379
2380         /*
2381          * keep track of user level virtual address
2382          */
2383         ctx->ctx_smpl_vaddr = (void *)vma->vm_start;
2384         *(unsigned long *)user_vaddr = vma->vm_start;
2385
2386         return 0;
2387
2388 error:
2389         kmem_cache_free(vm_area_cachep, vma);
2390 error_kmem:
2391         pfm_rvfree(smpl_buf, size);
2392
2393         return -ENOMEM;
2394 }
2395
2396 /*
2397  * XXX: do something better here
2398  */
2399 static int
2400 pfm_bad_permissions(struct task_struct *task)
2401 {
2402         const struct cred *tcred;
2403         uid_t uid = current_uid();
2404         gid_t gid = current_gid();
2405         int ret;
2406
2407         rcu_read_lock();
2408         tcred = __task_cred(task);
2409
2410         /* inspired by ptrace_attach() */
2411         DPRINT(("cur: uid=%d gid=%d task: euid=%d suid=%d uid=%d egid=%d sgid=%d\n",
2412                 uid,
2413                 gid,
2414                 tcred->euid,
2415                 tcred->suid,
2416                 tcred->uid,
2417                 tcred->egid,
2418                 tcred->sgid));
2419
2420         ret = ((uid != tcred->euid)
2421                || (uid != tcred->suid)
2422                || (uid != tcred->uid)
2423                || (gid != tcred->egid)
2424                || (gid != tcred->sgid)
2425                || (gid != tcred->gid)) && !capable(CAP_SYS_PTRACE);
2426
2427         rcu_read_unlock();
2428         return ret;
2429 }
2430
2431 static int
2432 pfarg_is_sane(struct task_struct *task, pfarg_context_t *pfx)
2433 {
2434         int ctx_flags;
2435
2436         /* valid signal */
2437
2438         ctx_flags = pfx->ctx_flags;
2439
2440         if (ctx_flags & PFM_FL_SYSTEM_WIDE) {
2441
2442                 /*
2443                  * cannot block in this mode
2444                  */
2445                 if (ctx_flags & PFM_FL_NOTIFY_BLOCK) {
2446                         DPRINT(("cannot use blocking mode when in system wide monitoring\n"));
2447                         return -EINVAL;
2448                 }
2449         } else {
2450         }
2451         /* probably more to add here */
2452
2453         return 0;
2454 }
2455
2456 static int
2457 pfm_setup_buffer_fmt(struct task_struct *task, struct file *filp, pfm_context_t *ctx, unsigned int ctx_flags,
2458                      unsigned int cpu, pfarg_context_t *arg)
2459 {
2460         pfm_buffer_fmt_t *fmt = NULL;
2461         unsigned long size = 0UL;
2462         void *uaddr = NULL;
2463         void *fmt_arg = NULL;
2464         int ret = 0;
2465 #define PFM_CTXARG_BUF_ARG(a)   (pfm_buffer_fmt_t *)(a+1)
2466
2467         /* invoke and lock buffer format, if found */
2468         fmt = pfm_find_buffer_fmt(arg->ctx_smpl_buf_id);
2469         if (fmt == NULL) {
2470                 DPRINT(("[%d] cannot find buffer format\n", task_pid_nr(task)));
2471                 return -EINVAL;
2472         }
2473
2474         /*
2475          * buffer argument MUST be contiguous to pfarg_context_t
2476          */
2477         if (fmt->fmt_arg_size) fmt_arg = PFM_CTXARG_BUF_ARG(arg);
2478
2479         ret = pfm_buf_fmt_validate(fmt, task, ctx_flags, cpu, fmt_arg);
2480
2481         DPRINT(("[%d] after validate(0x%x,%d,%p)=%d\n", task_pid_nr(task), ctx_flags, cpu, fmt_arg, ret));
2482
2483         if (ret) goto error;
2484
2485         /* link buffer format and context */
2486         ctx->ctx_buf_fmt = fmt;
2487         ctx->ctx_fl_is_sampling = 1; /* assume record() is defined */
2488
2489         /*
2490          * check if buffer format wants to use perfmon buffer allocation/mapping service
2491          */
2492         ret = pfm_buf_fmt_getsize(fmt, task, ctx_flags, cpu, fmt_arg, &size);
2493         if (ret) goto error;
2494
2495         if (size) {
2496                 /*
2497                  * buffer is always remapped into the caller's address space
2498                  */
2499                 ret = pfm_smpl_buffer_alloc(current, filp, ctx, size, &uaddr);
2500                 if (ret) goto error;
2501
2502                 /* keep track of user address of buffer */
2503                 arg->ctx_smpl_vaddr = uaddr;
2504         }
2505         ret = pfm_buf_fmt_init(fmt, task, ctx->ctx_smpl_hdr, ctx_flags, cpu, fmt_arg);
2506
2507 error:
2508         return ret;
2509 }
2510
2511 static void
2512 pfm_reset_pmu_state(pfm_context_t *ctx)
2513 {
2514         int i;
2515
2516         /*
2517          * install reset values for PMC.
2518          */
2519         for (i=1; PMC_IS_LAST(i) == 0; i++) {
2520                 if (PMC_IS_IMPL(i) == 0) continue;
2521                 ctx->ctx_pmcs[i] = PMC_DFL_VAL(i);
2522                 DPRINT(("pmc[%d]=0x%lx\n", i, ctx->ctx_pmcs[i]));
2523         }
2524         /*
2525          * PMD registers are set to 0UL when the context in memset()
2526          */
2527
2528         /*
2529          * On context switched restore, we must restore ALL pmc and ALL pmd even
2530          * when they are not actively used by the task. In UP, the incoming process
2531          * may otherwise pick up left over PMC, PMD state from the previous process.
2532          * As opposed to PMD, stale PMC can cause harm to the incoming
2533          * process because they may change what is being measured.
2534          * Therefore, we must systematically reinstall the entire
2535          * PMC state. In SMP, the same thing is possible on the
2536          * same CPU but also on between 2 CPUs.
2537          *
2538          * The problem with PMD is information leaking especially
2539          * to user level when psr.sp=0
2540          *
2541          * There is unfortunately no easy way to avoid this problem
2542          * on either UP or SMP. This definitively slows down the
2543          * pfm_load_regs() function.
2544          */
2545
2546          /*
2547           * bitmask of all PMCs accessible to this context
2548           *
2549           * PMC0 is treated differently.
2550           */
2551         ctx->ctx_all_pmcs[0] = pmu_conf->impl_pmcs[0] & ~0x1;
2552
2553         /*
2554          * bitmask of all PMDs that are accessible to this context
2555          */
2556         ctx->ctx_all_pmds[0] = pmu_conf->impl_pmds[0];
2557
2558         DPRINT(("<%d> all_pmcs=0x%lx all_pmds=0x%lx\n", ctx->ctx_fd, ctx->ctx_all_pmcs[0],ctx->ctx_all_pmds[0]));
2559
2560         /*
2561          * useful in case of re-enable after disable
2562          */
2563         ctx->ctx_used_ibrs[0] = 0UL;
2564         ctx->ctx_used_dbrs[0] = 0UL;
2565 }
2566
2567 static int
2568 pfm_ctx_getsize(void *arg, size_t *sz)
2569 {
2570         pfarg_context_t *req = (pfarg_context_t *)arg;
2571         pfm_buffer_fmt_t *fmt;
2572
2573         *sz = 0;
2574
2575         if (!pfm_uuid_cmp(req->ctx_smpl_buf_id, pfm_null_uuid)) return 0;
2576
2577         fmt = pfm_find_buffer_fmt(req->ctx_smpl_buf_id);
2578         if (fmt == NULL) {
2579                 DPRINT(("cannot find buffer format\n"));
2580                 return -EINVAL;
2581         }
2582         /* get just enough to copy in user parameters */
2583         *sz = fmt->fmt_arg_size;
2584         DPRINT(("arg_size=%lu\n", *sz));
2585
2586         return 0;
2587 }
2588
2589
2590
2591 /*
2592  * cannot attach if :
2593  *      - kernel task
2594  *      - task not owned by caller
2595  *      - task incompatible with context mode
2596  */
2597 static int
2598 pfm_task_incompatible(pfm_context_t *ctx, struct task_struct *task)
2599 {
2600         /*
2601          * no kernel task or task not owner by caller
2602          */
2603         if (task->mm == NULL) {
2604                 DPRINT(("task [%d] has not memory context (kernel thread)\n", task_pid_nr(task)));
2605                 return -EPERM;
2606         }
2607         if (pfm_bad_permissions(task)) {
2608                 DPRINT(("no permission to attach to  [%d]\n", task_pid_nr(task)));
2609                 return -EPERM;
2610         }
2611         /*
2612          * cannot block in self-monitoring mode
2613          */
2614         if (CTX_OVFL_NOBLOCK(ctx) == 0 && task == current) {
2615                 DPRINT(("cannot load a blocking context on self for [%d]\n", task_pid_nr(task)));
2616                 return -EINVAL;
2617         }
2618
2619         if (task->exit_state == EXIT_ZOMBIE) {
2620                 DPRINT(("cannot attach to  zombie task [%d]\n", task_pid_nr(task)));
2621                 return -EBUSY;
2622         }
2623
2624         /*
2625          * always ok for self
2626          */
2627         if (task == current) return 0;
2628
2629         if (!task_is_stopped_or_traced(task)) {
2630                 DPRINT(("cannot attach to non-stopped task [%d] state=%ld\n", task_pid_nr(task), task->state));
2631                 return -EBUSY;
2632         }
2633         /*
2634          * make sure the task is off any CPU
2635          */
2636         wait_task_inactive(task, 0);
2637
2638         /* more to come... */
2639
2640         return 0;
2641 }
2642
2643 static int
2644 pfm_get_task(pfm_context_t *ctx, pid_t pid, struct task_struct **task)
2645 {
2646         struct task_struct *p = current;
2647         int ret;
2648
2649         /* XXX: need to add more checks here */
2650         if (pid < 2) return -EPERM;
2651
2652         if (pid != task_pid_vnr(current)) {
2653
2654                 read_lock(&tasklist_lock);
2655
2656                 p = find_task_by_vpid(pid);
2657
2658                 /* make sure task cannot go away while we operate on it */
2659                 if (p) get_task_struct(p);
2660
2661                 read_unlock(&tasklist_lock);
2662
2663                 if (p == NULL) return -ESRCH;
2664         }
2665
2666         ret = pfm_task_incompatible(ctx, p);
2667         if (ret == 0) {
2668                 *task = p;
2669         } else if (p != current) {
2670                 pfm_put_task(p);
2671         }
2672         return ret;
2673 }
2674
2675
2676
2677 static int
2678 pfm_context_create(pfm_context_t *ctx, void *arg, int count, struct pt_regs *regs)
2679 {
2680         pfarg_context_t *req = (pfarg_context_t *)arg;
2681         struct file *filp;
2682         struct path path;
2683         int ctx_flags;
2684         int fd;
2685         int ret;
2686
2687         /* let's check the arguments first */
2688         ret = pfarg_is_sane(current, req);
2689         if (ret < 0)
2690                 return ret;
2691
2692         ctx_flags = req->ctx_flags;
2693
2694         ret = -ENOMEM;
2695
2696         fd = get_unused_fd();
2697         if (fd < 0)
2698                 return fd;
2699
2700         ctx = pfm_context_alloc(ctx_flags);
2701         if (!ctx)
2702                 goto error;
2703
2704         filp = pfm_alloc_file(ctx);
2705         if (IS_ERR(filp)) {
2706                 ret = PTR_ERR(filp);
2707                 goto error_file;
2708         }
2709
2710         req->ctx_fd = ctx->ctx_fd = fd;
2711
2712         /*
2713          * does the user want to sample?
2714          */
2715         if (pfm_uuid_cmp(req->ctx_smpl_buf_id, pfm_null_uuid)) {
2716                 ret = pfm_setup_buffer_fmt(current, filp, ctx, ctx_flags, 0, req);
2717                 if (ret)
2718                         goto buffer_error;
2719         }
2720
2721         DPRINT(("ctx=%p flags=0x%x system=%d notify_block=%d excl_idle=%d no_msg=%d ctx_fd=%d \n",
2722                 ctx,
2723                 ctx_flags,
2724                 ctx->ctx_fl_system,
2725                 ctx->ctx_fl_block,
2726                 ctx->ctx_fl_excl_idle,
2727                 ctx->ctx_fl_no_msg,
2728                 ctx->ctx_fd));
2729
2730         /*
2731          * initialize soft PMU state
2732          */
2733         pfm_reset_pmu_state(ctx);
2734
2735         fd_install(fd, filp);
2736
2737         return 0;
2738
2739 buffer_error:
2740         path = filp->f_path;
2741         put_filp(filp);
2742         path_put(&path);
2743
2744         if (ctx->ctx_buf_fmt) {
2745                 pfm_buf_fmt_exit(ctx->ctx_buf_fmt, current, NULL, regs);
2746         }
2747 error_file:
2748         pfm_context_free(ctx);
2749
2750 error:
2751         put_unused_fd(fd);
2752         return ret;
2753 }
2754
2755 static inline unsigned long
2756 pfm_new_counter_value (pfm_counter_t *reg, int is_long_reset)
2757 {
2758         unsigned long val = is_long_reset ? reg->long_reset : reg->short_reset;
2759         unsigned long new_seed, old_seed = reg->seed, mask = reg->mask;
2760         extern unsigned long carta_random32 (unsigned long seed);
2761
2762         if (reg->flags & PFM_REGFL_RANDOM) {
2763                 new_seed = carta_random32(old_seed);
2764                 val -= (old_seed & mask);       /* counter values are negative numbers! */
2765                 if ((mask >> 32) != 0)
2766                         /* construct a full 64-bit random value: */
2767                         new_seed |= carta_random32(old_seed >> 32) << 32;
2768                 reg->seed = new_seed;
2769         }
2770         reg->lval = val;
2771         return val;
2772 }
2773
2774 static void
2775 pfm_reset_regs_masked(pfm_context_t *ctx, unsigned long *ovfl_regs, int is_long_reset)
2776 {
2777         unsigned long mask = ovfl_regs[0];
2778         unsigned long reset_others = 0UL;
2779         unsigned long val;
2780         int i;
2781
2782         /*
2783          * now restore reset value on sampling overflowed counters
2784          */
2785         mask >>= PMU_FIRST_COUNTER;
2786         for(i = PMU_FIRST_COUNTER; mask; i++, mask >>= 1) {
2787
2788                 if ((mask & 0x1UL) == 0UL) continue;
2789
2790                 ctx->ctx_pmds[i].val = val = pfm_new_counter_value(ctx->ctx_pmds+ i, is_long_reset);
2791                 reset_others        |= ctx->ctx_pmds[i].reset_pmds[0];
2792
2793                 DPRINT_ovfl((" %s reset ctx_pmds[%d]=%lx\n", is_long_reset ? "long" : "short", i, val));
2794         }
2795
2796         /*
2797          * Now take care of resetting the other registers
2798          */
2799         for(i = 0; reset_others; i++, reset_others >>= 1) {
2800
2801                 if ((reset_others & 0x1) == 0) continue;
2802
2803                 ctx->ctx_pmds[i].val = val = pfm_new_counter_value(ctx->ctx_pmds + i, is_long_reset);
2804
2805                 DPRINT_ovfl(("%s reset_others pmd[%d]=%lx\n",
2806                           is_long_reset ? "long" : "short", i, val));
2807         }
2808 }
2809
2810 static void
2811 pfm_reset_regs(pfm_context_t *ctx, unsigned long *ovfl_regs, int is_long_reset)
2812 {
2813         unsigned long mask = ovfl_regs[0];
2814         unsigned long reset_others = 0UL;
2815         unsigned long val;
2816         int i;
2817
2818         DPRINT_ovfl(("ovfl_regs=0x%lx is_long_reset=%d\n", ovfl_regs[0], is_long_reset));
2819
2820         if (ctx->ctx_state == PFM_CTX_MASKED) {
2821                 pfm_reset_regs_masked(ctx, ovfl_regs, is_long_reset);
2822                 return;
2823         }
2824
2825         /*
2826          * now restore reset value on sampling overflowed counters
2827          */
2828         mask >>= PMU_FIRST_COUNTER;
2829         for(i = PMU_FIRST_COUNTER; mask; i++, mask >>= 1) {
2830
2831                 if ((mask & 0x1UL) == 0UL) continue;
2832
2833                 val           = pfm_new_counter_value(ctx->ctx_pmds+ i, is_long_reset);
2834                 reset_others |= ctx->ctx_pmds[i].reset_pmds[0];
2835
2836                 DPRINT_ovfl((" %s reset ctx_pmds[%d]=%lx\n", is_long_reset ? "long" : "short", i, val));
2837
2838                 pfm_write_soft_counter(ctx, i, val);
2839         }
2840
2841         /*
2842          * Now take care of resetting the other registers
2843          */
2844         for(i = 0; reset_others; i++, reset_others >>= 1) {
2845
2846                 if ((reset_others & 0x1) == 0) continue;
2847
2848                 val = pfm_new_counter_value(ctx->ctx_pmds + i, is_long_reset);
2849
2850                 if (PMD_IS_COUNTING(i)) {
2851                         pfm_write_soft_counter(ctx, i, val);
2852                 } else {
2853                         ia64_set_pmd(i, val);
2854                 }
2855                 DPRINT_ovfl(("%s reset_others pmd[%d]=%lx\n",
2856                           is_long_reset ? "long" : "short", i, val));
2857         }
2858         ia64_srlz_d();
2859 }
2860
2861 static int
2862 pfm_write_pmcs(pfm_context_t *ctx, void *arg, int count, struct pt_regs *regs)
2863 {
2864         struct task_struct *task;
2865         pfarg_reg_t *req = (pfarg_reg_t *)arg;
2866         unsigned long value, pmc_pm;
2867         unsigned long smpl_pmds, reset_pmds, impl_pmds;
2868         unsigned int cnum, reg_flags, flags, pmc_type;
2869         int i, can_access_pmu = 0, is_loaded, is_system, expert_mode;
2870         int is_monitor, is_counting, state;
2871         int ret = -EINVAL;
2872         pfm_reg_check_t wr_func;
2873 #define PFM_CHECK_PMC_PM(x, y, z) ((x)->ctx_fl_system ^ PMC_PM(y, z))
2874
2875         state     = ctx->ctx_state;
2876         is_loaded = state == PFM_CTX_LOADED ? 1 : 0;
2877         is_system = ctx->ctx_fl_system;
2878         task      = ctx->ctx_task;
2879         impl_pmds = pmu_conf->impl_pmds[0];
2880
2881         if (state == PFM_CTX_ZOMBIE) return -EINVAL;
2882
2883         if (is_loaded) {
2884                 /*
2885                  * In system wide and when the context is loaded, access can only happen
2886                  * when the caller is running on the CPU being monitored by the session.
2887                  * It does not have to be the owner (ctx_task) of the context per se.
2888                  */
2889                 if (is_system && ctx->ctx_cpu != smp_processor_id()) {
2890                         DPRINT(("should be running on CPU%d\n", ctx->ctx_cpu));
2891                         return -EBUSY;
2892                 }
2893                 can_access_pmu = GET_PMU_OWNER() == task || is_system ? 1 : 0;
2894         }
2895         expert_mode = pfm_sysctl.expert_mode; 
2896
2897         for (i = 0; i < count; i++, req++) {
2898
2899                 cnum       = req->reg_num;
2900                 reg_flags  = req->reg_flags;
2901                 value      = req->reg_value;
2902                 smpl_pmds  = req->reg_smpl_pmds[0];
2903                 reset_pmds = req->reg_reset_pmds[0];
2904                 flags      = 0;
2905
2906
2907                 if (cnum >= PMU_MAX_PMCS) {
2908                         DPRINT(("pmc%u is invalid\n", cnum));
2909                         goto error;
2910                 }
2911
2912                 pmc_type   = pmu_conf->pmc_desc[cnum].type;
2913                 pmc_pm     = (value >> pmu_conf->pmc_desc[cnum].pm_pos) & 0x1;
2914                 is_counting = (pmc_type & PFM_REG_COUNTING) == PFM_REG_COUNTING ? 1 : 0;
2915                 is_monitor  = (pmc_type & PFM_REG_MONITOR) == PFM_REG_MONITOR ? 1 : 0;
2916
2917                 /*
2918                  * we reject all non implemented PMC as well
2919                  * as attempts to modify PMC[0-3] which are used
2920                  * as status registers by the PMU
2921                  */
2922                 if ((pmc_type & PFM_REG_IMPL) == 0 || (pmc_type & PFM_REG_CONTROL) == PFM_REG_CONTROL) {
2923                         DPRINT(("pmc%u is unimplemented or no-access pmc_type=%x\n", cnum, pmc_type));
2924                         goto error;
2925                 }
2926                 wr_func = pmu_conf->pmc_desc[cnum].write_check;
2927                 /*
2928                  * If the PMC is a monitor, then if the value is not the default:
2929                  *      - system-wide session: PMCx.pm=1 (privileged monitor)
2930                  *      - per-task           : PMCx.pm=0 (user monitor)
2931                  */
2932                 if (is_monitor && value != PMC_DFL_VAL(cnum) && is_system ^ pmc_pm) {
2933                         DPRINT(("pmc%u pmc_pm=%lu is_system=%d\n",
2934                                 cnum,
2935                                 pmc_pm,
2936                                 is_system));
2937                         goto error;
2938                 }
2939
2940                 if (is_counting) {
2941                         /*
2942                          * enforce generation of overflow interrupt. Necessary on all
2943                          * CPUs.
2944                          */
2945                         value |= 1 << PMU_PMC_OI;
2946
2947                         if (reg_flags & PFM_REGFL_OVFL_NOTIFY) {
2948                                 flags |= PFM_REGFL_OVFL_NOTIFY;
2949                         }
2950
2951                         if (reg_flags & PFM_REGFL_RANDOM) flags |= PFM_REGFL_RANDOM;
2952
2953                         /* verify validity of smpl_pmds */
2954                         if ((smpl_pmds & impl_pmds) != smpl_pmds) {
2955                                 DPRINT(("invalid smpl_pmds 0x%lx for pmc%u\n", smpl_pmds, cnum));
2956                                 goto error;
2957                         }
2958
2959                         /* verify validity of reset_pmds */
2960                         if ((reset_pmds & impl_pmds) != reset_pmds) {
2961                                 DPRINT(("invalid reset_pmds 0x%lx for pmc%u\n", reset_pmds, cnum));
2962                                 goto error;
2963                         }
2964                 } else {
2965                         if (reg_flags & (PFM_REGFL_OVFL_NOTIFY|PFM_REGFL_RANDOM)) {
2966                                 DPRINT(("cannot set ovfl_notify or random on pmc%u\n", cnum));
2967                                 goto error;
2968                         }
2969                         /* eventid on non-counting monitors are ignored */
2970                 }
2971
2972                 /*
2973                  * execute write checker, if any
2974                  */
2975                 if (likely(expert_mode == 0 && wr_func)) {
2976                         ret = (*wr_func)(task, ctx, cnum, &value, regs);
2977                         if (ret) goto error;
2978                         ret = -EINVAL;
2979                 }
2980
2981                 /*
2982                  * no error on this register
2983                  */
2984                 PFM_REG_RETFLAG_SET(req->reg_flags, 0);
2985
2986                 /*
2987                  * Now we commit the changes to the software state
2988                  */
2989
2990                 /*
2991                  * update overflow information
2992                  */
2993                 if (is_counting) {
2994                         /*
2995                          * full flag update each time a register is programmed
2996                          */
2997                         ctx->ctx_pmds[cnum].flags = flags;
2998
2999                         ctx->ctx_pmds[cnum].reset_pmds[0] = reset_pmds;
3000                         ctx->ctx_pmds[cnum].smpl_pmds[0]  = smpl_pmds;
3001                         ctx->ctx_pmds[cnum].eventid       = req->reg_smpl_eventid;
3002
3003                         /*
3004                          * Mark all PMDS to be accessed as used.
3005                          *
3006                          * We do not keep track of PMC because we have to
3007                          * systematically restore ALL of them.
3008                          *
3009                          * We do not update the used_monitors mask, because
3010                          * if we have not programmed them, then will be in
3011                          * a quiescent state, therefore we will not need to
3012                          * mask/restore then when context is MASKED.
3013                          */
3014                         CTX_USED_PMD(ctx, reset_pmds);
3015                         CTX_USED_PMD(ctx, smpl_pmds);
3016                         /*
3017                          * make sure we do not try to reset on
3018                          * restart because we have established new values
3019                          */
3020                         if (state == PFM_CTX_MASKED) ctx->ctx_ovfl_regs[0] &= ~1UL << cnum;
3021                 }
3022                 /*
3023                  * Needed in case the user does not initialize the equivalent
3024                  * PMD. Clearing is done indirectly via pfm_reset_pmu_state() so there is no
3025                  * possible leak here.
3026                  */
3027                 CTX_USED_PMD(ctx, pmu_conf->pmc_desc[cnum].dep_pmd[0]);
3028
3029                 /*
3030                  * keep track of the monitor PMC that we are using.
3031                  * we save the value of the pmc in ctx_pmcs[] and if
3032                  * the monitoring is not stopped for the context we also
3033                  * place it in the saved state area so that it will be
3034                  * picked up later by the context switch code.
3035                  *
3036                  * The value in ctx_pmcs[] can only be changed in pfm_write_pmcs().
3037                  *
3038                  * The value in th_pmcs[] may be modified on overflow, i.e.,  when
3039                  * monitoring needs to be stopped.
3040                  */
3041                 if (is_monitor) CTX_USED_MONITOR(ctx, 1UL << cnum);
3042
3043                 /*
3044                  * update context state
3045                  */
3046                 ctx->ctx_pmcs[cnum] = value;
3047
3048                 if (is_loaded) {
3049                         /*
3050                          * write thread state
3051                          */
3052                         if (is_system == 0) ctx->th_pmcs[cnum] = value;
3053
3054                         /*
3055                          * write hardware register if we can
3056                          */
3057                         if (can_access_pmu) {
3058                                 ia64_set_pmc(cnum, value);
3059                         }
3060 #ifdef CONFIG_SMP
3061                         else {
3062                                 /*
3063                                  * per-task SMP only here
3064                                  *
3065                                  * we are guaranteed that the task is not running on the other CPU,
3066                                  * we indicate that this PMD will need to be reloaded if the task
3067                                  * is rescheduled on the CPU it ran last on.
3068                                  */
3069                                 ctx->ctx_reload_pmcs[0] |= 1UL << cnum;
3070                         }
3071 #endif
3072                 }
3073
3074                 DPRINT(("pmc[%u]=0x%lx ld=%d apmu=%d flags=0x%x all_pmcs=0x%lx used_pmds=0x%lx eventid=%ld smpl_pmds=0x%lx reset_pmds=0x%lx reloads_pmcs=0x%lx used_monitors=0x%lx ovfl_regs=0x%lx\n",
3075                           cnum,
3076                           value,
3077                           is_loaded,
3078                           can_access_pmu,
3079                           flags,
3080                           ctx->ctx_all_pmcs[0],
3081                           ctx->ctx_used_pmds[0],
3082                           ctx->ctx_pmds[cnum].eventid,
3083                           smpl_pmds,
3084                           reset_pmds,
3085                           ctx->ctx_reload_pmcs[0],
3086                           ctx->ctx_used_monitors[0],
3087                           ctx->ctx_ovfl_regs[0]));
3088         }
3089
3090         /*
3091          * make sure the changes are visible
3092          */
3093         if (can_access_pmu) ia64_srlz_d();
3094
3095         return 0;
3096 error:
3097         PFM_REG_RETFLAG_SET(req->reg_flags, PFM_REG_RETFL_EINVAL);
3098         return ret;
3099 }
3100
3101 static int
3102 pfm_write_pmds(pfm_context_t *ctx, void *arg, int count, struct pt_regs *regs)
3103 {
3104         struct task_struct *task;
3105         pfarg_reg_t *req = (pfarg_reg_t *)arg;
3106         unsigned long value, hw_value, ovfl_mask;
3107         unsigned int cnum;
3108         int i, can_access_pmu = 0, state;
3109         int is_counting, is_loaded, is_system, expert_mode;
3110         int ret = -EINVAL;
3111         pfm_reg_check_t wr_func;
3112
3113
3114         state     = ctx->ctx_state;
3115         is_loaded = state == PFM_CTX_LOADED ? 1 : 0;
3116         is_system = ctx->ctx_fl_system;
3117         ovfl_mask = pmu_conf->ovfl_val;
3118         task      = ctx->ctx_task;
3119
3120         if (unlikely(state == PFM_CTX_ZOMBIE)) return -EINVAL;
3121
3122         /*
3123          * on both UP and SMP, we can only write to the PMC when the task is
3124          * the owner of the local PMU.
3125          */
3126         if (likely(is_loaded)) {
3127                 /*
3128                  * In system wide and when the context is loaded, access can only happen
3129                  * when the caller is running on the CPU being monitored by the session.
3130                  * It does not have to be the owner (ctx_task) of the context per se.
3131                  */
3132                 if (unlikely(is_system && ctx->ctx_cpu != smp_processor_id())) {
3133                         DPRINT(("should be running on CPU%d\n", ctx->ctx_cpu));
3134                         return -EBUSY;
3135                 }
3136                 can_access_pmu = GET_PMU_OWNER() == task || is_system ? 1 : 0;
3137         }
3138         expert_mode = pfm_sysctl.expert_mode; 
3139
3140         for (i = 0; i < count; i++, req++) {
3141
3142                 cnum  = req->reg_num;
3143                 value = req->reg_value;
3144
3145                 if (!PMD_IS_IMPL(cnum)) {
3146                         DPRINT(("pmd[%u] is unimplemented or invalid\n", cnum));
3147                         goto abort_mission;
3148                 }
3149                 is_counting = PMD_IS_COUNTING(cnum);
3150                 wr_func     = pmu_conf->pmd_desc[cnum].write_check;
3151
3152                 /*
3153                  * execute write checker, if any
3154                  */
3155                 if (unlikely(expert_mode == 0 && wr_func)) {
3156                         unsigned long v = value;
3157
3158                         ret = (*wr_func)(task, ctx, cnum, &v, regs);
3159                         if (ret) goto abort_mission;
3160
3161                         value = v;
3162                         ret   = -EINVAL;
3163                 }
3164
3165                 /*
3166                  * no error on this register
3167                  */
3168                 PFM_REG_RETFLAG_SET(req->reg_flags, 0);
3169
3170                 /*
3171                  * now commit changes to software state
3172                  */
3173                 hw_value = value;
3174
3175                 /*
3176                  * update virtualized (64bits) counter
3177                  */
3178                 if (is_counting) {
3179                         /*
3180                          * write context state
3181                          */
3182                         ctx->ctx_pmds[cnum].lval = value;
3183
3184                         /*
3185                          * when context is load we use the split value
3186                          */
3187                         if (is_loaded) {
3188                                 hw_value = value &  ovfl_mask;
3189                                 value    = value & ~ovfl_mask;
3190                         }
3191                 }
3192                 /*
3193                  * update reset values (not just for counters)
3194                  */
3195                 ctx->ctx_pmds[cnum].long_reset  = req->reg_long_reset;
3196                 ctx->ctx_pmds[cnum].short_reset = req->reg_short_reset;
3197
3198                 /*
3199                  * update randomization parameters (not just for counters)
3200                  */
3201                 ctx->ctx_pmds[cnum].seed = req->reg_random_seed;
3202                 ctx->ctx_pmds[cnum].mask = req->reg_random_mask;
3203
3204                 /*
3205                  * update context value
3206                  */
3207                 ctx->ctx_pmds[cnum].val  = value;
3208
3209                 /*
3210                  * Keep track of what we use
3211                  *
3212                  * We do not keep track of PMC because we have to
3213                  * systematically restore ALL of them.
3214                  */
3215                 CTX_USED_PMD(ctx, PMD_PMD_DEP(cnum));
3216
3217                 /*
3218                  * mark this PMD register used as well
3219                  */
3220                 CTX_USED_PMD(ctx, RDEP(cnum));
3221
3222                 /*
3223                  * make sure we do not try to reset on
3224                  * restart because we have established new values
3225                  */
3226                 if (is_counting && state == PFM_CTX_MASKED) {
3227                         ctx->ctx_ovfl_regs[0] &= ~1UL << cnum;
3228                 }
3229
3230                 if (is_loaded) {
3231                         /*
3232                          * write thread state
3233                          */
3234                         if (is_system == 0) ctx->th_pmds[cnum] = hw_value;
3235
3236                         /*
3237                          * write hardware register if we can
3238                          */
3239                         if (can_access_pmu) {
3240                                 ia64_set_pmd(cnum, hw_value);
3241                         } else {
3242 #ifdef CONFIG_SMP
3243                                 /*
3244                                  * we are guaranteed that the task is not running on the other CPU,
3245                                  * we indicate that this PMD will need to be reloaded if the task
3246                                  * is rescheduled on the CPU it ran last on.
3247                                  */
3248                                 ctx->ctx_reload_pmds[0] |= 1UL << cnum;
3249 #endif
3250                         }
3251                 }
3252
3253                 DPRINT(("pmd[%u]=0x%lx ld=%d apmu=%d, hw_value=0x%lx ctx_pmd=0x%lx  short_reset=0x%lx "
3254                           "long_reset=0x%lx notify=%c seed=0x%lx mask=0x%lx used_pmds=0x%lx reset_pmds=0x%lx reload_pmds=0x%lx all_pmds=0x%lx ovfl_regs=0x%lx\n",
3255                         cnum,
3256                         value,
3257                         is_loaded,
3258