--- /dev/null
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "svm_fifo_segment.h"
+
+clib_error_t *
+hello_world (int verbose)
+{
+ svm_fifo_segment_create_args_t _a, *a = &_a;
+ svm_fifo_segment_private_t *sp;
+ svm_fifo_t *f;
+ int rv;
+ u8 *test_data;
+ u8 *retrieved_data = 0;
+ clib_error_t *error = 0;
+ int pid = getpid ();
+
+ memset (a, 0, sizeof (*a));
+
+ a->segment_name = "fifo-test1";
+ a->segment_size = 256 << 10;
+
+ rv = svm_fifo_segment_create (a);
+
+ if (rv)
+ return clib_error_return (0, "svm_fifo_segment_create returned %d", rv);
+
+ sp = svm_fifo_get_segment (a->new_segment_index);
+
+ f = svm_fifo_segment_alloc_fifo (sp, 4096);
+
+ if (f == 0)
+ return clib_error_return (0, "svm_fifo_segment_alloc_fifo failed");
+
+ test_data = format (0, "Hello world%c", 0);
+ vec_validate (retrieved_data, vec_len (test_data) - 1);
+
+ while (svm_fifo_max_enqueue (f) >= vec_len (test_data))
+ svm_fifo_enqueue_nowait (f, pid, vec_len (test_data), test_data);
+
+ while (svm_fifo_max_dequeue (f) >= vec_len (test_data))
+ svm_fifo_dequeue_nowait (f, pid, vec_len (retrieved_data),
+ retrieved_data);
+
+ while (svm_fifo_max_enqueue (f) >= vec_len (test_data))
+ svm_fifo_enqueue_nowait (f, pid, vec_len (test_data), test_data);
+
+ while (svm_fifo_max_dequeue (f) >= vec_len (test_data))
+ svm_fifo_dequeue_nowait (f, pid, vec_len (retrieved_data),
+ retrieved_data);
+
+ if (!memcmp (retrieved_data, test_data, vec_len (test_data)))
+ error = clib_error_return (0, "data test OK, got '%s'", retrieved_data);
+ else
+ error = clib_error_return (0, "data test FAIL!");
+
+ svm_fifo_segment_free_fifo (sp, f);
+
+ return error;
+}
+
+clib_error_t *
+master (int verbose)
+{
+ svm_fifo_segment_create_args_t _a, *a = &_a;
+ svm_fifo_segment_private_t *sp;
+ svm_fifo_t *f;
+ int rv;
+ u8 *test_data;
+ u8 *retrieved_data = 0;
+ int i;
+ int pid = getpid ();
+
+ memset (a, 0, sizeof (*a));
+
+ a->segment_name = "fifo-test1";
+ a->segment_size = 256 << 10;
+
+ rv = svm_fifo_segment_create (a);
+
+ if (rv)
+ return clib_error_return (0, "svm_fifo_segment_create returned %d", rv);
+
+ sp = svm_fifo_get_segment (a->new_segment_index);
+
+ f = svm_fifo_segment_alloc_fifo (sp, 4096);
+
+ if (f == 0)
+ return clib_error_return (0, "svm_fifo_segment_alloc_fifo failed");
+
+ test_data = format (0, "Hello world%c", 0);
+ vec_validate (retrieved_data, vec_len (test_data) - 1);
+
+ for (i = 0; i < 1000; i++)
+ svm_fifo_enqueue_nowait (f, pid, vec_len (test_data), test_data);
+
+ return clib_error_return (0, "master (enqueue) done");
+}
+
+clib_error_t *
+mempig (int verbose)
+{
+ svm_fifo_segment_create_args_t _a, *a = &_a;
+ svm_fifo_segment_private_t *sp;
+ svm_fifo_t *f;
+ svm_fifo_t **flist = 0;
+ int rv;
+ int i;
+
+ memset (a, 0, sizeof (*a));
+
+ a->segment_name = "fifo-test1";
+ a->segment_size = 256 << 10;
+
+ rv = svm_fifo_segment_create (a);
+
+ if (rv)
+ return clib_error_return (0, "svm_fifo_segment_create returned %d", rv);
+
+ sp = svm_fifo_get_segment (a->new_segment_index);
+
+ for (i = 0; i < 1000; i++)
+ {
+ f = svm_fifo_segment_alloc_fifo (sp, 4096);
+ if (f == 0)
+ break;
+ vec_add1 (flist, f);
+ }
+
+ fformat (stdout, "Try #1: created %d fifos...\n", vec_len (flist));
+ for (i = 0; i < vec_len (flist); i++)
+ {
+ f = flist[i];
+ svm_fifo_segment_free_fifo (sp, f);
+ }
+
+ _vec_len (flist) = 0;
+
+ for (i = 0; i < 1000; i++)
+ {
+ f = svm_fifo_segment_alloc_fifo (sp, 4096);
+ if (f == 0)
+ break;
+ vec_add1 (flist, f);
+ }
+
+ fformat (stdout, "Try #2: created %d fifos...\n", vec_len (flist));
+ for (i = 0; i < vec_len (flist); i++)
+ {
+ f = flist[i];
+ svm_fifo_segment_free_fifo (sp, f);
+ }
+
+ return 0;
+}
+
+clib_error_t *
+offset (int verbose)
+{
+ svm_fifo_segment_create_args_t _a, *a = &_a;
+ svm_fifo_segment_private_t *sp;
+ svm_fifo_t *f;
+ int rv;
+ u32 *test_data = 0;
+ u32 *recovered_data = 0;
+ int i;
+ int pid = getpid ();
+
+ memset (a, 0, sizeof (*a));
+
+ a->segment_name = "fifo-test1";
+ a->segment_size = 256 << 10;
+
+ rv = svm_fifo_segment_create (a);
+
+ if (rv)
+ return clib_error_return (0, "svm_fifo_segment_create returned %d", rv);
+
+ sp = svm_fifo_get_segment (a->new_segment_index);
+
+ f = svm_fifo_segment_alloc_fifo (sp, 200 << 10);
+
+ if (f == 0)
+ return clib_error_return (0, "svm_fifo_segment_alloc_fifo failed");
+
+ for (i = 0; i < (3 * 1024); i++)
+ vec_add1 (test_data, i);
+
+ /* Enqueue the first 1024 u32's */
+ svm_fifo_enqueue_nowait (f, pid, 4096 /* bytes to enqueue */ ,
+ (u8 *) test_data);
+
+ /* Enqueue the third 1024 u32's 2048 ahead of the current tail */
+ svm_fifo_enqueue_with_offset (f, pid, 4096, 4096, (u8 *) & test_data[2048]);
+
+ /* Enqueue the second 1024 u32's at the current tail */
+ svm_fifo_enqueue_nowait (f, pid, 4096 /* bytes to enqueue */ ,
+ (u8 *) & test_data[1024]);
+
+ vec_validate (recovered_data, (3 * 1024) - 1);
+
+ svm_fifo_dequeue_nowait (f, pid, 3 * 4096, (u8 *) recovered_data);
+
+ for (i = 0; i < (3 * 1024); i++)
+ {
+ if (recovered_data[i] != test_data[i])
+ {
+ clib_warning ("[%d] expected %d recovered %d", i,
+ test_data[i], recovered_data[i]);
+ return clib_error_return (0, "offset test FAILED");
+ }
+ }
+
+ return clib_error_return (0, "offset test OK");
+}
+
+clib_error_t *
+slave (int verbose)
+{
+ svm_fifo_segment_create_args_t _a, *a = &_a;
+ svm_fifo_segment_private_t *sp;
+ svm_fifo_segment_header_t *fsh;
+ svm_fifo_t *f;
+ ssvm_shared_header_t *sh;
+ int rv;
+ u8 *test_data;
+ u8 *retrieved_data = 0;
+ int pid = getpid ();
+ int i;
+
+ memset (a, 0, sizeof (*a));
+
+ a->segment_name = "fifo-test1";
+
+ rv = svm_fifo_segment_attach (a);
+
+ if (rv)
+ return clib_error_return (0, "svm_fifo_segment_attach returned %d", rv);
+
+ sp = svm_fifo_get_segment (a->new_segment_index);
+ sh = sp->ssvm.sh;
+ fsh = (svm_fifo_segment_header_t *) sh->opaque[0];
+
+ /* might wanna wait.. */
+ f = (svm_fifo_t *) fsh->fifos[0];
+
+ /* Lazy bastards united */
+ test_data = format (0, "Hello world%c", 0);
+ vec_validate (retrieved_data, vec_len (test_data) - 1);
+
+ for (i = 0; i < 1000; i++)
+ {
+ svm_fifo_dequeue_nowait (f, pid, vec_len (retrieved_data),
+ retrieved_data);
+ if (memcmp (retrieved_data, test_data, vec_len (retrieved_data)))
+ return clib_error_return (0, "retrieved data incorrect, '%s'",
+ retrieved_data);
+ }
+
+ return clib_error_return (0, "slave (dequeue) done");
+}
+
+
+int
+test_ssvm_fifo1 (unformat_input_t * input)
+{
+ clib_error_t *error = 0;
+ int verbose = 0;
+ int test_id = 0;
+
+ svm_fifo_segment_init (0x200000000ULL, 20);
+
+ while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
+ {
+ if (unformat (input, "verbose %d", &verbose))
+ ;
+ else if (unformat (input, "verbose"))
+ verbose = 1;
+ else if (unformat (input, "master"))
+ test_id = 1;
+ else if (unformat (input, "slave"))
+ test_id = 2;
+ else if (unformat (input, "mempig"))
+ test_id = 3;
+ else if (unformat (input, "offset"))
+ test_id = 4;
+ else
+ {
+ error = clib_error_create ("unknown input `%U'\n",
+ format_unformat_error, input);
+ goto out;
+ }
+ }
+
+ switch (test_id)
+ {
+ case 0:
+ error = hello_world (verbose);
+ break;
+
+ case 1:
+ error = master (verbose);
+ break;
+
+ case 2:
+ error = slave (verbose);
+ break;
+
+ case 3:
+ error = mempig (verbose);
+ break;
+
+ case 4:
+ error = offset (verbose);
+ break;
+
+ default:
+ error = clib_error_return (0, "test id %d unknown", test_id);
+ break;
+ }
+
+out:
+ if (error)
+ clib_error_report (error);
+
+ return 0;
+}
+
+
+
+int
+main (int argc, char *argv[])
+{
+ unformat_input_t i;
+ int r;
+
+ unformat_init_command_line (&i, argv);
+ r = test_ssvm_fifo1 (&i);
+ unformat_free (&i);
+ return r;
+}
+
+/*
+ * fd.io coding-style-patch-verification: ON
+ *
+ * Local Variables:
+ * eval: (c-set-style "gnu")
+ * End:
+ */
Code Review / vpp.git / blobdiff