Fix head

Author: savalet

src/copy_data.c

@@ -55,10 +55,12 @@ bool copy_data_value(dataframe_t *dst, dataframe_t *src, int row)
 __attribute__((nonnull(1, 2)))
 bool copy_data(dataframe_t *dst, dataframe_t *src, int nb_rows)
 {
+    int row = 0;
+
     dst->data = (void ***)malloc(sizeof(void *) * nb_rows);
     if ((void *)dst->data == NULL)
         return false;
-    for (int row = 0; row < nb_rows; row++) {
+    for (; row < nb_rows && row < src->nb_rows; row++) {
         dst->data[row] = (void **)malloc(sizeof(void *) * src->nb_columns);
 #pragma GCC diagnostic push
 #pragma GCC diagnostic ignored "-Wanalyzer-malloc-leak"
@@ -68,5 +70,6 @@ bool copy_data(dataframe_t *dst, dataframe_t *src, int nb_rows)
             return (free((void *)dst->data), false);
 #pragma GCC diagnostic pop
     }
+    dst->nb_rows = row + 1;
     return true;
 }

src/df_head.c

@@ -21,7 +21,6 @@ dataframe_t *df_head(dataframe_t *df, int nb_rows)
     if (new_df == NULL)
         return NULL;
     new_df->nb_columns = df->nb_columns;
-    new_df->nb_rows = nb_rows;
     if (!copy_columns_type(new_df, df) || !copy_columns_name(new_df, df) ||
         !copy_data(new_df, df, nb_rows))
         return (free(new_df), NULL);