/[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 297 by matthewc, Tue Jan 28 12:27:28 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 445  error(char *format, ...) Line 447  error(char *format, ...)
447          va_end(ap);          va_end(ap);
448  }  }
449    
450    /* report a warning */
451    void
452    warning(char *format, ...)
453    {
454            va_list ap;
455    
456            fprintf(stderr, "WARNING: ");
457    
458            va_start(ap, format);
459            vfprintf(stderr, format, ap);
460            va_end(ap);
461    }
462    
463  /* report an unimplemented protocol feature */  /* report an unimplemented protocol feature */
464  void  void
465  unimpl(char *format, ...)  unimpl(char *format, ...)
# Line 492  hexdump(unsigned char *p, unsigned int l Line 507  hexdump(unsigned char *p, unsigned int l
507  int  int
508  load_licence(unsigned char **data)  load_licence(unsigned char **data)
509  {  {
510          char path[PATH_MAX];          char *path;
511          char *home;          char *home;
512          struct stat st;          struct stat st;
513          int fd;          int fd;
# Line 501  load_licence(unsigned char **data) Line 516  load_licence(unsigned char **data)
516          if (home == NULL)          if (home == NULL)
517                  return -1;                  return -1;
518    
519          STRNCPY(path, home, sizeof(path));          path = xmalloc(strlen(home) + strlen(hostname) + 20);
520          strncat(path, "/.rdesktop/licence", sizeof(path) - strlen(path) - 1);          sprintf(path, "%s/.rdesktop/licence.%s", home, hostname);
521    
522          fd = open(path, O_RDONLY);          fd = open(path, O_RDONLY);
523          if (fd == -1)          if (fd == -1)
# Line 518  load_licence(unsigned char **data) Line 533  load_licence(unsigned char **data)
533  void  void
534  save_licence(unsigned char *data, int length)  save_licence(unsigned char *data, int length)
535  {  {
536          char path[PATH_MAX];          char *fpath;            /* file path for licence */
537            char *fname, *fnamewrk; /* file name for licence .inkl path. */
538          char *home;          char *home;
539          int fd;          uint32 y;
540            struct flock fnfl;
541            int fnfd, fnwrkfd, i, wlen;
542            struct stream s, *s_ptr;
543            uint32 len;
544    
545            /* Construct a stream, so that we can use macros to extract the
546             * licence.
547             */
548            s_ptr = &s;
549            s_ptr->p = data;
550            /* Skip first two bytes */
551            in_uint16(s_ptr, len);
552    
553            /* Skip three strings */
554            for (i = 0; i < 3; i++)
555            {
556                    in_uint32(s_ptr, len);
557                    s_ptr->p += len;
558                    /* Make sure that we won't be past the end of data after
559                     * reading the next length value
560                     */
561                    if ((s_ptr->p) + 4 > data + length)
562                    {
563                            printf("Error in parsing licence key.\n");
564                            printf("Strings %d end value %x > supplied length (%x)\n",
565                                   i, s_ptr->p, data + length);
566                            return;
567                    }
568            }
569            in_uint32(s_ptr, len);
570            if (s_ptr->p + len > data + length)
571            {
572                    printf("Error in parsing licence key.\n");
573                    printf("End of licence %x > supplied length (%x)\n", s_ptr->p + len, data + length);
574                    return;
575            }
576    
577          home = getenv("HOME");          home = getenv("HOME");
578          if (home == NULL)          if (home == NULL)
579                  return;                  return;
580    
581          STRNCPY(path, home, sizeof(path));          /* set and create the directory -- if it doesn't exist. */
582          strncat(path, "/.rdesktop", sizeof(path) - strlen(path) - 1);          fpath = xmalloc(strlen(home) + 11);
583          mkdir(path, 0700);          STRNCPY(fpath, home, strlen(home) + 1);
584    
585            sprintf(fpath, "%s/.rdesktop", fpath);
586            if (mkdir(fpath, 0700) == -1 && errno != EEXIST)
587            {
588                    perror("mkdir");
589                    exit(1);
590            }
591    
592          strncat(path, "/licence", sizeof(path) - strlen(path) - 1);          /* set the real licence filename, and put a write lock on it. */
593            fname = xmalloc(strlen(fpath) + strlen(hostname) + 10);
594            sprintf(fname, "%s/licence.%s", fpath, hostname);
595            fnfd = open(fname, O_RDONLY);
596            if (fnfd != -1)
597            {
598                    fnfl.l_type = F_WRLCK;
599                    fnfl.l_whence = SEEK_SET;
600                    fnfl.l_start = 0;
601                    fnfl.l_len = 1;
602                    fcntl(fnfd, F_SETLK, &fnfl);
603            }
604    
605            /* create a temporary licence file */
606            fnamewrk = xmalloc(strlen(fname) + 12);
607            for (y = 0;; y++)
608            {
609                    sprintf(fnamewrk, "%s.%lu", fname, y);
610                    fnwrkfd = open(fnamewrk, O_WRONLY | O_CREAT | O_EXCL, 0600);
611                    if (fnwrkfd == -1)
612                    {
613                            if (errno == EINTR || errno == EEXIST)
614                                    continue;
615                            perror("create");
616                            exit(1);
617                    }
618                    break;
619            }
620            /* write to the licence file */
621            for (y = 0; y < len;)
622            {
623                    do
624                    {
625                            wlen = write(fnwrkfd, s_ptr->p + y, len - y);
626                    }
627                    while (wlen == -1 && errno == EINTR);
628                    if (wlen < 1)
629                    {
630                            perror("write");
631                            unlink(fnamewrk);
632                            exit(1);
633                    }
634                    y += wlen;
635            }
636    
637          fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0600);          /* close the file and rename it to fname */
638          if (fd == -1)          if (close(fnwrkfd) == -1)
639          {          {
640                  perror("open");                  perror("close");
641                  return;                  unlink(fnamewrk);
642                    exit(1);
643            }
644            if (rename(fnamewrk, fname) == -1)
645            {
646                    perror("rename");
647                    unlink(fnamewrk);
648                    exit(1);
649            }
650            /* close the file lock on fname */
651            if (fnfd != -1)
652            {
653                    fnfl.l_type = F_UNLCK;
654                    fnfl.l_whence = SEEK_SET;
655                    fnfl.l_start = 0;
656                    fnfl.l_len = 1;
657                    fcntl(fnfd, F_SETLK, &fnfl);
658                    close(fnfd);
659          }          }
660    
         write(fd, data, length);  
         close(fd);  
661  }  }
662  #endif  #endif

Legend:
Removed from v.279  
changed lines
  Added in v.297

  ViewVC Help
Powered by ViewVC 1.1.26