k_order_welfare MEX: check number of arguments

2022-07-05 11:55:12 +02:00 · 2022-07-05 11:55:12 +02:00 · 218bb1a15f
parent 0a9dc0137a
commit 218bb1a15f
1 changed files with 2 additions and 0 deletions
--- a/mex/sources/k_order_welfare/k_order_welfare.cc
+++ b/mex/sources/k_order_welfare/k_order_welfare.cc
@ -87,6 +87,8 @@ extern "C" {
  void mexFunction(int nlhs, mxArray *plhs[],
                   int nrhs, const mxArray *prhs[])
  {
+    if (nlhs != 1 || nrhs != 3)
+      mexErrMsgTxt("Must have exactly 3 input arguments and 1 output argument");

    const mxArray *dr_mx = prhs[0];
    const mxArray *M_mx = prhs[1];