/[rdesktop]/sourceforge.net/trunk/rdesktop/rdesktop.c
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /sourceforge.net/trunk/rdesktop/rdesktop.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 279 by n-ki, Tue Nov 26 10:09:14 2002 UTC revision 289 by astrand, Thu Jan 16 13:48:02 2003 UTC
# Line 21  Line 21 
21  #include <stdarg.h>             /* va_list va_start va_end */  #include <stdarg.h>             /* va_list va_start va_end */
22  #include <unistd.h>             /* read close getuid getgid getpid getppid gethostname */  #include <unistd.h>             /* read close getuid getgid getpid getppid gethostname */
23  #include <fcntl.h>              /* open */  #include <fcntl.h>              /* open */
24    #include <errno.h>              /* save licence uses it. */
25  #include <pwd.h>                /* getpwuid */  #include <pwd.h>                /* getpwuid */
 #include <limits.h>             /* PATH_MAX */  
26  #include <termios.h>            /* tcgetattr tcsetattr */  #include <termios.h>            /* tcgetattr tcsetattr */
27  #include <sys/stat.h>           /* stat */  #include <sys/stat.h>           /* stat */
28  #include <sys/time.h>           /* gettimeofday */  #include <sys/time.h>           /* gettimeofday */
# Line 137  main(int argc, char *argv[]) Line 137  main(int argc, char *argv[])
137          uint32 flags;          uint32 flags;
138          char *p;          char *p;
139          int c;          int c;
140            int username_option = 0;
141    
142          flags = RDP_LOGON_NORMAL;          flags = RDP_LOGON_NORMAL;
143          prompt_password = False;          prompt_password = False;
# Line 149  main(int argc, char *argv[]) Line 150  main(int argc, char *argv[])
150                  {                  {
151                          case 'u':                          case 'u':
152                                  STRNCPY(username, optarg, sizeof(username));                                  STRNCPY(username, optarg, sizeof(username));
153                                    username_option = 1;
154                                  break;                                  break;
155    
156                          case 'd':                          case 'd':
# Line 259  main(int argc, char *argv[]) Line 261  main(int argc, char *argv[])
261                  *p = 0;                  *p = 0;
262          }          }
263    
264          if (username[0] == 0)          if (!username_option)
265          {          {
266                  pw = getpwuid(getuid());                  pw = getpwuid(getuid());
267                  if ((pw == NULL) || (pw->pw_name == NULL))                  if ((pw == NULL) || (pw->pw_name == NULL))
# Line 492  hexdump(unsigned char *p, unsigned int l Line 494  hexdump(unsigned char *p, unsigned int l
494  int  int
495  load_licence(unsigned char **data)  load_licence(unsigned char **data)
496  {  {
497          char path[PATH_MAX];          char *path;
498          char *home;          char *home;
499          struct stat st;          struct stat st;
500          int fd;          int fd;
# Line 501  load_licence(unsigned char **data) Line 503  load_licence(unsigned char **data)
503          if (home == NULL)          if (home == NULL)
504                  return -1;                  return -1;
505    
506          STRNCPY(path, home, sizeof(path));          path = xmalloc(strlen(home) + strlen(hostname) + 20);
507          strncat(path, "/.rdesktop/licence", sizeof(path) - strlen(path) - 1);          sprintf(path, "%s/.rdesktop/licence.%s", home, hostname);
508    
509          fd = open(path, O_RDONLY);          fd = open(path, O_RDONLY);
510          if (fd == -1)          if (fd == -1)
# Line 518  load_licence(unsigned char **data) Line 520  load_licence(unsigned char **data)
520  void  void
521  save_licence(unsigned char *data, int length)  save_licence(unsigned char *data, int length)
522  {  {
523          char path[PATH_MAX];          char *fpath;            /* file path for licence */
524            char *fname, *fnamewrk; /* file name for licence .inkl path. */
525          char *home;          char *home;
526          int fd;          uint32 y;
527            struct flock fnfl;
528            int fnfd, fnwrkfd, i, wlen;
529            struct stream s, *s_ptr;
530            uint32 len;
531    
532            /* Construct a stream, so that we can use macros to extract the
533             * licence.
534             */
535            s_ptr = &s;
536            s_ptr->p = data;
537            /* Skip first two bytes */
538            in_uint16(s_ptr, len);
539    
540            /* Skip three strings */
541            for (i = 0; i < 3; i++)
542            {
543                    in_uint32(s_ptr, len);
544                    s_ptr->p += len;
545                    /* Make sure that we won't be past the end of data after
546                     * reading the next length value
547                     */
548                    if ((s_ptr->p) + 4 > data + length)
549                    {
550                            printf("Error in parsing licence key.\n");
551                            printf("Strings %d end value %x > supplied length (%x)\n",
552                                   i, s_ptr->p, data + length);
553                            return;
554                    }
555            }
556            in_uint32(s_ptr, len);
557            if (s_ptr->p + len > data + length)
558            {
559                    printf("Error in parsing licence key.\n");
560                    printf("End of licence %x > supplied length (%x)\n", s_ptr->p + len, data + length);
561                    return;
562            }
563    
564          home = getenv("HOME");          home = getenv("HOME");
565          if (home == NULL)          if (home == NULL)
566                  return;                  return;
567    
568          STRNCPY(path, home, sizeof(path));          /* set and create the directory -- if it doesn't exist. */
569          strncat(path, "/.rdesktop", sizeof(path) - strlen(path) - 1);          fpath = xmalloc(strlen(home) + 11);
570          mkdir(path, 0700);          STRNCPY(fpath, home, strlen(home) + 1);
571    
572          strncat(path, "/licence", sizeof(path) - strlen(path) - 1);          sprintf(fpath, "%s/.rdesktop", fpath);
573            if (mkdir(fpath, 0700) == -1 && errno != EEXIST)
574            {
575                    perror("mkdir");
576                    exit(1);
577            }
578    
579          fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0600);          /* set the real licence filename, and put a write lock on it. */
580          if (fd == -1)          fname = xmalloc(strlen(fpath) + strlen(hostname) + 10);
581            sprintf(fname, "%s/licence.%s", fpath, hostname);
582            fnfd = open(fname, O_RDONLY);
583            if (fnfd != -1)
584            {
585                    fnfl.l_type = F_WRLCK;
586                    fnfl.l_whence = SEEK_SET;
587                    fnfl.l_start = 0;
588                    fnfl.l_len = 1;
589                    fcntl(fnfd, F_SETLK, &fnfl);
590            }
591    
592            /* create a temporary licence file */
593            fnamewrk = xmalloc(strlen(fname) + 12);
594            for (y = 0;; y++)
595            {
596                    sprintf(fnamewrk, "%s.%lu", fname, y);
597                    fnwrkfd = open(fnamewrk, O_WRONLY | O_CREAT | O_EXCL, 0600);
598                    if (fnwrkfd == -1)
599                    {
600                            if (errno == EINTR || errno == EEXIST)
601                                    continue;
602                            perror("create");
603                            exit(1);
604                    }
605                    break;
606            }
607            /* write to the licence file */
608            for (y = 0; y < len;)
609          {          {
610                  perror("open");                  do
611                  return;                  {
612                            wlen = write(fnwrkfd, s_ptr->p + y, len - y);
613                    }
614                    while (wlen == -1 && errno == EINTR);
615                    if (wlen < 1)
616                    {
617                            perror("write");
618                            unlink(fnamewrk);
619                            exit(1);
620                    }
621                    y += wlen;
622            }
623    
624            /* close the file and rename it to fname */
625            if (close(fnwrkfd) == -1)
626            {
627                    perror("close");
628                    unlink(fnamewrk);
629                    exit(1);
630            }
631            if (rename(fnamewrk, fname) == -1)
632            {
633                    perror("rename");
634                    unlink(fnamewrk);
635                    exit(1);
636            }
637            /* close the file lock on fname */
638            if (fnfd != -1)
639            {
640                    fnfl.l_type = F_UNLCK;
641                    fnfl.l_whence = SEEK_SET;
642                    fnfl.l_start = 0;
643                    fnfl.l_len = 1;
644                    fcntl(fnfd, F_SETLK, &fnfl);
645                    close(fnfd);
646          }          }
647    
         write(fd, data, length);  
         close(fd);  
648  }  }
649  #endif  #endif
Legend:
Removed from v.279  
changed lines
  Added in v.289
  ViewVC Help
Powered by ViewVC 1.1.26