Читать онлайн «Разработка приложений в среде Linux. Второе издание». Страница 140

 42:    exit(1);

 43:   }

 44:  }

 45:

 46:  int main(int argc, const char ** argv) {

 47:   pam_handle_t * pamh;

 48:   struct passwd * pw;

 49:   char * username=NULL, * service=NULL;

 50:   int account = 1, session = 0;

 51:   int c;

 52:   poptContext optCon;

 53:   struct poptOption optionsTable[] = {

 54:    { "username", 'u', POPT_ARG_STRING, &username, 0,

 55:      "Имя пользователя для аутентификации", "<имя_пользователя>" },

 56:    { "service", 'S', РОPT_ARG_STRING, &service, 0,

 57:      "Имя службы для инициализации как (pamsample)",

 58:      "<служба>" },

 59:    { "account", 'a', POPT_ARG_NONE|POPT_ARGFLAG_XOR,

 60:      &account, 0,

 61:      "включение/выключение управления учетными записями (включено)", "" },

 62:    { "session", 's', POPT_ARG_NONE|POPT_ARGFLAG_XOR,

 63:      &session, 0,

 64:      "включение/выключение запуска сеанса (выключено)", "" },

 65:    POPT_AUTOHELP

 66:    POPT_TABLEEND

 67:   };

 68:

 69:   optCon = poptGetContext("pamexample", argc, argv,

 70:    optionsTable, 0);

 71:   if ((c = poptGetNextOpt(optCon)) < -1) {

 72:    fprintf(stderr, "%s: %s\n",

 73:    poptBadOption(optCon, POPT_BADOPTION_NOALIAS),

 74:    poptStrerror(c));

 75:    return 1;

 76:   }

 77:   poptFreeContext(optCon);

 78:

 79:   if (!service) {

 80:    /* Обратите внимание, что обычное приложение не должно предоставлять

 81:     * этот параметр пользователю; он присутствует здесь, чтобы можно было

 82:     * проверить это приложение, не производя изменений в системе,

 83:     * требующих доступа root.

 84:     */

 85:   service = "pamexample";

 86:  }

 87:

 88:  if (!username) {

 89:   /* по умолчанию для текущего пользователя */

 90:   if (!(pw = getpwuid (getuid())) ) {

 91:    fprintf(stderr, "Имя пользователя не существует");

 92:    exit(1);

 93:   }

 94:   username = strdup(pw->pw_name);

 95:  }

 96:

 97:  с = pam_start(service, username, &my_conv, &pamh);

 98:  check_success(pamh, c);

 99:

100:  с = pam_authenticate(pamh, 0);

101:  check_success(pamh, c);

102:

103:  if (account) {

104:   /* если аутентификация не была закончена, управление

105:    * учетной записью не определено

106:    */

107:   с = pam_acct_mgmt(pamh, 0);

108:   check_success(pamh, с);

109:  }

110:

111:  if (session) {

112:   /* В случае необходимости мы могли бы организовывать здесь ветвление */

113:   с = pam_open_session(pamh, 0);

114:   check_success(pamh, с);

115:

116:   /* Обратите внимание, что здесь не устанавливается uid, gid

117:      или дополнительные группы */

118:   с = pam_setcred(pamh, 0);

119:

120:   /* В случае необходимости мы могли бы закрыть здесь полномочия */

121:

122:   /* Вызов оболочки, которая была "аутентифицирована" */

123:   printf("Запуск оболочки...\n");

124:   system("exec bash -");

125:

126:   /* Здесь мы могли бы использовать wait4(), если бы организовывали

127:      ветвление вместо вызова system() */

128:   с = pam_close_session(pamh, 0);

129:   check_success(pamh, с);

130:  }

131:

132:  /* Реальные приложения могли бы сообщать о сбое вместо

133:   * выхода, что мы и делали в check_success на каждой стадии,

134:   * поэтому в таких случаях с может иметь значения, отличные

135:   * от PAM_SUCCESS.

136:   */

137:  с = pam_end(pamh, с);

138:  check_success(pamh, с);

139:

140:  return 0;

141: }

В этом приложении показаны все локальные заголовочные файлы для исходного кода, рассмотренного в книге.

1: /* libhello.h */

2:

3: #ifndef LIBHELLO_H_

4: #define LIBHELLO_H_

5:

6: void print_hello(void);

7:

8: #endif /* LIBHELLO_H_ */

1: /* ptypair.h */

2:

3: #ifndef _PTYPAIR_H

4: #define _PTYPAIR_H

5: int get_master_pty(char **name);

6: int get_slave_pty(char *name);

7: #endif /* _PTYPAIR_H */

1: /* sockutil.h */

2:

3: void die(char * message);

4: void copyData(int from, int to);

5: #ifndef CMSG_DATA

6: #define CMSG_DATA (cmsg) ((cmsg)->cmsg_data)

7: #endif

  1: /* ladsh4.c */

  2:

  3: #define _GNU_SOURCE

  4:

  5: #include <ctype.h>

  6: #include <errno.h>

  7: #include <fcntl.h>

  8: #include <glob.h>

  9: #include <signal.h>

 10: #include <stdio.h>

 11: #include <stdlib.h>

 12: #include <string.h>

 13: #include <sys/ioctl.h>

 14: #include <sys/wait.h>

 15: #include <unistd.h>

 16:

 17: #define MAX_COMMAND_LEN 250 /* максимальная длина одной

 18:                                командной строки */

 19: #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"

 20:

 21: struct jobSet {

 22:  struct job * head; /* заголовок списка выполняющихся заданий */

 23:  struct job * fg; /* текущее высокоприоритетное задание */

 24: };

 25:

 26: enum redirectionType { REDIRECT_INPUT, REDIRECT_OVERWRITE,

 27:                        REDIRECT_APPEND };

 28:

 29: struct redirectionSpecifier {

 30:  enum redirectionType type; /* тип переадресации */

 31:  int fd;                    /* переадресация fd */

 32:  char * filename;           /* файл, в который будет переадресовано fd */

 33: };

 34:

 35: struct childProgram {

 36:  pid_t pid;                                 /* 0 в случае выхода */

 37:  char ** argv;                              /* имя программы и аргументы */

 38:  int numRedirections;                       /* элементы в массиве переадресации */

 39:  struct redirectionSpecifier* redirections; /* переадресации ввода-вывода */

 40:  glob_t globResult;                         /* результат универсализации параметра */

 41:  int freeGlob;                              /* нужно ли освобождать globResult? */

 42:  int isStopped;                             /* выполняется ли в данный момент программа?*/

 43: };

 44:

 45: struct job {

 46:  int jobId;        /* номер задания */

 47:  int numProgs;     /* количество программ в задании */

 48:  int runningProgs; /* количество выполняющихся программ */

 49:  char * text;      /* имя задания */

 50:  char * cmdBuf;    /* буфер, на который ссылаются различные массивы argv */

 51:  pid_t pgrp;       /* идентификатор группы процесса для задания */

 52:  struct childProgram* progs; /* массив программ в задании */

 53:  struct job* next; /* для отслеживания фоновых команд */

 54:  int stoppedProgs; /* количество активных, но приостановленных программ */

 55: };

 56:

 57: void freeJob (struct job * cmd) {

 58:  int i;

 59:

 60:  for (i = 0; i <cmd->numProgs; i++) {

 61:   free(cmd->progs[i].argv);

 62:   if (cmd->progs[i].redirections)

 63:    free(cmd->progs[i].redirections);

 64:   if (cmd->progs[i].freeGlob)

 65:    globfree(&cmd->progs[i].globResult);

 66:  }

 67:  free(cmd->progs);

 68:  if (cmd->text) free(cmd->text);

 69:   free(cmd->cmdBuf);

 70: }

 71:

 72: int getCommand(FILE * source, char * command) {

 73:  if (source == stdin) {

 74:   printf("# ");

 75:   fflush(stdout);

 76:  }

 77:

 78:  if (!fgets(command, MAX_COMMAND_LEN, source)) {

 79:   if (source == stdin) printf("\n");

 80:   return 1;

 81:  }

 82:

 83:  /* удаление хвостового символа новой строки */

 84:  command[strlen(command) - 1] = '\0';

 85:

 86:  return 0;

 87: }

 88:

 89: void globLastArgument(struct childProgram * prog, int * argcPtr,

 90:  int * argcAllocedPtr) {

 91:  int argc = *argcPtr;

 92:  int argcAlloced = *argcAllocedPtr;

 93:  int rc;

 94:  int flags;

 95:  int i;

 96:  char * src, * dst;

 98:  if (argc>1) { /* cmd->globResult уже инициализирован */

 99:   flags = GLOB_APPEND;

100:   i = prog->globResult.gl_pathc;

101:  } else {

102:   prog->freeGlob = 1;

103:   flags = 0;