u16 link_width;
        u16 link_speed;
 
-       response_data_size = sizeof(struct opa_port_status_rsp) +
-                               num_vls * sizeof(struct _vls_pctrs);
+       response_data_size = struct_size(rsp, vls, num_vls);
        if (response_data_size > sizeof(pmp->data)) {
                pmp->mad_hdr.status |= OPA_PM_STATUS_REQUEST_TOO_LARGE;
                return reply((struct ib_mad_hdr *)pmp);
        }
 
        /* Sanity check */
-       response_data_size = sizeof(struct opa_port_data_counters_msg) +
-                               num_vls * sizeof(struct _vls_dctrs);
+       response_data_size = struct_size(req, port[0].vls, num_vls);
 
        if (response_data_size > sizeof(pmp->data)) {
                pmp->mad_hdr.status |= IB_SMP_INVALID_FIELD;
                return reply((struct ib_mad_hdr *)pmp);
        }
 
-       response_data_size = sizeof(struct opa_port_error_counters64_msg) +
-                               num_vls * sizeof(struct _vls_ectrs);
+       response_data_size = struct_size(req, port[0].vls, num_vls);
 
        if (response_data_size > sizeof(pmp->data)) {
                pmp->mad_hdr.status |= IB_SMP_INVALID_FIELD;