将主要分为一个功能

时间:2012-05-04 15:30:32

标签: c function

这是我的代码:

#include <stdio.h>
#include <stdlib.h>

int main()
{
        char c, *data, *temp;
        data=malloc(sizeof(char));
        int i, conta=0;
        printf("Write hostname: ");
        for (i=0;;i++) {
            c=getchar();
            conta++;
            if (c=='\n') break; /* Se premo invio esco dal ciclo */
            data[i]=c; /* Inserisco il carattere nell'array */
            temp=realloc(data,(i+2)*sizeof(char)); /* do memoria aggiuntiva al puntatore (+2 per carattere NUL) */
            if ( temp != NULL ) {
                    data=temp; /* Se la riallocazione va a buon fine assegno temp a data */
            } else {
                    free(data); /* altrimenti libero il puntatore, stampo messaggio errore e chiudo programma */
                    printf("Error allocating memory!\n");
                    return EXIT_FAILURE;
            }    
        }
    /* Stampo la stringa letta, libero puntatore e chiudo programma */
    for (i=0;i<=conta;i++){
            printf("%c", data[i]);
        }
        printf("\n");
        free(data);
        return EXIT_SUCCESS;
}

我想将我的主要分成一个可以获得用户输入的函数(主机名,用户名)。
我想要的是:

#include <stdio.h>
#include <stdlib.h>

function read_input();

int main()
{
    printf("Write hostname: ");
    read_input();
    printf("Write username: ");
    read_input();
    for (i=0;i<=conta;i++){
      printf("%c", data[i]);
    }
    printf("\n");
    free(data);
    return EXIT_SUCCESS;
}

read_input(){
....
}

但我不知道如何开始制作这个功能...
我不知道如何创建一个函数,它将返回conta(对于for cicle),指针data(必须释放它,并且需要打印输出)。
我是否要制作一个结构并将其传递给函数返回?

1 个答案:

答案 0 :(得分:1)

您可以这样组织代码:

#include <stdio.h>
#include <stdlib.h>

int read_input(char data[]);

int main()
{
    char *data;
    int i, conta;
    data=malloc(sizeof(char));
    printf("Write hostname: ");
    conta = read_input(data);
    /* Stampo la stringa letta, libero puntatore e chiudo programma */
    for (i=0;i<=conta;i++){
        printf("%c", data[i]);
    }
    printf("\n");
    free(data);
    return EXIT_SUCCESS;
}

int read_input(char data[])
{
    char c, *temp;
    int i, conta = 0;
    for (i=0;;i++) {
        c=getchar();
        conta++;
        if (c=='\n') break; /* Se premo invio esco dal ciclo */
        data[i]=c; /* Inserisco il carattere nell'array */
        temp=realloc(data,(i+2)*sizeof(char)); /* do memoria aggiuntiva al puntatore (+2 per carattere NUL) */
        if ( temp != NULL ) {
                data=temp; /* Se la riallocazione va a buon fine assegno temp a data */
        } else {
                free(data); /* altrimenti libero il puntatore, stampo messaggio errore e chiudo programma */
                printf("Error allocating memory!\n");
                return EXIT_FAILURE;
        }
    }
    return conta;
}