/* 
 * The contents of this file are subject to the Netscape Public
 * License Version 1.1 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of
 * the License at http://www.mozilla.org/NPL/
 *  
 * Software distributed under the License is distributed on an "AS
 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * rights and limitations under the License.
 *  
 * The Original Code is Mozilla Communicator client code, released
 * March 31, 1998.
 * 
 * The Initial Developer of the Original Code is Netscape
 * Communications Corporation. Portions created by Netscape are
 * Copyright (C) 1998-1999 Netscape Communications Corporation. All
 * Rights Reserved.
 * 
 * Contributor(s): 
 */

/******************************************************
 *
 *  argpin.c - Returns pin for token specified in a
 *  command line paramenter.
 *
 ******************************************************/

#include <stdio.h>
#include <string.h>

#include "argpin.h"

struct SVRCOREArgPinObj
{
  SVRCOREPinObj base;

  char *tokenName;
  char *password;
  SVRCOREPinObj *alt;
};
static const struct SVRCOREPinMethods vtable;

/*  XXXceb these are two hacks to fix a problem with the debug builds
 *  of svrcore.  With the optimizer turned off, there is a situation
 *  in user.c, where these two functions need to be available for the
 *  linker (they are imported, and no lib exports them, since they are
 *  declared static on XP_UNIX platforms) The short term hack solution
 *  is to define them here.  Yeah, it is ugly but, it will need to be
 *  here, until a new version of svrcore is done. 
 */


void echoOff(int fd)
{
}
 
void echoOn(int fd)
{
}


/* ------------------------------------------------------------ */
SVRCOREError
SVRCORE_CreateArgPinObj(SVRCOREArgPinObj **out, const char * tokenName, const char *password, SVRCOREPinObj *pinObj)
{
  SVRCOREError err = 0;
  SVRCOREArgPinObj *obj = 0;

  do {
    obj = (SVRCOREArgPinObj*)malloc(sizeof (SVRCOREArgPinObj));
    if (!obj) { err = 1; break; }

    obj->base.methods = &vtable;
    obj->tokenName=NULL;
    obj->password=NULL;
    obj->alt=pinObj;

    if ( tokenName == NULL) {
      PK11SlotInfo *slot = PK11_GetInternalKeySlot();

      obj->tokenName = strdup(PK11_GetTokenName(slot));
      PK11_FreeSlot(slot);
    }
    else
    {
      obj->tokenName = strdup(tokenName);
    }
    if (obj->tokenName == NULL) { err = 1; break; }

    obj->password = strdup(password);
    if (obj->password == NULL) { err = 1; break; }
  } while(0);

  if (err)
  {
    SVRCORE_DestroyArgPinObj(obj);
    obj = 0;
  }

  *out = obj;
  return err;
}

void
SVRCORE_DestroyArgPinObj(SVRCOREArgPinObj *obj)
{
  if (obj->tokenName) free(obj->tokenName);
  if (obj->password)
  {
    memset(obj->password, 0, strlen(obj->password));
    free(obj->password);
  }
  if (obj) free(obj);
}

static void destroyObject(SVRCOREPinObj *obj)
{
  SVRCORE_DestroyArgPinObj((SVRCOREArgPinObj*)obj);
}

static char *getPin(SVRCOREPinObj *obj, const char *tokenName, PRBool retry)
{
    SVRCOREArgPinObj *p = (SVRCOREArgPinObj*)obj;

    if (tokenName == NULL) return NULL;

    /* On first attempt, return the password if the token name
     * matches.
     */
    if (!retry && strcmp(p->tokenName, tokenName) == 0)
    {
      return strdup(p->password);
    }

    if (p->alt != NULL) return SVRCORE_GetPin(p->alt, tokenName, retry);

    return (NULL);
}

/*
 * VTable
 */
static const SVRCOREPinMethods vtable =
{ 0, 0, destroyObject, getPin };