fflush(stdout);
 }
 
+static void cs_etm__set_trace_param_etmv3(struct cs_etm_trace_params *t_params,
+                                         struct cs_etm_auxtrace *etm, int idx,
+                                         u32 etmidr)
+{
+       u64 **metadata = etm->metadata;
+
+       t_params[idx].protocol = cs_etm__get_v7_protocol_version(etmidr);
+       t_params[idx].etmv3.reg_ctrl = metadata[idx][CS_ETM_ETMCR];
+       t_params[idx].etmv3.reg_trc_id = metadata[idx][CS_ETM_ETMTRACEIDR];
+}
+
+static void cs_etm__set_trace_param_etmv4(struct cs_etm_trace_params *t_params,
+                                         struct cs_etm_auxtrace *etm, int idx)
+{
+       u64 **metadata = etm->metadata;
+
+       t_params[idx].protocol = CS_ETM_PROTO_ETMV4i;
+       t_params[idx].etmv4.reg_idr0 = metadata[idx][CS_ETMV4_TRCIDR0];
+       t_params[idx].etmv4.reg_idr1 = metadata[idx][CS_ETMV4_TRCIDR1];
+       t_params[idx].etmv4.reg_idr2 = metadata[idx][CS_ETMV4_TRCIDR2];
+       t_params[idx].etmv4.reg_idr8 = metadata[idx][CS_ETMV4_TRCIDR8];
+       t_params[idx].etmv4.reg_configr = metadata[idx][CS_ETMV4_TRCCONFIGR];
+       t_params[idx].etmv4.reg_traceidr = metadata[idx][CS_ETMV4_TRCTRACEIDR];
+}
+
+static int cs_etm__init_trace_params(struct cs_etm_trace_params *t_params,
+                                    struct cs_etm_auxtrace *etm)
+{
+       int i;
+       u32 etmidr;
+       u64 architecture;
+
+       for (i = 0; i < etm->num_cpu; i++) {
+               architecture = etm->metadata[i][CS_ETM_MAGIC];
+
+               switch (architecture) {
+               case __perf_cs_etmv3_magic:
+                       etmidr = etm->metadata[i][CS_ETM_ETMIDR];
+                       cs_etm__set_trace_param_etmv3(t_params, etm, i, etmidr);
+                       break;
+               case __perf_cs_etmv4_magic:
+                       cs_etm__set_trace_param_etmv4(t_params, etm, i);
+                       break;
+               default:
+                       return -EINVAL;
+               }
+       }
+
+       return 0;
+}
+
 static int cs_etm__init_decoder_params(struct cs_etm_decoder_params *d_params,
                                       struct cs_etm_queue *etmq,
                                       enum cs_etm_decoder_operation mode)
 static void cs_etm__dump_event(struct cs_etm_auxtrace *etm,
                               struct auxtrace_buffer *buffer)
 {
-       int i, ret;
+       int ret;
        const char *color = PERF_COLOR_BLUE;
        struct cs_etm_decoder_params d_params;
        struct cs_etm_trace_params *t_params;
        if (!t_params)
                return;
 
-       for (i = 0; i < etm->num_cpu; i++) {
-               if (etm->metadata[i][CS_ETM_MAGIC] == __perf_cs_etmv3_magic) {
-                       u32 etmidr = etm->metadata[i][CS_ETM_ETMIDR];
-
-                       t_params[i].protocol =
-                                       cs_etm__get_v7_protocol_version(etmidr);
-                       t_params[i].etmv3.reg_ctrl =
-                                       etm->metadata[i][CS_ETM_ETMCR];
-                       t_params[i].etmv3.reg_trc_id =
-                                       etm->metadata[i][CS_ETM_ETMTRACEIDR];
-               } else if (etm->metadata[i][CS_ETM_MAGIC] ==
-                                                     __perf_cs_etmv4_magic) {
-                       t_params[i].protocol = CS_ETM_PROTO_ETMV4i;
-                       t_params[i].etmv4.reg_idr0 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR0];
-                       t_params[i].etmv4.reg_idr1 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR1];
-                       t_params[i].etmv4.reg_idr2 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR2];
-                       t_params[i].etmv4.reg_idr8 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR8];
-                       t_params[i].etmv4.reg_configr =
-                                       etm->metadata[i][CS_ETMV4_TRCCONFIGR];
-                       t_params[i].etmv4.reg_traceidr =
-                                       etm->metadata[i][CS_ETMV4_TRCTRACEIDR];
-               }
-       }
+       if (cs_etm__init_trace_params(t_params, etm))
+               goto out_free;
 
        /* Set decoder parameters to simply print the trace packets */
        if (cs_etm__init_decoder_params(&d_params, NULL,
 static struct cs_etm_queue *cs_etm__alloc_queue(struct cs_etm_auxtrace *etm,
                                                unsigned int queue_nr)
 {
-       int i;
        struct cs_etm_decoder_params d_params;
        struct cs_etm_trace_params  *t_params = NULL;
        struct cs_etm_queue *etmq;
        if (!t_params)
                goto out_free;
 
-       for (i = 0; i < etm->num_cpu; i++) {
-               if (etm->metadata[i][CS_ETM_MAGIC] == __perf_cs_etmv3_magic) {
-                       u32 etmidr = etm->metadata[i][CS_ETM_ETMIDR];
-
-                       t_params[i].protocol =
-                                       cs_etm__get_v7_protocol_version(etmidr);
-                       t_params[i].etmv3.reg_ctrl =
-                                       etm->metadata[i][CS_ETM_ETMCR];
-                       t_params[i].etmv3.reg_trc_id =
-                                       etm->metadata[i][CS_ETM_ETMTRACEIDR];
-               } else if (etm->metadata[i][CS_ETM_MAGIC] ==
-                                                       __perf_cs_etmv4_magic) {
-                       t_params[i].protocol = CS_ETM_PROTO_ETMV4i;
-                       t_params[i].etmv4.reg_idr0 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR0];
-                       t_params[i].etmv4.reg_idr1 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR1];
-                       t_params[i].etmv4.reg_idr2 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR2];
-                       t_params[i].etmv4.reg_idr8 =
-                                       etm->metadata[i][CS_ETMV4_TRCIDR8];
-                       t_params[i].etmv4.reg_configr =
-                                       etm->metadata[i][CS_ETMV4_TRCCONFIGR];
-                       t_params[i].etmv4.reg_traceidr =
-                                       etm->metadata[i][CS_ETMV4_TRCTRACEIDR];
-               }
-       }
+       if (cs_etm__init_trace_params(t_params, etm))
+               goto out_free;
 
        /* Set decoder parameters to simply print the trace packets */
        if (cs_etm__init_decoder_params(&d_params, etmq,